ToolChain-CRC: Conformal Risk Control for Agentic AI Under Retrieval and Tool-Use Drift
On the map: LLM agents and tool use → Tool learning and function calling
Abstract (arXiv)
Modern AI agents retrieve documents, call tools, check intermediate information, and then produce a final answer or action. This creates a risk-control problem that is not visible from the final answer alone. A final response may look acceptable even when the retrieval was weak, a tool output was wrong, or an earlier step was unsupported. We propose ToolChain-CRC, a conformal risk-control method for retrieval-augmented and tool-using agents under drift. The method treats each agent run as a full trajectory of actions, observations, and final output. It builds step-level risk scores, combines them into a trajectory risk score, calibrates an accept-or-intervene rule, and adds an anytime alarm that can stop risky runs before the final answer. We prove trajectory-level risk control under exchangeable calibration runs, give a drift-aware extension with auditable constants, and prove an anytime escalation rule through a supermartingale construction. Experiments cover synthetic tool-chain drift, RAG/tool-use stress tests, public SQuAD-derived retrieval tasks, an API-free agentic QA case study, ablations, target-risk sensitivity checks, 20-seed robustness checks, a drift-margin audit, and a live RAG/tool-use agent benchmark. Across these settings, final-answer-only calibration can miss retrieval and tool failures, while trajectory-level calibration keeps accepted-trajectory risk below the target.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Placed in Tool learning and function calling, one of 16 directions in LLM agents and tool use. Builds on 10 of its 15 references that are in the map. First posted on arXiv 16 Jun 2026.
Builds on
10 of 15 references.
References · 10
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
- 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 792
- 2022Conformal Risk Controlcited by 299
- 2018Time-uniform, nonparametric, nonasymptotic confidence sequencescited by 229
- 2022Game-theoretic statistics and safe anytime-valid inferencecited by 155
- 2025Tools in the Loop: Quantifying Uncertainty of LLM Question Answering Systems That Use Toolscited by 9
- 2026Conformal Selective Acting: Anytime-Valid Risk Control for RLVR-Trained LLMscited by 8
- 2026Anytime-Valid Conformal Risk Controlcited by 5
- 2025SafePath: Conformal Prediction for Safe LLM-Based Autonomous Navigationcited by 4
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Beyond Aggregate Risk: Role-Stratified Conformal Risk Control for LLM Tool Callscited by 1
- 2026What Does a Skill Actually Do? Estimands and Evaluation Validity for Tool and Skill Use in LLM Agents: A Critical Reviewcited by 1
- 2026Signed Evidence Flow: Conflict-Aware and Stability-Calibrated Data Analysiscited by 0
- 2026What Can Component-Replacement Evidence Establish? A Critical Scoping Review of Local Decisions in LLM Agentscited by 0
- 2026G-CARB: Graph-Localized Conformal Agent Risk Budget for Compositional Harmcited by 0
- 2026Finding Blind Spots in AppWorld and WorkArena Task Verifierscited by 0
- 2026Finding Blind Spots in AppWorld and WorkArena Task Verifiers6 Oct 2026
- 2026G-CARB: Graph-Localized Conformal Agent Risk Budget for Compositional Harm4 Oct 2026
- 2026What Can Component-Replacement Evidence Establish? A Critical Scoping Review of Local Decisions in LLM Agents30 Sep 2026
- 2026What Does a Skill Actually Do? Estimands and Evaluation Validity for Tool and Skill Use in LLM Agents: A Critical Review27 Sep 2026
- 2026Beyond Aggregate Risk: Role-Stratified Conformal Risk Control for LLM Tool Calls27 Jul 2026
- 2026Signed Evidence Flow: Conflict-Aware and Stability-Calibrated Data Analysis20 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026Risk-Aware Reranking for Agentic Tool Retrieval
- 2026SoK: Agentic Retrieval-Augmented Generation (RAG): Taxonomy, Architectures, Evaluation, and Research Directions
- 2026AgentTune: A Unified Training Lifecycle for Tool-Using LLM Agents
- 2026TRACER: Trajectory Risk Aggregation for Critical Episodes in Agentic Reasoning
- 2026AgentRiskBOM: A Risk-Scoping Security Bill of Materials for Agentic AI Systems
- 2026ARTIS: Agentic Risk-Aware Test-Time Scaling via Iterative Simulation
- 2026TAPO: Tool-Aware Policy Optimization via Credit Transfer for Multimodal Search Agents
- 2025Dynamic System Instructions and Tool Exposure for Efficient Agentic LLMs