提出新方法提升推理时对齐效果,避免算力增加反导致性能下降。
Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time Alignment
- 用悲观原则设计拒绝采样机制,主动规避奖励欺骗
- 在真实覆盖率下表现优于传统Best-of-N,且算力越大越稳定
- 适合追求可靠推理结果的高阶应用开发者
推理时计算为提升语言模型性能提供了强大扩展维度。然而,像Best-of-N采样这类方法若盲目增加计算量,可能因奖励欺骗导致性能下降。本文聚焦推理时对齐问题:在给定提示和一个不完美奖励模型的前提下,如何通过额外计算提升预训练策略输出的质量。我们从响应质量与算力消耗两个角度分析算法表现,揭示预训练策略对高质量响应的覆盖范围对性能与算力扩展至关重要。首先,当覆盖条件严格时,理想N值的Best-of-N可达到最优性能,但当N过大时会因奖励欺骗而表现退化,且在更现实的覆盖条件下无法获得紧致保证。其次,我们提出新算法InferenceTimePessimism,通过推理时计算主动引入悲观性,利用拒绝采样缓解奖励欺骗;理论上证明其性能最优且不随N增大而退化,即具备算力单调性。实验验证了该方法在多种任务与模型上的有效性。
原文摘要 · Abstract (English)
Inference-time computation offers a powerful axis for scaling the performance of language models. However, naively increasing computation in techniques like Best-of-N sampling can lead to performance degradation due to reward hacking. Toward a theoretical understanding of how to best leverage additional computation, we focus on inference-time alignment, which we formalize as the problem of improving the quality of responses drawn from a pre-trained policy, given a prompt of interest and access to an imperfect reward model. We analyze the performance of inference-time alignment algorithms in terms of (i) response quality, and (ii) compute, and provide new results that highlight the importance of the pre-trained policy's coverage over high-quality responses for performance and compute scaling: 1. We show that Best-of-$N$ alignment with an ideal choice for $N$ can achieve optimal performance under stringent notions of coverage, but provably suffers from reward hacking when $N$ is large, and fails to achieve tight guarantees under more realistic coverage conditions. 2. We introduce $\texttt{InferenceTimePessimism}$, a new algorithm which mitigates reward hacking through deliberate use of inference-time compute, implementing the principle of pessimism in the face of uncertainty via rejection sampling; we prove that its performance is optimal and does not degrade with $N$, meaning it is scaling-monotonic. We complement our theoretical results with an experimental evaluation that demonstrate the benefits of $\texttt{InferenceTimePessimism}$ across a variety of tasks and models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。