Prospective Hindsight: Self-Calibrating Reinforcement Learning via Prediction-Reality Gaps
cs.LG, cs.AI
Submitted: 2026-10-02
Updated: 2026-10-02
Comments: NeurIPS 2026
License: http://creativecommons.org/licenses/by/4.0/
The gist: Reinforcement learning for long-horizon agents relies on purely retrospective training signals: credit is assigned only after observing environmental consequences, leaving the agent's belief at
Terminology
Abstract
Reinforcement learning for long-horizon agents relies on purely retrospective training signals: credit is assigned only after observing environmental consequences, leaving the agent's belief at action time invisible to the gradient. We introduce Prospective Hindsight (PH), a self-calibrating training principle that augments any retrospective base method with a signal derived from the gap between the agent's prospective prediction (before feedback) and the retrospective evaluation (after feedback). This per-rollout surprise identifies samples where the agent's self-model is most inaccurate and amplifies their gradient contribution through a stop-gradient surprise-weighted advantage. Since the prospective predictor shares parameters with the policy, the two co-evolve, progressively shifting focus to the agent's remaining blind spots. We connect this principle to a privileged-information gap and show that minimizing the surprise residual provides a descent pathway on the agent's miscalibration rate; calibration thus emerges as a byproduct of optimization rather than from an added objective. On single-turn verifiable tasks and a multi-turn personal-agent task (under GRPO, on-policy distillation, and their combination), PH improves both task performance and calibration, with consistent gains across model scales. Notably, the dominant miscalibration mode shifts structurally between regimes, overconfident failures in single-turn, underconfident successes in multi-turn, yet the same training principle addresses both successfully.
Sources
- gpt-oss-120b & gpt-oss-20b Model Card
- Aligning Language Models from User Interactions
- Rethinking Agentic Reinforcement Learning In Large Language Models
- Beyond Binary Rewards: Training LMs to Reason About Their Uncertainty
- HDPO: Hybrid Distillation Policy Optimization via Privileged Self-Distillation
- Agentic Reinforced Policy Optimization
- UProp: Investigating the Uncertainty Propagation of LLMs in Multi-Step Agentic Decision-Making
- SciKnowEval: Evaluating Multi-level Scientific Knowledge of Large Language Models
- Overconfidence is Key: Verbalized Uncertainty Evaluation in Large Language and Vision-Language Models
- Reinforcement Learning via Self-Distillation
- Language Models (Mostly) Know What They Know
- Why Language Models Hallucinate
- Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?
- Taming Overconfidence in LLMs: Reward Calibration in RLHF
- Teaching Models to Express Their Uncertainty in Words
- Unifying distillation and privileged information
- Uncertainty Quantification in LLM Agents: Foundations, Emerging Challenges, and Opportunities
- Olmo 3
- CRISP: Compressed Reasoning via Iterative Self-Policy Distillation
- Proximal Policy Optimization Algorithms
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks