Best-of-Tails: Bridging Optimism and Pessimism in Inference-Time Alignment
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
- 2025Optimal Stopping vs Best-of-N for Inference Time Optimization1 Oct
- 2025Best-of-N through the Smoothing Lens: KL Divergence and Regret Analysis8 Jul
- 2025Inference-Time Reward Hacking in Large Language Models24 Jun
- 2025Guided Speculative Inference for Efficient Test-Time Alignment of LLMs4 Jun
- 2025AdaBoN: Adaptive Best-of-N Alignment17 May
- 2025Soft Best-of-n Sampling for Model Alignment6 May
- 2025Sample, Don't Search: Rethinking Test-Time Alignment for Language Models4 Apr
- 2025Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time Alignment27 Mar
- 2024InfAlign: Inference-aware language model alignment27 Dec
- 2024Correcting the Mythos of KL-Regularization: Direct Alignment without Overoptimization via Chi-Squared Preference Optimization18 Jul
- 2024Variational Best-of-N Alignment8 Jul
- 2024BoNBoN Alignment for Large Language Models and the Sweetness of Best-of-n Sampling2 Jun
5 more
- 2024Asymptotics of Language Model Alignment2 Apr
- 2024Regularized Best-of-N Sampling with Minimum Bayes Risk Objective for Language Model Alignment1 Apr
- 2023The Unlocking Spell on Base LLMs: Rethinking Alignment via In-Context Learning4 Dec
- 2023Statistical Rejection Sampling Improves Preference Optimization13 Sep
- 2023OpenAssistant Conversations -- Democratizing Large Language Model Alignment14 Apr
Other references · 38
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,155
- 2023Mistral 7Bcited by 3,065
- 2020Learning to summarize from human feedbackcited by 2,562
26 more
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,928
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
- 2023Self-RAG: Learning to Retrieve, Generate, and Critique through Self-Reflectioncited by 1,800
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
- 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
- 2023AlpacaFarm: A Simulation Framework for Methods that Learn from Human Feedbackcited by 738
- 2023RAFT: Reward rAnked FineTuning for Generative Foundation Model Alignmentcited by 672
- 2024Interpretable Preferences via Multi-Objective Reward Modeling and Mixture-of-Expertscited by 369
- 2024ReST-MCTS*: LLM Self-Training via Process Reward Guided Tree Searchcited by 338
- 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 293
- 2023Reward Model Ensembles Help Mitigate Overoptimizationcited by 225
- 2024Meta-Rewarding Language Models: Self-Improving Alignment with LLM-as-a-Meta-Judgecited by 202
- 2024Regularizing Hidden States Enables Learning Generalizable Reward Model for LLMscited by 142
- 2022RL with KL penalties is better viewed as Bayesian inferencecited by 130
- 2024Theoretical guarantees on the best-of-n alignment policycited by 116
- 2025Reward-Guided Speculative Decoding for Efficient LLM Reasoningcited by 106
- 2024Self-Improvement in Language Models: The Sharpening Mechanismcited by 102
- 2023Self-Evaluation Improves Selective Generation in Large Language Modelscited by 74
- 2024BOND: Aligning LLMs with Best-of-N Distillationcited by 73
- 2024Correlated Proxies: A New Definition and Improved Mitigation for Reward Hackingcited by 50
- 2024Rethinking Bradley-Terry Models in Preference-Based Reward Modeling: Foundations, Theory, and Alternativescited by 48
- 2024Catastrophic Goodhart: regularizing RLHF with KL divergence does not mitigate heavy-tailed reward misspecificationcited by 20
- 2019Tsallis Reinforcement Learning: A Unified Framework for Maximum Entropy Reinforcement Learningcited by 18
- 2023The Sample Complexity of Approximate Rejection Sampling with Applications to Smoothed Online Learningcited by 13
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Revisiting the (Sub)Optimality of Best-of-N for Inference-Time Alignment
- 2026From Curiosity to Caution: Mitigating Reward Hacking for Best-of-N with Pessimism
- 2026Gradient-Guided Reward Optimization for Inference-time Alignment
- 2025Reward-Shifted Speculative Sampling Is An Efficient Test-Time Weak-to-Strong Aligner
- 2024Inference Time Alignment with Reward-Guided Tree Search
- 2026Safe Inference-Time Alignment via Lagrangian Reward Augmentation
- 2026Predicting and improving test-time scaling laws via reward tail-guided search
- 2026Best-of-Better-N: Generating Pre-Aligned Responses with In-Context Learning