mlsanityData to 7 Oct 2026

Theoretical Limits of Language Model Alignment

arXiv 2605.07105 ↗First posted 8 May 2026cs.LG cs.CL cs.CYcited by 0

On the map: LLM alignment / RLHF / preference optimisation → LLM preference optimization methods

Abstract (arXiv)

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 √textttKL 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.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in LLM preference optimization methods, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 35 of its 48 references that are in the map. First posted on arXiv 8 May 2026.

Builds on

35 of 48 references.

Closest ancestors · 5

Same direction, newest first

  1. 2025Soft Best-of-n Sampling for Model Alignment6 May
  2. 2024Information Theoretic Guarantees For Policy Alignment In Large Language Models9 Jun
  3. 2024BoNBoN Alignment for Large Language Models and the Sweetness of Best-of-n Sampling2 Jun
  4. 2024Asymptotics of Language Model Alignment2 Apr
  5. 2022Training language models to follow instructions with human feedback4 Mar

Other references · 30

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  4. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  5. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  6. 2017Deep reinforcement learning from human preferencescited by 3,846
  7. 2020Learning to summarize from human feedbackcited by 2,562
  8. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  9. 2016Concrete Problems in AI Safetycited by 1,582
  10. 2022WebShop: Towards Scalable Real-World Web Interaction with Grounded Language Agentscited by 1,197
  11. 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
  12. 2023BeaverTails: Towards Improved Safety Alignment of LLM via a Human-Preference Datasetcited by 828
18 more
  1. 2023Enhancing Chat Language Models by Scaling High-quality Instructional Conversationscited by 825
  2. 2020Pre-trained Models for Natural Language Processing: A Surveycited by 504
  3. 2023AI Alignment: A Comprehensive Surveycited by 256
  4. 2024PKU-SafeRLHF: Towards Multi-Level Safety Alignment for LLMs with Human Preferencecited by 252
  5. 2023Reward Model Ensembles Help Mitigate Overoptimizationcited by 225
  6. 2023Helping or Herding? Reward Model Ensembles Mitigate but do not Eliminate Reward Hackingcited by 195
  7. 2023AI Deception: A Survey of Examples, Risks, and Potential Solutionscited by 185
  8. 2023Controlled Decoding from Language Modelscited by 133
  9. 2022Calibrating Sequence likelihood Improves Conditional Language Generationcited by 121
  10. 2024Theoretical guarantees on the best-of-n alignment policycited by 116
  11. 2024Large Language Models Reflect the Ideology of their Creatorscited by 56
  12. 2024The N+ Implementation Details of RLHF with PPO: A Case Study on TL;DR Summarizationcited by 49
  13. 2025Values in the Wild: Discovering and Analyzing Values in Real-World Language Model Interactionscited by 47
  14. 2024Improving Reinforcement Learning from Human Feedback with Efficient Reward Model Ensemblecited by 43
  15. 2024Shopping MMLU: A Massive Multi-Task Online Shopping Benchmark for Large Language Modelscited by 20
  16. 2025AI Alignment at Your Discretioncited by 20
  17. 2024Scalable Ensembling For Mitigating Reward Overoptimisationcited by 14
  18. 2025DSO: Direct Steering Optimization for Bias Mitigationcited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024Rethinking the Role of Proxy Rewards in Language Model Alignment
  2. 2023Language Model Alignment with Elastic Reset
  3. 2025Towards Reliable, Uncertainty-Aware Alignment
  4. 2024Elephant in the Room: Unveiling the Impact of Reward Model Quality in Alignment
  5. 2025Energy-Based Reward Models for Robust Language Model Alignment
  6. 2025Alignment of large language models with constrained learning
  7. 2024Correcting the Mythos of KL-Regularization: Direct Alignment without Overoptimization via Chi-Squared Preference Optimization
  8. 2025A Principled Loss Function for Direct Language Model Alignment