arXiv:2606.22716cs.AIcs.CL2026-06

提出自适应仅正确奖励机制,稳定大模型高效推理训练。

Beyond Penalizing Mistakes: Stabilizing Efficiency Training in Large Reasoning Models via Adaptive Correct-Only Rewards

论文配图:Beyond Penalizing Mistakes: Stabilizing Efficiency Training in Large Reasoning Models via Adaptive Correct-Only Rewards
图 1 · 摘自论文原文
  • 仅对正确回答给予简洁性奖励,避免惩罚错误回答导致的奖励崩溃。
  • 在数学推理任务中减少60%以上生成token,同时提升整体准确率。
  • 适合追求高效且稳定推理能力的模型训练场景。

训练大语言模型实现高效推理是一项关键挑战。尽管在组相对策略优化(GRPO)中引入长度惩罚奖励旨在减少冗余,但常引发奖励崩溃,严重损害推理能力。通过系统评估多种奖励配置,我们发现根本原因在于:GRPO的组归一化在错误答案持续受长度惩罚时产生分歧优势。因此,对错误答案施加长度惩罚的方法在持续优化下具有结构性崩溃风险。进一步地,仅对正确答案施加惩罚虽可避免此问题,但仍可能因响应过度压缩而引发随机性崩溃。为稳健防止两种失效模式,本文提出ACOER(自适应仅正确效率奖励)。ACOER通过将简洁性奖励仅限于正确完成项,消除结构化惩罚循环;并通过动态预算归一化与控制环路惩罚调整,防止随机压缩。在多个数学推理基准上评估显示,相比基线模型,ACOER在减少超过60%生成token的同时提升了整体准确率,建立了面向效率感知优化的根本稳定方法。

原文摘要 · Abstract (English)

Training large language models to reason efficiently is a critical challenge. While integrating length-penalizing rewards into Group Relative Policy Optimization (GRPO) aims to reduce verbosity, it frequently triggers reward collapse, severely degrading reasoning capabilities. Through a systematic evaluation of various reward configurations, we identify the root mechanism: GRPO's group normalization creates divergent advantages when incorrect answers receive continuous length penalties. Consequently, methods penalizing the length of incorrect answers are structurally prone to collapse under sustained optimization. Furthermore, restricting penalties exclusively to correct answers avoids this primary failure, but leaves the model susceptible to a stochastic collapse driven by response over-compression. To robustly prevent both failure modes, we propose ACOER (Adaptive Correct-Only Efficiency Reward). ACOER eliminates the structural penalty loop by isolating brevity bonuses to correct completions and prevents stochastic compression via dynamic budget normalization and control-loop penalty adjustments. Evaluated across diverse mathematical reasoning benchmarks, ACOER improves overall accuracy compared to the base model while reducing token generation by over 60%, establishing a fundamentally stable approach for efficiency-aware optimization.

高效推理奖励设计模型训练

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