用新奖励机制让大模型推理更简洁高效,少废话多得分。
Efficient Reasoning via Reward Model
- 训练专门评分推理简洁度的奖励模型,引导模型精炼表达。
- 在数学题上提升8.1%准确率,响应长度减少19.9%。
- 适合追求高效推理的模型部署与工业级应用者。
基于可验证奖励的强化学习(RLVR)已被证明能提升大语言模型(LLMs)的推理能力,推动大型推理模型(LRMs)的发展。然而,DeepSeek-R1 和 OpenAI o1 等 LRMs 常产生冗长且包含无关步骤的推理过程,即‘过度思考’,显著增加计算成本。现有方法多通过在奖励函数中加入长度惩罚缓解此问题,但常引发长度坍缩和训练坍缩,导致性能不佳。为此,我们提出一种训练简洁性奖励模型(CRM)的流程,并设计新的简洁性奖励函数(CRF),显式建模结果奖励与简洁性评分之间的依赖关系,从而促进更高效、更优的推理。理论分析表明,新奖励函数具有更低方差与更好收敛性。实证方面,在五个数学基准数据集上的实验显示,该方法在 Qwen2.5-7B 上实现 8.1% 的准确率提升与 19.9% 的响应词元数减少,且对 Llama、Mistral 等其他大模型具有良好泛化能力。代码与数据集已公开:https://anonymous.4open.science/r/CRM。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards (RLVR) has been shown to enhance the reasoning capabilities of large language models (LLMs), enabling the development of large reasoning models (LRMs). However, LRMs such as DeepSeek-R1 and OpenAI o1 often generate verbose responses containing redundant or irrelevant reasoning step-a phenomenon known as overthinking-which substantially increases computational costs. Prior efforts to mitigate this issue commonly incorporate length penalties into the reward function, but we find they frequently suffer from two critical issues: length collapse and training collapse, resulting in sub-optimal performance. To address them, we propose a pipeline for training a Conciseness Reward Model (CRM) that scores the conciseness of reasoning path. Additionally, we introduce a novel reward formulation named Conciseness Reward Function (CRF) with explicit dependency between the outcome reward and conciseness score, thereby fostering both more effective and more efficient reasoning. From a theoretical standpoint, we demonstrate the superiority of the new reward from the perspective of variance reduction and improved convergence properties. Besides, on the practical side, extensive experiments on five mathematical benchmark datasets demonstrate the method's effectiveness and token efficiency, which achieves an 8.1% accuracy improvement and a 19.9% reduction in response token length on Qwen2.5-7B. Furthermore, the method generalizes well to other LLMs including Llama and Mistral. The implementation code and datasets are publicly available for reproduction: https://anonymous.4open.science/r/CRM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。