解决大模型推理时过度自信问题,提升准确率与自知力。
C$^2$GSPG: Confidence-calibrated Group Sequence Policy Gradient towards Self-aware Reasoning
- 用序列级概率定义模型置信度,避免逐词偏差。
- 通过交叉熵正则化校准置信度,使预测更可信。
- 适合需要可靠推理的数学逻辑任务,如竞赛题求解。
强化学习方法,如组相对策略优化(GRPO)及其变体,在构建推理模型中起核心作用。然而,这些方法常存在严重过度自信问题,阻碍自知推理模型的发展。本文提出一种简单而有效的置信度校准组序列策略梯度方法(C²GSPG),在提升推理性能的同时抑制过度自信。我们提出组序列策略梯度(GSPG)框架,消除GRPO及其变体中常见的逐标记偏差。该框架通过归一化的序列级概率定义每个推理问题的模型置信度,并应用交叉熵正则化器将置信度校准至序列奖励。我们证明,在二元奖励下,置信度校准正则化器与GSPG目标始终同向梯度,具有协同性;对于非二元奖励,采用非线性奖励归一化和自适应正则化截断,缓解两者潜在冲突。将C²GSPG应用于后训练的大语言模型,在逻辑与数学推理任务中,其表现优于现有最先进方法,在推理准确率与置信度校准方面均具优势。代码已开源:https://github.com/HaotianLiu123/CCGSPG。
原文摘要 · Abstract (English)
Reinforcement Learning (RL) methods, exemplified by Group Relative Policy Optimization (GRPO) and its variants, play a central role in developing reasoning models. However, these methods often suffer from a critical overconfidence issue, which prevents them from achieving self-aware reasoning models. In this study, we propose a simple yet effective confidence-calibration group sequence policy gradient method, called C$^2$GSPG, which simultaneously enhances reasoning performance while suppressing overconfidence. In principle, we propose a Group Sequence Policy Gradient (GSPG) framework for learning reasoning models, which eliminates the token-level bias commonly appearing in GRPO and its variants. In this framework, we define the model confidence for each reasoning problem using the normalized sequence-level probability, and then apply a cross-entropy regularizer to calibrate the model confidence to the sequence's reward. We demonstrate that the confidence calibration regularizer and GSPG are collaborative for binary rewards, as their objectives always share the same gradient direction. For non-binary rewards, we apply nonlinear reward normalization and adaptive regularizer clipping, mitigating the potential conflict between the two objectives. Applying C$^2$GSPG to post-train large language models in logical and mathematical reasoning tasks, we show its superiority over state-of-the-art methods in both reasoning accuracy and confidence calibration. The code of C$^2$GSPG is available at https://github.com/HaotianLiu123/CCGSPG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。