通过过滤不可靠奖励样本,提升强化学习中策略优化的稳定性与效果。
Policy Filtration for RLHF to Mitigate Noise in Reward Models
- 基于奖励模型得分可靠性动态筛选训练样本,降低噪声干扰。
- 在代码生成任务中,70亿参数模型在HumanEval上提升7.9%准确率。
- 适用于需要复杂推理的奖励评估场景,尤其适合追求高精度的RLHF研究者。
尽管存在直接策略优化方法,但前沿大语言模型仍常通过人类反馈强化学习(RLHF)进行微调,以在奖励模型指导下生成更优响应。其主要挑战在于中间奖励模型的不准确性,尤其是在需要复杂推理的任务中,奖励模型对响应评分的能力有限。我们发现,不同奖励值对应的响应,其奖励模型的可靠性存在差异。为此,我们提出针对近端策略优化(PF-PPO)的策略过滤机制(Policy Filtration),通过剔除可能不可靠的样本,提升策略学习中的信噪比。为选择合适的过滤策略,我们使用过滤样本上奖励与实际评分间的决定系数(R²)作为指标,衡量奖励信号与真实性能的一致性。大量实验验证了该方法在代码生成和数学推理任务上的有效性:在代码生成中,PF-PPO使70亿参数模型在HumanEval(+7.9%)、MBPP(+0.7%)及我们自建的LeetCode Contest(+10.0%)上达到当前最优表现;在数学推理中,也对Ape210K与CMATH多个基准实现了性能提升。代码已开源于 https://github.com/DtYXs/verl/tree/pf-ppo。
原文摘要 · Abstract (English)
While direct policy optimization methods exist, pioneering LLMs are fine-tuned with reinforcement learning from human feedback (RLHF) to generate better responses under the supervision of a reward model learned from preference data. One major challenge of RLHF is the inaccuracy of the intermediate reward model, especially in the tasks that requires complex reasoning for the reward model to score a response. We find that the reliability of the reward model varies across responses assigned with different rewards. This motivates us to filter the samples whose rewards may be unreliable to improve the signal-to-noise ratio during policy learning, resulting in Policy Filtration for Proximal Policy Optimization (PF-PPO). To choose a proper policy filtering strategy, we use the coefficient of determination (R2) between the rewards and actual scores on filtered samples as the metrics to help us find promising strategies since it measures how well the rewards filtered by PF-PPO indicate real performance. We provide extensive experiments to validate the effectiveness of PF-PPO in code generation and math reasoning tasks. In code generation, PF-PPO achieves the state-of-the-art performance of 7-billion-parameter models on HumanEval (+7.9%), MBPP (+0.7%), and LeetCode Contest (+10.0%) which is a more challenging benchmark created by us. In math reasoning, PF-PPO yields performance increase using different reward models and benchmarks (Ape210K and CMATH). Code is available on https://github.com/DtYXs/verl/tree/pf-ppo.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。