让大模型从错误中学习,通过分步判断提升推理训练效果
Stepwise Guided Policy Optimization: Coloring your Incorrect Reasoning in GRPO
- 用分步裁判模型增强组内回复多样性,解决全错样本无法更新的问题
- 在9个推理任务上验证,7B到32B模型均提升早期和中期训练性能
- 无需裁判生成正确答案,适用于各类推理模型微调
强化学习已证明能有效提升大语言模型的推理能力。广泛采用的组相对策略优化(GRPO)在训练近期推理模型时表现良好,但在所有组内响应均错误(即全负样本组)时无法更新策略,这暴露了与人类智能的差距:人类能从错误中学习,而GRPO会丢弃这些失败信号。本文提出一种简单框架,通过引入组内回复多样性并结合分步裁判模型(可直接训练或基于现有LLM适配),缓解全负样本问题。在简化设定下,我们证明该多样化策略能加速GRPO的学习动态。随后,在九个推理基准(包括基础版和蒸馏版)上,对7B、14B、32B三种规模模型进行离线与在线训练验证,结果表明SGPO整体提升平均性能,尤其在早期和中期训练阶段(此时全负样本占比较高)效果显著;但改进并非在每个基准上均匀出现,取决于负样本的结构与信息量。最后,SGPO无需裁判模型生成正确解,区别于知识蒸馏方法。
原文摘要 · Abstract (English)
Reinforcement learning (RL) has proven effective in strengthening the reasoning capabilities of large language models (LLMs). A widely adopted method, Group Relative Policy Optimization (GRPO), has shown strong empirical results in training recent reasoning models, but it fails to update the policy when all responses within a group are incorrect (i.e., all-negative-sample groups). This limitation highlights a gap between artificial and human intelligence: unlike humans, who can learn from mistakes, GRPO discards these failure signals. We introduce a simple framework to mitigate the all-negative-sample issue by incorporating response diversity within groups using a step-wise judge model, which can be trained directly or adapted from existing LLMs. In a simplified setting, we prove that this diversification accelerates GRPO's learning dynamics. We then empirically validate Stepwise Guided Policy Optimization (SGPO) across model sizes (7B, 14B, 32B) in both offline and online training on nine reasoning benchmarks (including base and distilled variants). Overall, SGPO improves average performance and is effective in early and mid-training when all-negative groups are prevalent, while improvements are not uniform across every benchmark and depend on the structure and informativeness of negative samples. Finally, SGPO does not require the judge model to generate correct solutions, distinguishing it from knowledge distillation methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。