arXiv:2508.05613cs.CLcs.AI2025-08被引 2

Cooper协同优化策略与奖励模型,解决大模型强化学习中的奖励欺骗问题。

Cooper: Co-Optimizing Policy and Reward Models in Reinforcement Learning for Large Language Models

  • 联合优化策略与奖励模型,动态构建正负样本对提升鲁棒性。
  • 在Qwen2.5-1.5B-Instruct上实现0.54%准确率提升,缓解奖励劫持。
  • 适用于需高可靠性推理的大模型强化学习场景。

大语言模型在推理任务中表现卓越,强化学习是提升其推理能力的关键方法。当前主流奖励范式分为基于模型的奖励与基于规则的奖励,但前者易受奖励劫持影响,后者鲁棒性不足。为此,本文提出Cooper框架,通过协同优化策略模型与奖励模型,利用规则奖励的高精度识别正确响应,并动态构建正负样本对持续训练奖励模型,增强鲁棒性并抑制奖励劫持。为支持该框架,引入混合标注策略以高效生成奖励模型训练数据,并提出参考答案输入的奖励建模范式,据此训练出VerifyRM模型,在VerifyBench上表现优于同规模其他模型。基于VerifyRM与Cooper进行强化学习实验表明,该方法不仅有效缓解奖励劫持,还提升了端到端强化学习性能,例如在Qwen2.5-1.5B-Instruct上实现0.54%平均准确率提升。结果表明,动态更新奖励模型是应对奖励劫持的有效路径,为奖励模型在强化学习中的融合提供了新思路。

原文摘要 · Abstract (English)

Large language models (LLMs) have demonstrated remarkable performance in reasoning tasks, where reinforcement learning (RL) serves as a key algorithm for enhancing their reasoning capabilities. Currently, there are two mainstream reward paradigms: model-based rewards and rule-based rewards. However, both approaches suffer from limitations: rule-based rewards lack robustness, while model-based rewards are vulnerable to reward hacking. To address these issues, we propose Cooper(Co-optimizing Policy Model and Reward Model), a RL framework that jointly optimizes both the policy model and the reward model. Cooper leverages the high precision of rule-based rewards when identifying correct responses, and dynamically constructs and selects positive-negative sample pairs for continued training the reward model. This design enhances robustness and mitigates the risk of reward hacking. To further support Cooper, we introduce a hybrid annotation strategy that efficiently and accurately generates training data for the reward model. We also propose a reference-based reward modeling paradigm, where the reward model takes a reference answer as input. Based on this design, we train a reward model named VerifyRM, which achieves higher accuracy on VerifyBench compared to other models of the same size. We conduct reinforcement learning using both VerifyRM and Cooper. Our experiments show that Cooper not only alleviates reward hacking but also improves end-to-end RL performance, for instance, achieving a 0.54% gain in average accuracy on Qwen2.5-1.5B-Instruct. Our findings demonstrate that dynamically updating reward model is an effective way to combat reward hacking, providing a reference for better integrating reward models into RL.

强化学习大模型奖励模型推理优化

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