arXiv:2511.07483cs.AIcs.LG2025-11EMNLP被引 2

用置信度增强奖励模型,让小模型也能练出靠谱推理链

Beyond Correctness: Confidence-Aware Reward Modeling for Enhancing Large Language Model Reasoning

  • 基于回答置信度设计奖励机制,不只看对错
  • 在多个科学类基准上超越现有开源模型表现
  • 适合资源有限但想提升小模型推理能力的研究者

大型语言模型的后训练范式正从传统的指令微调和人类偏好对齐转向以推理能力为核心的强化学习。然而,大量技术报告指出,纯规则型奖励强化学习常导致推理链质量差或推理过程与最终答案不一致,尤其在小规模模型上更为明显。在强化学习探索过程中,模型因知识不足可能采用低质量推理链,偶尔随机得出正确答案并被规则判官奖励,限制了资源有限组织对小模型直接进行强化学习训练的可能性。本文提出一种面向增强STEM推理能力的置信度感知奖励模型。不同于传统方法,该模型不仅惩罚错误答案,也惩罚高正确率但低置信度的回答,从而促进更稳健、逻辑一致的推理过程。通过静态评估、Best-of-N推理测试及PPO强化学习训练验证了方法的有效性。实验表明,本方法在多个科学类基准上优于若干最先进的开源奖励模型。代码与模型已公开于https://github.com/qianxiHe147/C2RM。

原文摘要 · Abstract (English)

Recent advancements in large language models (LLMs) have shifted the post-training paradigm from traditional instruction tuning and human preference alignment toward reinforcement learning (RL) focused on reasoning capabilities. However, numerous technical reports indicate that purely rule-based reward RL frequently results in poor-quality reasoning chains or inconsistencies between reasoning processes and final answers, particularly when the base model is of smaller scale. During the RL exploration process, models might employ low-quality reasoning chains due to the lack of knowledge, occasionally producing correct answers randomly and receiving rewards based on established rule-based judges. This constrains the potential for resource-limited organizations to conduct direct reinforcement learning training on smaller-scale models. We propose a novel confidence-based reward model tailored for enhancing STEM reasoning capabilities. Unlike conventional approaches, our model penalizes not only incorrect answers but also low-confidence correct responses, thereby promoting more robust and logically consistent reasoning. We validate the effectiveness of our approach through static evaluations, Best-of-N inference tests, and PPO-based RL training. Our method outperforms several state-of-the-art open-source reward models across diverse STEM benchmarks. We release our codes and model in https://github.com/qianxiHe147/C2RM.

强化学习推理增强置信度小模型

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