SAP paper teaches LLM agents to predict their own failure (AUROC 0.85)

SAP · hf · 2026-10-14

SAP researchers propose outcome-grounded confidence calibration: an LLM agent's verbalized confidence is trained (under GRPO) against the eventual success of a multi-step episode rather than single-step correctness. Offline RL on a Qwen2.5-7B agent across ALFWorld, TextWorld and WebShop lifts per-step AUROC from near chance to 0.68-0.85, with the top-10% most-confident steps succeeding 4.3x more often. Online RL on live ALFWorld cuts expected calibration error from 0.42 to below 0.12 and improves selective execution without hurting task success. A new evaluation protocol separates episode depth, difficulty, and attempt-level foresight: the trained agent spots doomed tasks from step one, though distinguishing succeeding vs failing attempts on the same task remains harder. Useful for runtime control decisions: continue, verify, or hand off to humans.

Original post →

More from coding & agent

coding & agent channel →