Papers
arxiv:2610.18614

Can an Agent Learn to Predict Its Own Failure? Outcome-Grounded Confidence Calibration in LLM Agents

Published on Oct 13
ยท Submitted by
Ashutosh Hathidara
on Oct 14
ยท SAP SAP

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

Paper author Paper submitter

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.

Sign up or log in to comment

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

Cite arxiv.org/abs/2610.18614 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2610.18614 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2610.18614 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.