arXiv:2604.04648cs.LG2026-04被引 2

用悲观策略抑制大模型生成中的奖励劫持问题。

From Curiosity to Caution: Mitigating Reward Hacking for Best-of-N with Pessimism

  • 引入悲观机制,通过误差模型降低异常响应的奖励分。
  • 在BoN采样中提升N值时仍保持性能,避免奖励劫持导致下降。
  • 适合追求高可靠性推理的场景,尤其适用于对齐与评估任务。

推理阶段的计算扩展已成为提升语言模型性能的有效范式,但如何有效利用额外算力仍存疑问。主流方法Best-of-N(BoN)通过生成N个候选回复,依据奖励模型评分并选择最高分者,虽可提升性能,却易受奖励劫持影响:随着N增大,模型倾向于选择利用奖励模型缺陷的回应,而非真正高质量生成。已有缓解方法如更强奖励模型或强正则化,或无法彻底解决过优化,或过于保守而浪费算力。本文探索强化学习中的悲观原则,提出‘谨慎’(caution)策略——基于典型响应训练误差模型,以预测误差降低异常响应的奖励估计。实验表明,该方法简单高效,显著缓解了BoN中的奖励劫持。理论分析在简化线性设定下证明其优于标准BoN。结果不仅确立了谨慎作为实用解决方案,还暗示基于好奇心的方法可普遍用于大模型中的分布外检测。

原文摘要 · Abstract (English)

Inference-time compute scaling has emerged as a powerful paradigm for improving language model performance on a wide range of tasks, but the question of how best to use the additional compute remains open. A popular approach is BoN sampling, where N candidate responses are generated, scored according to a reward model, and the highest-scoring response is selected. While this approach can improve performance, it is vulnerable to reward hacking, where performance degrades as N increases due to the selection of responses that exploit imperfections in the reward model instead of genuinely improving generation quality. Prior attempts to mitigate reward hacking, via stronger reward models or heavy-handed distributional regularization, either fail to fully address over-optimization or are too conservative to exploit additional compute. In this work, we explore the principle of pessimism in RL, which uses lower confidence bounds on value estimates to avoid OOD actions with uncertain reward estimates. Our approach, termed as caution, can be seen as the reverse of curiosity: where curiosity rewards prediction error as a signal of novelty, caution penalizes prediction error as a signal of distributional uncertainty. Practically, caution trains an error model on typical responses and uses its prediction error to lower reward estimates for atypical ones. Our extensive empirical evaluation demonstrates that caution is a simple, computationally efficient approach that substantially mitigates reward hacking in BoN sampling. We also provide a theoretical analysis in a simplified linear setting, which shows that caution provably improves over the standard BoN approach. Together, our results not only establish caution as a practical solution to reward hacking, but also provide evidence that curiosity-based approaches can be a general OOD detection technique in LLM settings.

奖励劫持BoN采样悲观策略大模型对齐

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