arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2511.04439cs.AIcs.LG
收藏

CoRPO:在GRPO中添加正确性偏差以提高泛化能力

CoRPO: Adding a Correctness Bias to GRPO Improves Generalization

Anisha Garg, Claire Zhang, Nishit Neema, David Bick, Ganesh Venkatesh, Joel Hestness

AI总结:

CoRPO通过引入正确性偏差改进GRPO,提升模型在跨领域任务中的泛化能力与推理稳定性。

摘要:

组相对策略优化(GRPO)已成为通过强化学习训练大语言模型推理能力的标准方法。通过使用组均值奖励而非学习的批评者来估计优势,GRPO使从可验证奖励(RLVR)中高效扩展强化学习成为可能。然而,我们发现了一个根本性限制:GRPO的均值基线可以简单地将正优势分配给错误的解决方案,因为它们优于表现差的组平均。这导致了优势的高估和错误行为的强化。为了解决这个问题,我们提出了正确性相对策略优化(CoRPO),这是一种对GRPO目标的简单修改,它将最小基线裁剪到固定的正确性阈值。我们显示,基线裁剪引入了保护偏差到优势估计中,这在减轻过拟合的同时保持了有效的探索。经验上,训练CoRPO的模型在跨领域推理中表现更好,更一致地泛化到非领域(OOD)任务。当在编码任务上训练时,CoRPO在数学任务上优于GRPO,反之亦然,表明CoRPO学习了稳健、可转移的推理模式,而不是任务特定的解决方案。

英文摘要:

Group-Relative Policy Optimization (GRPO) has emerged as the standard for training reasoning capabilities in large language models through reinforcement learning. By estimating advantages using group-mean rewards rather than a learned critic, GRPO has enabled efficient scaling of reinforcement learning from verifiable rewards (RLVR). However, we identify a fundamental limitation: GRPO's mean baseline can assign positive advantages to incorrect solutions simply because they outperform a poorly-performing group average. It leads to overestimation of advantages and reinforcement of incorrect behaviours. To address this, we propose Correctness-Relative Policy Optimization (CoRPO), a simple modification to the GRPO objective that clips the minimum baseline to a fixed correctness threshold. We show that baseline clipping introduces a protective bias to advantage estimation that mitigates overfitting while preserving effective exploration. Empirically, CoRPO-trained models improve cross-domain reasoning, generalizing more consistently to out-of-domain (OOD) tasks. When trained on coding tasks, CoRPO outperforms GRPO on math, and vice-versa, indicating that CoRPO learns robust, transferable reasoning patterns rather than task-specific solutions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑