CaSKG calibrates LLM agent skill graphs, tops rival on every benchmark

LLM agents that build up libraries of reusable skills, procedures they can call on again for later tasks, run into a retrieval problem once those libraries grow: how does the agent find the right skill for the task at hand? Zhiyuan Li and co-authors describe three existing options and why each falls short. Prompting the full library into context preserves everything but gets expensive as the library grows. Vector retrieval keeps that cost down by returning a compact set of similar skills, but it treats each skill as an isolated piece of text, missing how skills relate to one another in a workflow. Graph-based retrieval can recover that workflow context, connecting skills the way steps in a procedure connect, but only when the edges that carry the relevance are actually reliable.
The authors' answer is CaSKG, a counterfactual-causal skill graph framework that calibrates those procedural connections before retrieval happens rather than trusting them as built. CaSKG first assembles a high-recall directed candidate graph by drawing on semantic, lexical, input/output and structural evidence between skills, with repair evidence and an optional LLM judge sharpening the candidate scores further. It then runs direction-conditioned textual counterfactual probes on the graph: removing a connection between two skills, substituting one skill for another, and reordering pairs, to test which connections actually carry relevance rather than just surface similarity. The results of these probes are combined with Bayesian smoothing into a state-filtered weighted graph, which the agent expands based on its current task state. Because the calibration happens offline, the resulting graph can be dropped in without changing the downstream agent's own decision-making policy or how it interfaces with tasks.
The authors tested CaSKG across six different LLM backbones on two benchmarks, ALFWorld ID-140 and ScienceWorld U211, comparing it against Graph-of-Skills (GoS), a prior graph-based retrieval method. CaSKG scored highest in all twelve combinations of model and benchmark. Against GoS, it raised the six-model macro-average score on ScienceWorld from 72.62 to 80.50, and raised ALFWorld success from 80.01% to 86.79%. It also needed fewer environment steps on average to complete tasks on both benchmarks, though the paper does not give the exact size of that reduction. Follow-up qualitative and ablation analysis found that the calibrated edges specifically help retrieval preserve prerequisites, state-changing actions, verification routines and final completion steps within a retrieved skill sequence.
The authors position calibrating edge confidence this way as an effective route to compact, executable skill retrieval that scales as a library grows. Code for CaSKG is available on GitHub.
Key facts
- CaSKG calibrates an LLM agent's skill graph before retrieval by running counterfactual probes, removing, substituting and reordering skill pairs, to test which connections carry genuine relevance.
- Tested across six LLM backbones on ALFWorld ID-140 and ScienceWorld U211, CaSKG scored highest in all twelve model-benchmark combinations.
- Against the Graph-of-Skills (GoS) baseline, CaSKG raised the six-model macro-average ScienceWorld score from 72.62 to 80.50 and ALFWorld success from 80.01% to 86.79%.
- CaSKG also reduced the average number of environment steps needed on both benchmarks, though the paper gives no exact figure for the reduction.
- The calibrated graph is built offline and can be used without changing the downstream agent's policy or task interface; code is available on GitHub.
Why it matters
Agents that reuse skills need a way to find the right one as their library grows, and the existing options each trade something away. Prompting the whole library into context keeps everything available but gets expensive as the library grows. Vector retrieval keeps costs down but treats each skill as standalone text, losing the workflow context of how skills connect to each other. Graph-based retrieval can recover that context, but only if the graph's connections are actually reliable, and building a graph from surface similarity alone does not guarantee that. CaSKG's contribution is a way to calibrate those connections against causal evidence, using counterfactual tests, before the agent ever retrieves from the graph, rather than trusting a graph that was only ever checked for similarity.
Who it affects
This is aimed at researchers and engineers building LLM agents that maintain a growing library of reusable skills, procedures for multi-step tasks the agent can call on again later. It was evaluated specifically on ALFWorld and ScienceWorld, simulated environments used in agent research, across six different underlying language models, so it speaks most directly to teams working with similar skill-library architectures rather than to agents that lack a persistent skill memory.
How to use it
CaSKG is a preprocessing step: it builds a calibrated, weighted skill graph offline, before any task runs. That graph then plugs into retrieval at inference time without changing the downstream agent's own decision-making policy or how it interfaces with tasks, so an existing skill-library agent could in principle drop in a CaSKG-calibrated graph without further modification. The paper states that code is available on GitHub, at https://github.com/ZhiyuanLi218/Caskg.
How solid is it
This is a full evaluation, not just a proposal: six different LLM backbones, two benchmark suites, and CaSKG comes out ahead of the Graph-of-Skills baseline in all twelve resulting model-benchmark combinations. The authors back the headline numbers with separate qualitative and ablation analysis of which parts of a retrieved skill sequence the calibrated edges help preserve, specifically prerequisites, state-changing actions, verification routines, and final completion steps. On Hugging Face, the paper has drawn light attention so far: 5 points and 2 comments.
Risks and caveats
These are the authors' own reported results, without independent replication. The paper states that CaSKG reduces mean environment steps on both benchmarks but never says by how much. It names full-library prompting and vector retrieval as the other two retrieval approaches but gives no quantified comparison against either, only against Graph-of-Skills. It also does not explain what the 'task score' metric measures or how it is computed, nor what distinguishes the specific ALFWorld ID-140 and ScienceWorld U211 task sets from other splits of those benchmarks. No institutions or publication venue appear in the text itself.