mlsanityData to 7 Oct 2026

Best-of-Tails: Bridging Optimism and Pessimism in Inference-Time Alignment

arXiv 2603.06797 ↗First posted 6 Mar 2026cs.AI cs.LGcited by 2

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

Abstract (arXiv)

Inference-time alignment effectively steers large language models (LLMs) by generating multiple candidates from a reference model and selecting among them with an imperfect reward model. However, current strategies face a fundamental dilemma: ``optimistic'' approaches like Best-of-N suffer from reward hacking, while ``pessimistic'' regularized methods often stifle the exploration needed to discover high-quality responses. In this work, we formalize this trade-off through the lens of regret minimization, demonstrating that the optimal strategy depends critically on the tail behavior of the reward distribution. We show theoretically that light-tailed regimes favor optimism to unearth high-quality outliers, whereas heavy-tailed regimes require pessimism to guard against reward mis-calibration in the extremes. Guided by this insight, we introduce Best-of-Tails (BoT), an adaptive inference-time alignment framework that uses Tsallis divergence as a tunable regularizer to provide a finer granularity of interpolation between these extremes. BoT uses the Hill estimator to characterize reward-tail heaviness on a per-prompt basis and dynamically adjusts its selection rule to balance exploration gains against alignment error. Across math, multiple-choice reasoning, and human-preference evaluations, BoT improves alignment performance across a range of reference and reward model configurations relative to fixed-strategy baselines.

Key facts. Cited by 2 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 55 of its 64 references that are in the map. First posted on arXiv 6 Mar 2026.

Builds on

55 of 64 references.

Closest ancestors · 17

Same direction, newest first

  1. 2025Optimal Stopping vs Best-of-N for Inference Time Optimization1 Oct
  2. 2025Best-of-N through the Smoothing Lens: KL Divergence and Regret Analysis8 Jul
  3. 2025Inference-Time Reward Hacking in Large Language Models24 Jun
  4. 2025Guided Speculative Inference for Efficient Test-Time Alignment of LLMs4 Jun
  5. 2025AdaBoN: Adaptive Best-of-N Alignment17 May
  6. 2025Soft Best-of-n Sampling for Model Alignment6 May
  7. 2025Sample, Don't Search: Rethinking Test-Time Alignment for Language Models4 Apr
  8. 2025Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time Alignment27 Mar
  9. 2024InfAlign: Inference-aware language model alignment27 Dec
  10. 2024Correcting the Mythos of KL-Regularization: Direct Alignment without Overoptimization via Chi-Squared Preference Optimization18 Jul
  11. 2024Variational Best-of-N Alignment8 Jul
  12. 2024BoNBoN Alignment for Large Language Models and the Sweetness of Best-of-n Sampling2 Jun
5 more
  1. 2024Asymptotics of Language Model Alignment2 Apr
  2. 2024Regularized Best-of-N Sampling with Minimum Bayes Risk Objective for Language Model Alignment1 Apr
  3. 2023The Unlocking Spell on Base LLMs: Rethinking Alignment via In-Context Learning4 Dec
  4. 2023Statistical Rejection Sampling Improves Preference Optimization13 Sep
  5. 2023OpenAssistant Conversations -- Democratizing Large Language Model Alignment14 Apr

Other references · 38

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2024The Llama 3 Herd of Modelscited by 14,192
  3. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  4. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  5. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  6. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  7. 2020Scaling Laws for Neural Language Modelscited by 6,411
  8. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  9. 2017Deep reinforcement learning from human preferencescited by 3,846
  10. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,155
  11. 2023Mistral 7Bcited by 3,065
  12. 2020Learning to summarize from human feedbackcited by 2,562
26 more
  1. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,928
  2. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
  3. 2023Self-RAG: Learning to Retrieve, Generate, and Critique through Self-Reflectioncited by 1,800
  4. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  5. 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
  6. 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
  7. 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
  8. 2023AlpacaFarm: A Simulation Framework for Methods that Learn from Human Feedbackcited by 738
  9. 2023RAFT: Reward rAnked FineTuning for Generative Foundation Model Alignmentcited by 672
  10. 2024Interpretable Preferences via Multi-Objective Reward Modeling and Mixture-of-Expertscited by 369
  11. 2024ReST-MCTS*: LLM Self-Training via Process Reward Guided Tree Searchcited by 338
  12. 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 293
  13. 2023Reward Model Ensembles Help Mitigate Overoptimizationcited by 225
  14. 2024Meta-Rewarding Language Models: Self-Improving Alignment with LLM-as-a-Meta-Judgecited by 202
  15. 2024Regularizing Hidden States Enables Learning Generalizable Reward Model for LLMscited by 142
  16. 2022RL with KL penalties is better viewed as Bayesian inferencecited by 130
  17. 2024Theoretical guarantees on the best-of-n alignment policycited by 116
  18. 2025Reward-Guided Speculative Decoding for Efficient LLM Reasoningcited by 106
  19. 2024Self-Improvement in Language Models: The Sharpening Mechanismcited by 102
  20. 2023Self-Evaluation Improves Selective Generation in Large Language Modelscited by 74
  21. 2024BOND: Aligning LLMs with Best-of-N Distillationcited by 73
  22. 2024Correlated Proxies: A New Definition and Improved Mitigation for Reward Hackingcited by 50
  23. 2024Rethinking Bradley-Terry Models in Preference-Based Reward Modeling: Foundations, Theory, and Alternativescited by 48
  24. 2024Catastrophic Goodhart: regularizing RLHF with KL divergence does not mitigate heavy-tailed reward misspecificationcited by 20
  25. 2019Tsallis Reinforcement Learning: A Unified Framework for Maximum Entropy Reinforcement Learningcited by 18
  26. 2023The Sample Complexity of Approximate Rejection Sampling with Applications to Smoothed Online Learningcited by 13

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Revisiting the (Sub)Optimality of Best-of-N for Inference-Time Alignment
  2. 2026From Curiosity to Caution: Mitigating Reward Hacking for Best-of-N with Pessimism
  3. 2026Gradient-Guided Reward Optimization for Inference-time Alignment
  4. 2025Reward-Shifted Speculative Sampling Is An Efficient Test-Time Weak-to-Strong Aligner
  5. 2024Inference Time Alignment with Reward-Guided Tree Search
  6. 2026Safe Inference-Time Alignment via Lagrangian Reward Augmentation
  7. 2026Predicting and improving test-time scaling laws via reward tail-guided search
  8. 2026Best-of-Better-N: Generating Pre-Aligned Responses with In-Context Learning