arXiv:2503.15477cs.LGcs.AI2025-03NeurIPS被引 70

高准确率未必好,奖励模型还需足够波动才能高效训练。

What Makes a Reward Model a Good Teacher? An Optimization Perspective

  • 从优化角度发现:奖励方差过低会导致强化学习目标平坦化
  • 即使准确率100%,低方差也会使训练速度极慢
  • 适合关注奖励模型设计与训练效率的AI研究者

强化学习中的人类反馈(RLHF)成功高度依赖奖励模型的质量。然而,当前主要以准确性评估奖励模型,却未明确准确性是否足以反映其作为“教师”的有效性。本文从优化视角分析发现:无论奖励模型多准确,若其诱导的奖励方差过低,将导致RLHF目标函数出现平坦区域,从而引发优化困难。即便准确率为100%,低方差仍可能使训练速度极慢,表现不如一些准确性较低但方差更高的模型。此外,一个对某语言模型有效的奖励模型,可能对另一模型造成低方差问题。实验基于最大80亿参数模型验证了奖励方差、准确性与最大化速率之间的复杂关系。研究揭示:除准确性外,奖励模型必须能诱发足够奖励方差,才能实现高效优化。

原文摘要 · Abstract (English)

The success of Reinforcement Learning from Human Feedback (RLHF) critically depends on the quality of the reward model. However, while this quality is primarily evaluated through accuracy, it remains unclear whether accuracy fully captures what makes a reward model an effective teacher. We address this question from an optimization perspective. First, we prove that regardless of how accurate a reward model is, if it induces low reward variance, then the RLHF objective suffers from a flat landscape. Consequently, even a perfectly accurate reward model can lead to extremely slow optimization, underperforming less accurate models that induce higher reward variance. We additionally show that a reward model that works well for one language model can induce low reward variance, and thus a flat objective landscape, for another. These results establish a fundamental limitation of evaluating reward models solely based on accuracy or independently of the language model they guide. Experiments using models of up to 8B parameters corroborate our theory, demonstrating the interplay between reward variance, accuracy, and reward maximization rate. Overall, our findings highlight that beyond accuracy, a reward model needs to induce sufficient variance for efficient optimization.

强化学习奖励模型优化效率

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