cs.AIAug 26, 2026

CaSKG: Counterfactual-Causal Skill Graphs for Scalable Agent Skill Retrieval

Authors: Zhiyuan Li, Linyuan Gao, Xuechun Ding, Hongwei Chen, Yuan Wu, Yi Chang

Organizations: School of Artificial Intelligence, Jilin University Ant Group Changchun, China · Ant Group Hangzhou, China

Abstract

Reusable skill libraries allow large language model (LLM) agents to reuse procedural knowledge across tasks, but they also turn memory access into a challenging retrieval problem. Full-library prompting preserves coverage at high context cost, vector retrieval returns compact neighborhoods but treats skills as independent text, and graph-based retrieval can recover workflow context only when the edges that carry relevance are reliable. We propose CaSKG, a counterfactual-causal skill graph framework that calibrates procedural relations before retrieval. CaSKG first builds a high-recall directed candidate graph from semantic, lexical, input/output, and structural evidence, with repair evidence and an optional LLM judge further refining candidate scores. It then applies direction-conditioned textual counterfactual probes that remove, substitute, and reorder skill pairs, aggregates the evidence with Bayesian smoothing, and publishes a state-filtered weighted graph for task-conditioned expansion. The graph is constructed offline and used without changing the downstream agent policy or task interface. Across six LLM backbones on ALFWorld ID-140 and ScienceWorld U211, CaSKG achieves the highest task score in all twelve combinations of model and benchmark. Relative to Graph-of-Skills (GoS), it improves the six-model macro-average ScienceWorld score from 72.62 to 80.50 and ALFWorld success from 80.01% to 86.79%, while reducing mean environment steps on both benchmarks. Qualitative and ablation analyses further show that calibrated edges help retrieval preserve prerequisites, state-changing actions, verification routines, and final completion steps. These results position edge-confidence calibration as an effective route to compact and executable skill retrieval at scale\footnote{Code is available at: https://github.com/ZhiyuanLi218/Caskg }.

Figures & tables

Explore similar work

CardsList
  1. Graph-of-Skills: Dependency-Aware Structural Retrieval for Massive Agent Skills

    Apr 7, 2026Dawei Liu, Zongxia Li, Hongyang Du +4Tool-Augmented Language Model AgentsGraph-Based Retrieval

  2. SkillDAG: Self-Evolving Typed Skill Graphs for LLM Skill Selection at Scale

    Jun 2, 2026Tong Bai, Zhenglin Wan, Pengfei Zhou +3LLM Agent Skill RetrievalDynamic Graph Learning

  3. SkillTrace: Traversing a Query-Skill Graph for Composable LLM Agents

    Aug 3, 2026Yue Yao, Shengyuan Wang, Xin Chen +4Graph-Based RetrievalLLM Agent Skill Retrieval