arXiv:2506.19248cs.LG2025-06NeurIPS被引 28

大模型推理时过度优化奖励模型会引发性能下降,本文提出新方法有效缓解此问题。

Inference-Time Reward Hacking in Large Language Models

  • 通过引入惩罚项控制奖励与模型熵的权衡,实现推理时更稳健的输出优化。
  • 在数学、推理和人类偏好任务中,新方法显著提升奖励得分并减少偏差。
  • 适合关注大模型对齐安全性和推理优化的研究者与工程实践者。

大型语言模型通常通过优化奖励模型来提升性能,但奖励模型仅是复杂目标(如正确性、有用性和安全性)的代理指标。过度优化存在偏差的奖励模型会导致对齐目标偏离,即所谓“奖励劫持”。本文研究推理阶段的奖励劫持现象,分析Best-of-$n$(BoN)与Soft Best-of-$n$(SBoN)机制下的表现,并提出近似最优奖励-熵权衡策略的Best-of-Poisson(BoP)。结果表明,实践中观察到的奖励先升后降模式是这类推理机制的固有特性。为此,我们提出HedgeTune算法,高效寻找最优推理参数。实验显示,通过适度惩罚奖励过拟合,可有效缓解奖励劫持,在数学、推理及人类偏好任务中实现更好的奖励与失真平衡。

原文摘要 · Abstract (English)

A common paradigm to improve the performance of large language models is optimizing for a reward model. Reward models assign a numerical score to an LLM's output that indicates, for example, how likely it is to align with user preferences or safety goals. However, reward models are never perfect. They inevitably function as proxies for complex desiderata such as correctness, helpfulness, and safety. By overoptimizing for a misspecified reward, we can subvert intended alignment goals and reduce overall performance, a phenomenon commonly referred to as reward hacking. In this work, we characterize reward hacking in inference-time alignment and demonstrate when and how we can mitigate it by hedging on the proxy reward. We study this phenomenon under Best-of-$n$ (BoN) and Soft Best-of-$n$ (SBoN), and we introduce Best-of-Poisson (BoP) that provides an efficient, near-exact approximation of the optimal reward-KL divergence policy at inference time. We show that the characteristic pattern of hacking as observed in practice (where the true reward first increases before declining) is an inevitable property of a broad class of inference-time mechanisms, including BoN and BoP. To counter this effect, we introduce HedgeTune, an efficient algorithm to find the optimal inference-time parameter. We demonstrate that hedging mitigates reward hacking and achieves superior reward-distortion tradeoffs on math, reasoning, and human-preference setups.

大模型对齐奖励劫持推理优化

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