arXiv:2510.13501cs.AI2025-10被引 4

用模型自信度做奖励,零训练提升大模型推理能力

Confidence as a Reward: Transforming LLMs into Reward Models

  • 用模型输出的逐令牌自信度作为奖励信号,无需额外训练
  • 在MATH500和RewardMATH上超越多数训练型奖励模型
  • 可筛选高质量数据,适配需要精准判断的闭合任务

奖励模型能显著提升大语言模型的推理能力,但通常需大量标注数据和昂贵训练。为降低门槛,无训练方法如LLM-as-a-Judge利用模型自身推理能力评估回答,表现良好。近期研究发现,模型自信度可有效作为奖励指标,区分思维链(CoT)与非CoT路径。本文系统研究了零训练的‘自信度即奖励’(CRew)方法,通过模型最终答案的逐令牌自信度作为奖励代理,尤其适用于闭合任务。在数学推理任务上的实验表明,CRew在MATH500和RewardMATH基准上优于现有无训练方法,甚至超越多数训练型奖励模型。我们还发现CRew得分与模型实际推理性能强相关,并能有效过滤高质量训练数据。基于此,提出CRew-DPO训练策略,结合自信度与正确性信号构建偏好数据,微调后模型判别能力持续优于现有自训练方法。

原文摘要 · Abstract (English)

Reward models can significantly enhance the reasoning capabilities of large language models (LLMs), but they typically require extensive curated data and costly training. To mitigate these challenges, training-free approaches such as LLM-as-a-Judge leverage the intrinsic reasoning abilities of LLMs to evaluate responses, achieving promising results. Recent works have also indicated that model confidence can serve effectively as a reward metric, distinguishing between chain-of-thought (CoT) and non-CoT paths. However, the concept of using confidence as a reward has not been comprehensively studied. In this work, we systematically investigate Confidence-as-a-Reward (CRew), a simple yet powerful training-free method that utilizes token-level confidence in the model's final answers as a proxy for reward, especially suitable for close-ended tasks. Through extensive experiments on mathematical reasoning tasks, we demonstrate that CRew outperforms existing training-free reward approaches on the MATH500 and RewardMATH benchmarks, and even surpasses most trained reward models. We further identify a strong correlation between CRew scores and the actual reasoning performance of the model. Additionally, we find that CRew can effectively filter high-quality training data. Building upon these insights, we propose CRew-DPO, a training strategy that constructs preference data from confidence scores combined with correctness signals. Finetuning with CRew-DPO further enhances the model's judging capabilities and consistently outperforms existing self-training methods.

奖励模型自信度推理增强零训练

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