arXiv:2604.12632cs.LGcs.AI2026-04ACL被引 2

提升大模型推理的可信度,让错答更不自信,对答更准。

Calibration-Aware Policy Optimization for Reasoning LLMs

  • 引入校准感知的优势估计,让模型学习时关注不确定性。
  • 在数学推理任务上校准度提升15%,准确率不降反升。
  • 支持低置信度时拒绝回答,有效减少幻觉,适合实际部署。

组相对策略优化(GRPO)虽提升了大模型推理能力,但常导致过度自信:错误回答的困惑度反而低于正确回答,使相对校准性能下降(以曲线下面积AUC衡量)。现有方法或校准改进有限,或牺牲推理准确率。我们证明,此类算法的校准退化源于其忽略不确定性的优势估计,导致优化梯度与校准目标错位,以准确率为代价换取校准恶化。为此提出校准感知策略优化(CAPO),采用理论一致且具有后悔界保障的逻辑回归型AUC代理损失,实现不确定性感知的优势估计。进一步引入噪声掩蔽机制,确保学习过程稳定,同时优化校准与准确率。多数学推理基准测试表明,CAPO-1.5B在校准度上最高提升15%,准确率不低于甚至超过GRPO;在下游推理时缩放任务中,准确率再提升5%。当允许在低置信度下拒绝回答时,CAPO达成精度-覆盖率的帕累托最优,凸显其在抑制幻觉方面的实用价值。

原文摘要 · Abstract (English)

Group Relative Policy Optimization (GRPO) enhances LLM reasoning but often induces overconfidence, where incorrect responses yield lower perplexity than correct ones, degrading relative calibration as described by the Area Under the Curve (AUC). Existing approaches either yield limited improvements in calibration or sacrifice gains in reasoning accuracy. We first prove that this degradation in GRPO-style algorithms stems from their uncertainty-agnostic advantage estimation, which inevitably misaligns optimization gradients with calibration. This leads to improved accuracy at the expense of degraded calibration. We then propose Calibration-Aware Policy Optimization (CAPO). It adopts a logistic AUC surrogate loss that is theoretically consistent and admits regret bound, enabling uncertainty-aware advantage estimation. By further incorporating a noise masking mechanism, CAPO achieves stable learning dynamics that jointly optimize calibration and accuracy. Experiments on multiple mathematical reasoning benchmarks show that CAPO-1.5B significantly improves calibration by up to 15% while achieving accuracy comparable to or better than GRPO, and further boosts accuracy on downstream inference-time scaling tasks by up to 5%. Moreover, when allowed to abstain under low-confidence conditions, CAPO achieves a Pareto-optimal precision-coverage trade-off, highlighting its practical value for hallucination mitigation.

大模型推理校准优化幻觉抑制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。