arXiv:2605.07105cs.LGcs.CL2026-05

揭示语言模型对齐的理论极限,解释为何当前方法仍有提升空间。

Theoretical Limits of Language Model Alignment

论文配图:Theoretical Limits of Language Model Alignment
图 1 · 摘自论文原文
  • 用杰弗里斯散度推导出对齐最优奖励增益公式
  • 发现best-of-N几乎逼近理论极限,而PPO等方法仍明显落后
  • 解释奖励欺骗现象并证明集成可缓解该问题,适合研究对齐算法者

语言模型对齐旨在使模型输出符合人类偏好,同时保持基础模型能力。主流方法包括强化学习(在KL约束下最大化期望奖励)和best-of-$N$(从$N$个独立采样中选最高奖励输出)。尽管广泛应用,其在固定KL预算下的理论极限仍不明确。本文通过信息论分析,推导出给定KL预算下可实现的最大期望奖励增益,其闭式解由杰弗里斯散度决定,而非先前分析中的$\ exttt{KL}$平方根。进一步将表达式重写为基模型下的协方差形式,提出仅需基模型样本即可估计对齐潜力的实用估算器。扩展至代理奖励场景,表明理想与代理对齐的差距随奖励误差增大而扩大,且当KL惩罚系数减小时更严重。证明奖励集成可缓解奖励欺骗,为实际应用提供理论支持。实验计算了安全与摘要任务下两个模型的KL-奖励帕累托前沿,显示best-of-$N$接近理论极限,而PPO和GRPO仍显著落后。结果解释了对齐领域多个经验现象,表明需算法改进以在低推理成本下实现最优对齐。

原文摘要 · Abstract (English)

Language model (LM) alignment improves model outputs to reflect human preferences while preserving the capabilities of the base model. The most common alignment approaches are (i) reinforcement learning, which maximizes the expected reward under a KL-divergence constraint, and (ii) best-of-$N$ alignment, which selects the highest-reward output among $N$ independent samples. Despite their widespread use, the fundamental limits of reward improvement under a KL budget remain poorly understood. We characterize the information-theoretic limits of KL-regularized alignment by deriving the maximum achievable expected reward gain for a fixed KL-divergence budget. Our first result provides a closed-form expression for the optimal reward improvement, governed by a Jeffreys divergence term rather than the $\sqrt{\texttt{KL}}$ used in prior analyses. We further reformulate this expression as a covariance under the base model, yielding a practical estimator that predicts achievable alignment gains from base model samples alone. We extend our analysis to the proxy reward setting, showing that the gap between ideal and proxy alignment (reward hacking) grows with the magnitude of reward error and when the KL penalty factor decreases. We then prove that reward ensembling mitigates reward hacking, providing a theoretical justification for this technique used in practice. Empirically, we compute the KL-reward Pareto frontier for two tasks for LMs, safety and summarization, and show that best-of-$N$ closely approaches the theoretical limit, while PPO and GRPO remain substantially suboptimal. Our theoretical results shed light on several empirically observed phenomena in the alignment literature and suggest that algorithmic improvements are needed to achieve optimal alignment without high inference costs.

语言模型对齐理论强化学习奖励欺骗

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