Can an Agent Learn to Predict Its Own Failure? Outcome-Grounded Confidence Calibration in LLM Agents
Abstract
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.
Community
Can an Agent Learn to Predict Its Own Failure? ๐ค
As part of our latest work, "Can an Agent Learn to Predict Its Own Failure? Outcome-Grounded Confidence Calibration in LLM Agents", we try to investigate this.
Most confidence-training approaches focus on whether a model's current answer or action is correct. But for an agent working through a long-horizon task, the more useful question is: Will this attempt eventually succeed?
๐ฏ We introduce a way to train agents to align their step-by-step confidence with the final outcome of a task. The approach uses the outcomes of sampled rollouts as learning signals, so the agent learns to treat confidence as a forecast of eventual success, not just a judgment of its next move.
In live ALFWorld experiments, calibration training:
๐ Reduced expected calibration error from 0.42 to as low as 0.05
๐ฆ Raised success among the most confident early steps from 49% to as high as 66%
โ
Improved early triage without reducing overall task success
The key finding is that the clearest gain is in recognizing which tasks the agent is likely to solve or fail. Distinguishing a successful attempt from a failing attempt at the same task still remains harder and is an important challenge for future work.
Get this paper in your agent:
hf papers read 2610.18614 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper