🤖 AI Summary
This work addresses the issue of overconfidence in large language models trained via reinforcement learning with verifiable rewards (RLVR), where models often exhibit degraded calibration and assign high confidence to incorrect answers. The study is the first to reveal a gradient conflict between optimizing policy accuracy and minimizing calibration error. To resolve this, the authors propose the Decoupled Confidence and Policy Optimization (DCPO) framework, which disentangles confidence estimation from reasoning objectives and jointly optimizes both goals. Experiments demonstrate that DCPO achieves calibration performance significantly superior to existing methods while maintaining reasoning accuracy comparable to GRPO, thereby effectively mitigating overconfidence without compromising task performance.
📝 Abstract
Reinforcement Learning from Verifiable Rewards (RLVR) significantly enhances large language models (LLMs) reasoning but severely suffers from calibration degeneration, where models become excessively over-confident in incorrect answers. Previous studies devote to directly incorporating calibration objective into existing optimization target. However, our theoretical analysis demonstrates that there exists a fundamental gradient conflict between the optimization for maximizing policy accuracy and minimizing calibration error. Building on this insight, we propose DCPO, a simple yet effective framework that systematically decouples reasoning and calibration objectives. Extensive experiments demonstrate that our DCPO not only preserves accuracy on par with GRPO but also achieves the best calibration performance and substantially mitigates the over-confidence issue. Our study provides valuable insights and practical solution for more reliable LLM deployment.