通过动态调整奖励策略,解决大模型推理中过长思考导致的训练不稳问题。
Stable Reinforcement Learning for Efficient Reasoning
- 根据每组生成结果的正确率动态切换长度惩罚或0/1奖励
- 在GSM8K等5个基准上提升平均准确率1.48%,推理长度减少47.3%
- 适合追求高效高准确推理的大模型应用者
Deepseek-R1的成功引发学界对GRPO等强化学习方法的关注。然而,基于0/1结果的规则奖励无法调控思维链(CoT)生成过程中的中间推理,导致严重过思考现象。尽管已有研究引入长度惩罚奖励以鼓励更短且正确的完成,但发现这类奖励加剧了强化学习训练的不稳定性:随着完成长度下降,模型准确率突然崩溃,常出现在训练早期。为此,我们提出GRPO-λ,一种高效且稳定的GRPO变体,通过监测每查询采样组内完成结果的正确率动态调整奖励策略。正确率低时,避免长度惩罚以保护推理质量,切换为无长度惩罚的0/1奖励;正确率高时则维持长度惩罚以提升效率。实验表明,该方法有效避免了长度惩罚引发的训练不稳,同时保持最优准确率-效率权衡。在GSM8K、GPQA、MATH-500、AMC 2023和AIME 2024五个基准上,平均准确率提升1.48%,思维链序列长度减少47.3%。
原文摘要 · Abstract (English)
The success of Deepseek-R1 has drawn the LLM community's attention to reinforcement learning (RL) methods like GRPO. However, such rule-based 0/1 outcome reward methods lack the capability to regulate the intermediate reasoning processes during chain-of-thought (CoT) generation, leading to severe overthinking phenomena. In response, recent studies have designed reward functions to reinforce models' behaviors in producing shorter yet correct completions. Nevertheless, we observe that these length-penalty reward functions exacerbate RL training instability: as the completion length decreases, model accuracy abruptly collapses, often occurring early in training. To address this issue, we propose a simple yet effective solution GRPO-$λ$, an efficient and stabilized variant of GRPO, which dynamically adjusts the reward strategy by monitoring the correctness ratio among completions within each query-sampled group. A low correctness ratio indicates the need to avoid length penalty that compromises CoT quality, triggering a switch to length-agnostic 0/1 rewards that prioritize reasoning capability. A high ratio maintains length penalties to boost efficiency. Experimental results show that our approach avoids training instability caused by length penalty while maintaining the optimal accuracy-efficiency trade-off. On the GSM8K, GPQA, MATH-500, AMC 2023, and AIME 2024 benchmarks, it improves average accuracy by 1.48% while reducing CoT sequence length by 47.3%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。