mlsanityData to 7 Oct 2026

Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time Alignment

arXiv 2503.21878 ↗First posted 27 Mar 2025cs.AI cs.LG stat.MLcited by 91

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

Abstract (arXiv)

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 textttInferenceTimePessimism, 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 textttInferenceTimePessimism across a variety of tasks and models.

Key facts. Cited by 91 papers in the map (data through 7 Oct 2026). Placed in LLM preference optimization methods, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 64 of its 102 references that are in the map. First posted on arXiv 27 Mar 2025.

Builds on

64 of 102 references.

Closest ancestors · 19

Same direction, newest first

  1. 2024InfAlign: Inference-aware language model alignment27 Dec
  2. 2024TreeBoN: Enhancing Inference-Time Alignment with Speculative Tree-Search and Best-of-N Sampling18 Oct
  3. 2024GenARM: Reward Guided Generation with Autoregressive Reward Model for Test-time Alignment10 Oct
  4. 2024Correcting the Mythos of KL-Regularization: Direct Alignment without Overoptimization via Chi-Squared Preference Optimization18 Jul
  5. 2024Variational Best-of-N Alignment8 Jul
  6. 2024Decoding-Time Language Model Alignment with Multiple Objectives27 Jun
  7. 2024Cascade Reward Sampling for Efficient Decoding-Time Alignment24 Jun
  8. 2024Information Theoretic Guarantees For Policy Alignment In Large Language Models9 Jun
  9. 2024Self-Play with Adversarial Critic: Provable and Scalable Offline Alignment for Language Models6 Jun
  10. 2024The Importance of Online Data: Understanding Preference Fine-tuning via Coverage3 Jun
  11. 2024BoNBoN Alignment for Large Language Models and the Sweetness of Best-of-n Sampling2 Jun
  12. 2024Transfer Q Star: Principled Decoding for LLM Alignment30 May
7 more
  1. 2024Robust Preference Optimization through Reward Model Distillation29 May
  2. 2024Asymptotics of Language Model Alignment2 Apr
  3. 2024Regularized Best-of-N Sampling with Minimum Bayes Risk Objective for Language Model Alignment1 Apr
  4. 2024RS-DPO: A Hybrid Rejection Sampling and Direct Preference Optimization Method for Alignment of Large Language Models15 Feb
  5. 2024Decoding-time Realignment of Language Models5 Feb
  6. 2024ARGS: Alignment as Reward-Guided Search23 Jan
  7. 2023Statistical Rejection Sampling Improves Preference Optimization13 Sep

Other references · 45

  1. 2024The Llama 3 Herd of Modelscited by 14,192
  2. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  3. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  4. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  5. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  6. 2024Qwen2.5 Technical Reportcited by 4,713
  7. 2023Mistral 7Bcited by 3,065
  8. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,928
  9. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
  10. 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,883
  11. 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,722
  12. 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
33 more
  1. 2023RAFT: Reward rAnked FineTuning for Generative Foundation Model Alignmentcited by 672
  2. 2024Interpretable Preferences via Multi-Objective Reward Modeling and Mixture-of-Expertscited by 369
  3. 2024ReST-MCTS*: LLM Self-Training via Process Reward Guided Tree Searchcited by 338
  4. 2024Inference Scaling Laws: An Empirical Analysis of Compute-Optimal Inference for Problem-Solving with Language Modelscited by 282
  5. 2024Chain of Thought Empowers Transformers to Solve Inherently Serial Problemscited by 256
  6. 2024Meta-Rewarding Language Models: Self-Improving Alignment with LLM-as-a-Meta-Judgecited by 202
  7. 2023Helping or Herding? Reward Model Ensembles Mitigate but do not Eliminate Reward Hackingcited by 195
  8. 2024Regularizing Hidden States Enables Learning Generalizable Reward Model for LLMscited by 142
  9. 2024From Decoding to Meta-Generation: Inference-time Algorithms for Large Language Modelscited by 136
  10. 2024Theoretical guarantees on the best-of-n alignment policycited by 116
  11. 2024Provably Mitigating Overoptimization in RLHF: Your SFT Loss is Implicitly an Adversarial Regularizercited by 109
  12. 2024Exploratory Preference Optimization: Harnessing Implicit Q*-Approximation for Sample-Efficient RLHFcited by 106
  13. 2021Dueling RL: Reinforcement Learning with Trajectory Preferencescited by 105
  14. 2024ARC Prize 2024: Technical Reportcited by 103
  15. 2024Self-Improvement in Language Models: The Sharpening Mechanismcited by 102
  16. 2023Is RLHF More Difficult than Standard RL?cited by 96
  17. 2024Inference-Aware Fine-Tuning for Best-of-N Sampling in Large Language Modelscited by 86
  18. 2024Probabilistic Inference in Language Models via Twisted Sequential Monte Carlocited by 85
  19. 2024How to Evaluate Reward Models for RLHFcited by 84
  20. 2024Value-Incentivized Preference Optimization: A Unified Approach to Online and Offline RLHFcited by 73
  21. 2024BOND: Aligning LLMs with Best-of-N Distillationcited by 73
  22. 2023Reinforcement Learning with Human Feedback: Learning Dynamic Choices via Pessimismcited by 69
  23. 2024REBEL: Reinforcement Learning via Regressing Relative Rewardscited by 66
  24. 2024Active Preference Optimization for Sample Efficient RLHFcited by 61
  25. 2024PAD: Personalized Alignment of LLMs at Decoding-Timecited by 58
  26. 2024Online Iterative Reinforcement Learning from Human Feedback with General Preference Modelcited by 52
  27. 2024The Limits of Inference Scaling Through Resamplingcited by 51
  28. 2023Making RL with Preference-based Feedback Efficient via Randomizationcited by 46
  29. 2024Dataset Reset Policy Optimization for RLHFcited by 41
  30. 2024West-of-N: Synthetic Preferences for Self-Improving Reward Modelscited by 27
  31. 2023Provable Reward-Agnostic Preference-Based Reinforcement Learningcited by 25
  32. 2024Exploration-Driven Policy Optimization in RLHF: Theoretical Insights on Efficient Data Utilizationcited by 25
  33. 2024Sail into the Headwind: Alignment via Robust Rewards and Dynamic Labels against Reward Hackingcited by 13

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 33; 2026: 58.

91 papers.

Sort:

By area:

  1. 2025Chasing the Tail: Effective Rubric-based Reward Modeling for Large Language Model Post-Trainingcited by 46
  2. 2025Inference-Time Reward Hacking in Large Language Modelscited by 32
  3. 2025Energy-Based Transformers are Scalable Learners and Thinkerscited by 30
  4. 2025RewardAnything: Generalizable Principle-Following Reward Modelscited by 27
  5. 2026Scalable Power Sampling: Unlocking Efficient, Training-Free Reasoning for LLMs via Distribution Sharpeningcited by 27
  6. 2025Incentivizing Agentic Reasoning in LLM Judges via Tool-Integrated Reinforcement Learningcited by 16
  7. 2025Let Me Think! A Long Chain-of-Thought Can Be Worth Exponentially Many Short Onescited by 12
  8. 2025ROC-n-reroll: How verifier imperfection affects test-time scalingcited by 11
  9. 2025Guided Speculative Inference for Efficient Test-Time Alignment of LLMscited by 9
  10. 2025When Life Gives You Samples: The Benefits of Scaling up Inference Compute for Multilingual LLMscited by 9
  11. 2025PiCSAR: Probabilistic Confidence Selection And Ranking for Reasoning Chainscited by 9
  12. 2025Token Hidden Reward: Steering Exploration-Exploitation in Group Relative Deep Reinforcement Learningcited by 9
38 more
  1. 2025AdaBoN: Adaptive Best-of-N Alignmentcited by 8
  2. 2025Revisiting Test-Time Scaling: A Survey and a Diversity-Aware Method for Efficient Reasoningcited by 8
  3. 2025Best-of-Majority: Minimax-Optimal Strategy for Pass@k Inference Scalingcited by 8
  4. 2025Enhancing Large Language Model Reasoning with Reward Models: An Analytical Surveycited by 7
  5. 2025Learning to Answer from Correct Demonstrationscited by 6
  6. 2025On the Limits of Test-Time Compute: Sequential Reward Filtering for Better Inferencecited by 6
  7. 2026Likelihood-Based Reward Designs for General LLM Reasoningcited by 6
  8. 2025Learning a Pessimistic Reward Model in RLHFcited by 5
  9. 2025Building Coding Agents via Entropy-Enhanced Multi-Turn Preference Optimizationcited by 5
  10. 2025Information-Theoretic Reward Modeling for Stable RLHF: Detecting and Mitigating Reward Hackingcited by 5
  11. 2025A*-Decoding: Token-Efficient Inference Scalingcited by 4
  12. 2025Making, not Taking, the Best of Ncited by 4
  13. 2025Demystifying LLM-as-a-Judge: Analytically Tractable Model for Inference-Time Scalingcited by 4
  14. 2026Sampling for Quality: Training-Free Reward-Guided LLM Decoding via Sequential Monte Carlocited by 4
  15. 2026On the Generalization Gap in Self-Evolving Language Model Reasoningcited by 4
  16. 2025textttSPECS: Faster Test-Time Scaling through Speculative Draftscited by 3
  17. 2025Best-of-∞ -- Asymptotic Performance of Test-Time LLM Ensemblingcited by 3
  18. 2025Test-time Verification via Optimal Transport: Coverage, ROC, & Sub-optimalitycited by 3
  19. 2025ThoughtProbe: Classifier-Guided LLM Thought Space Exploration via Probing Representationscited by 3
  20. 2025In Good GRACEs: Principled Teacher Selection for Knowledge Distillationcited by 3
  21. 2026ATLAS: Agentic Test-time Learning-to-Allocate Scalingcited by 3
  22. 2025SDA: Steering-Driven Distribution Alignment for Open LLMs without Fine-Tuningcited by 2
  23. 2026Inference-time Unlearning Using Conformal Predictioncited by 2
  24. 2026Small Reward Models via Backward Inferencecited by 2
  25. 2026PETS: A Principled Framework Towards Optimal Trajectory Allocation for Efficient Test-Time Self-Consistencycited by 2
  26. 2026Best-of-Tails: Bridging Optimism and Pessimism in Inference-Time Alignmentcited by 2
  27. 2026Jailbreak Scaling Laws for Large Language Models: Polynomial-Exponential Crossovercited by 2
  28. 2026From Curiosity to Caution: Mitigating Reward Hacking for Best-of-N with Pessimismcited by 2
  29. 2026Agentic Systems as Boosting Weak Reasoning Modelscited by 2
  30. 2026Understanding Reasoning from Pretraining to Post-Trainingcited by 2
  31. 2026TailSFT: Filtered Fine-Tuning Improves Post-Training Performancecited by 2
  32. 2026Predicting and improving test-time scaling laws via reward tail-guided searchcited by 1
  33. 2026MATA: Multi-Agent Framework for Reliable and Flexible Table Question Answeringcited by 1
  34. 2026Temper-Then-Tilt: Principled Unlearning for Generative Models through Tempering and Classifier Guidancecited by 1
  35. 2026Power and Limitations of Aggregation in Compound AI Systemscited by 1
  36. 2026Provable and Practical In-Context Policy Optimization for Self-Improvementcited by 1
  37. 2026Adaptive Generate-Rank-Verify: Inference-Time Search with Costly Verificationcited by 1
  38. 2026What Am I Missing? Question-Answering as Hidden State Probingcited by 1

50 most cited of 91 All 91 (JSON)

Similar, not cited

Close in wording, no citation either way

  1. 2026Revisiting the (Sub)Optimality of Best-of-N for Inference-Time Alignment
  2. 2024Inference Time Alignment with Reward-Guided Tree Search
  3. 2026Efficient Best-of-N policy evaluation for inference-time alignment
  4. 2025Soft Best-of-n Sampling for Model Alignment
  5. 2025Inference-time Alignment in Continuous Space
  6. 2025On Almost Surely Safe Alignment of Large Language Models at Inference-Time
  7. 2026Compute Aligned Training: Optimizing for Test Time Inference
  8. 2024Fast Best-of-N Decoding via Speculative Rejection