arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2510.18924cs.LGcs.AI

噪声校正的GRPO:从噪声奖励到无偏梯度

Noise-corrected GRPO: From Noisy Rewards to Unbiased Gradients

Department of Machine Learning, Mohamed bin Zayed University of Artificial Intelligence, Abu Dhabi, UAE;Technology Innovation Institute, Abu Dhabi, UAE;Department of Robotics, Khalifa University, Abu Dhabi, UAE

Omar El Mansouri, Fathinah Asma Izzati, Mohamed El Amine Seddik, Salem Lahlou

AI总结:

本文提出了一种噪声鲁棒的GRPO框架,通过校正奖励中的噪声来获得无偏梯度估计,从而提升强化学习在噪声环境中的性能。

摘要:

人类反馈的强化学习(RLHF)或可验证奖励(RLVR)是对大语言模型进行对齐或构建最新SOTA推理模型的标准范式,但其对不一致或错误奖励产生的噪声非常敏感。然而,此类噪声与广泛使用的基于组的策略优化方法之间的相互作用仍不为人知。我们引入了一种噪声鲁棒的组相对策略优化(GRPO)和正确执行GRPO(Dr.GRPO)框架,该框架明确将奖励损坏建模为伯努利噪声。我们的方法在估计奖励翻转概率后应用噪声校正,以消除学习信号的偏差,从而获得可证明无偏的梯度估计。理论分析表明,基于组的方法本质上可以缓解个体层面的噪声,而我们的校正策略增强了这种鲁棒性。实验表明,在应用我们的噪声校正到标准奖励模型使用时,数学和代码任务中均观察到一致的改进,特别是在现实奖励模型条件下,数学任务的准确性提高了高达6.7个百分点,代码任务提高了1.5个百分点。这项工作将监督学习中的标签噪声校正与现代RLHF相结合,提供了理论洞察和实用算法,以应对噪声现实世界部署。

英文摘要:

Reinforcement learning from human feedback (RLHF) or verifiable rewards (RLVR), the standard paradigm for aligning LLMs or building recent SOTA reasoning models, is highly sensitive to noise from inconsistent or erroneous rewards. Yet, the interaction between such noise and widely used group-based policy optimization methods remains underexplored. We introduce a noise-robust Group Relative Policy Optimization (GRPO) and Done Right GRPO (Dr.GRPO) framework that explicitly models reward corruption as Bernoulli noise. Our method applies noise correction after estimating reward flip probabilities to debias the learning signal, yielding provably unbiased gradient estimates. Theoretical analysis shows that group-based methods inherently mitigate individual-level noise, and our correction strategy amplifies this robustness. Empirically, we observe consistent improvements across math and code tasks when applying our noise correction to standard reward model usage, with particular gains of up to 6.7 percentage points in accuracy on math tasks and 1.5 on code tasks under realistic reward model conditions. This work bridges label-noise correction from supervised learning with modern RLHF, offering both theoretical insights and a practical algorithm for noisy real-world deployment.

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

↑