arXiv:2503.18130cs.LGcs.AI2025-03ICLR被引 13

提出新方法抑制大模型奖励过优化,提升对齐人类意图的稳定性。

Mitigating Reward Over-Optimization in RLHF via Behavior-Supported Regularization

  • 用奖励数据分布定义行为策略,识别并约束分布外响应
  • 通过行为支持贝尔曼算子降低分布外值估计偏差,减少异常输出
  • 理论保证单调提升,实验验证在真实人类偏好上表现更优

基于人类反馈的强化学习(RLHF)是使大语言模型对齐人类价值观的有效方法。然而,奖励过优化仍是未解难题,导致模型在奖励模型下的表现与真实人类目标产生偏差。主要诱因是奖励模型对分布外(OOD)响应评估时的外推误差。现有方法无法有效抑制强化学习过程中分布外响应频率上升,也难以应对分布外评估带来的外推误差。本文提出行为支持策略优化(BSPO),将奖励训练数据的下一个词分布定义为行为策略,以建模奖励模型的分布内(ID)区域。基于此,引入行为支持贝尔曼算子,对所有分布外值进行惩罚,同时不影响分布内值。该机制有效减少了强化学习过程中的分布外响应生成,避免了由奖励模型外推误差引发的过估计问题。理论上,我们证明了BSPO能保证支持策略的单调改进,直至收敛至最优行为支持策略。大量实验证明,相比基线方法,BSPO能更有效地防止因分布外评估导致的奖励过优化,并找到最优分布内策略。

原文摘要 · Abstract (English)

Reinforcement learning from human feedback (RLHF) is an effective method for aligning large language models (LLMs) with human values. However, reward over-optimization remains an open challenge leading to discrepancies between the performance of LLMs under the reward model and the true human objectives. A primary contributor to reward over-optimization is the extrapolation error that arises when the reward model evaluates out-of-distribution (OOD) responses. However, current methods still fail to prevent the increasing frequency of OOD response generation during the reinforcement learning (RL) process and are not effective at handling extrapolation errors from OOD responses. In this work, we propose the Behavior-Supported Policy Optimization (BSPO) method to mitigate the reward over-optimization issue. Specifically, we define behavior policy as the next token distribution of the reward training dataset to model the in-distribution (ID) region of the reward model. Building on this, we introduce the behavior-supported Bellman operator to regularize the value function, penalizing all OOD values without impacting the ID ones. Consequently, BSPO reduces the generation of OOD responses during the RL process, thereby avoiding overestimation caused by the reward model's extrapolation errors. Theoretically, we prove that BSPO guarantees a monotonic improvement of the supported policy until convergence to the optimal behavior-supported policy. Empirical results from extensive experiments show that BSPO outperforms baselines in preventing reward over-optimization due to OOD evaluation and finding the optimal ID policy.

强化学习大模型对齐奖励过优化分布外检测

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