arXiv:2501.19358cs.LG2025-01ICML被引 17

发现强化学习中能量损耗加剧会引发奖励滥用,提出新算法抑制此现象。

The Energy Loss Phenomenon in RLHF: A New Perspective on Mitigating Reward Hacking

  • 通过分析语言模型最后一层能量损耗,揭示其与奖励滥用的关联。
  • 实验显示能量损耗过快时,模型上下文相关性下降,易产生奖励滥用。
  • 提出EPPO算法,通过惩罚能量损耗提升鲁棒性,适合改进大模型对齐训练。

本文识别出强化学习从人类反馈(RLHF)中的能量损耗现象及其与奖励滥用的关系。具体而言,在强化学习过程中,大型语言模型(LLM)最后一层的能量损耗逐渐增加,且能量损耗过度增长是奖励滥用的特征。除了实证分析,我们还提供了理论基础,证明在温和条件下,能量损耗的增加会降低语言模型上下文相关性的上界,而这是奖励滥用的关键表现——上下文相关性下降通常意味着模型过度拟合奖励模型偏好的模式。为应对该问题,我们提出了能量损耗感知的PPO算法(EPPO),在奖励计算中惩罚模型最后一层的能量损耗增长,以防止其过度增加,从而缓解奖励滥用。理论上,我们证明了EPPO可被概念化为一种熵正则化强化学习算法,提供了对其有效性的深层理解。在多种大语言模型和任务上的广泛实验表明,能量损耗现象具有普遍性,且EPPO在缓解奖励滥用和提升RLHF性能方面效果显著。

原文摘要 · Abstract (English)

This work identifies the Energy Loss Phenomenon in Reinforcement Learning from Human Feedback (RLHF) and its connection to reward hacking. Specifically, energy loss in the final layer of a Large Language Model (LLM) gradually increases during the RL process, with an excessive increase in energy loss characterizing reward hacking. Beyond empirical analysis, we further provide a theoretical foundation by proving that, under mild conditions, the increased energy loss reduces the upper bound of contextual relevance in LLMs, which is a critical aspect of reward hacking as the reduced contextual relevance typically indicates overfitting to reward model-favored patterns in RL. To address this issue, we propose an Energy loss-aware PPO algorithm (EPPO) which penalizes the increase in energy loss in the LLM's final layer during reward calculation to prevent excessive energy loss, thereby mitigating reward hacking. We theoretically show that EPPO can be conceptually interpreted as an entropy-regularized RL algorithm, which provides deeper insights into its effectiveness. Extensive experiments across various LLMs and tasks demonstrate the commonality of the energy loss phenomenon, as well as the effectiveness of EPPO in mitigating reward hacking and improving RLHF performance.

RLHF奖励滥用大模型对齐强化学习

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