Can an Agent Learn to Predict Its Own Failure? Outcome-Grounded Confidence Calibration in LLM Agents
New in LLM agents and tool use; no direction yet. LLM agents and tool use
Abstract (arXiv)
Controlling an acting LLM agent means deciding, at each step of a long-horizon task, whether to trust its next action, abstain, or hand off to a human. That decision needs a per-step confidence signal predicting whether the current rollout will eventually succeed. The agent's own self-reported verbalized confidence is the natural candidate, but existing reinforcement learning methods calibrate it against the correctness of a decision at the current step. We propose a method to calibrate it against the eventual outcome of a multi-step episode: under GRPO, the group of rollouts for a task supplies success labels, and a proper scoring rule scores each step's confidence against the realized outcome. In offline RL on recorded trajectories of a Qwen2.5-7B agent in ALFWorld, TextWorld, and WebShop environments, the AUROC of step confidence against episode outcome rises from near chance to 0.68-0.85, and the top-10% of steps the agent is most confident about succeed ~4.3x more often than average. In online RL on the live ALFWorld agent, four calibration rewards improve early-step AUROC, expected calibration error goes from 0.42 to below 0.12, and selective execution improves over both the untrained agent and a format-only RL control, without degrading task success. We also introduce an evaluation protocol that separates episode depth, task difficulty, and attempt-level foresight, which shows where the gain lies: the trained agent recognizes from its first step which tasks it will fail, while telling a succeeding attempt from a failing one at the same task is harder, with some early-window signal emerging on TextWorld.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 32 references that are in the map. First posted on arXiv 13 Oct 2026.
Builds on
21 of 32 references.
References · 21
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2023WebArena: A Realistic Web Environment for Building Autonomous Agentscited by 2,006
- 2022WebShop: Towards Scalable Real-World Web Interaction with Grounded Language Agentscited by 1,251
- 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,205
- 2023Can LLMs Express Their Uncertainty? An Empirical Evaluation of Confidence Elicitation in LLMscited by 974
- 2017Selective Classification for Deep Neural Networkscited by 852
- 2023Just Ask for Calibration: Strategies for Eliciting Calibrated Confidence Scores from Language Models Fine-Tuned with Human Feedbackcited by 812
- 2019ALFRED: A Benchmark for Interpreting Grounded Instructions for Everyday Taskscited by 783
- 2022Teaching Models to Express Their Uncertainty in Wordscited by 751
- 2022ScienceWorld: Is your Agent Smarter than a 5th Grader?cited by 327
- 2018TextWorld: A Learning Environment for Text-based Gamescited by 310
9 more
- 2020Selective Question Answering under Domain Shiftcited by 238
- 2019Interactive Fiction Games: A Colossal Adventurecited by 168
- 2025Beyond Binary Rewards: Training LMs to Reason About Their Uncertaintycited by 117
- 2024Taming Overconfidence in LLMs: Reward Calibration in RLHFcited by 112
- 2024SaySelf: Teaching LLMs to Express Confidence with Self-Reflective Rationalescited by 105
- 2025Rewarding Doubt: A Reinforcement Learning Approach to Calibrated Confidence Expression of Large Language Modelscited by 52
- 2026The Confidence Dichotomy: Analyzing and Mitigating Miscalibration in Tool-Use Agentscited by 16
- 2026Calibration-Aware Policy Optimization for Reasoning LLMscited by 5
- 2026Balancing Classification and Calibration Performance in Decision-Making LLMs via Calibration Aware Reinforcement Learningcited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Critic Experience Bank: Self-Evolving Step-Level Confidence Estimation for LLM Agents
- 2026Confidence Reasoning Graphs: Structured Confidence Estimation for LLM Agents
- 2026Testing-Driven Reliability Audit of Trajectory-Based Early Outcome Prediction for LLM Agents: Target-Specific Calibration Transfer Persists Within a Single Benchmark
- 2026Do Agents Know When They Succeed? Calibrating Agent Confidence from Internal Representations
- 2026Agentic Confidence Calibration
- 2026Shared Experience, Separate Learning: Companion Confidence Calibration for LLMs
- 2026Process Supervision of Confidence Margin for Calibrated LLM Reasoning
- 2026Confident at the moment of action: belief miscalibration in LLM play under hidden information