arXiv:2603.06797cs.AIcs.LG2026-03被引 1

通过动态调整推理时对齐策略,平衡探索与精准性。

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

  • 基于奖励分布尾部特性,自适应选择乐观或悲观策略
  • 在数学和多选推理任务中优于固定策略基线
  • 适合需要高可靠响应的场景,如安全关键应用

推理时对齐通过从参考模型生成多个候选并用不完美奖励模型进行选择来引导大语言模型。然而,现有方法面临根本性困境:‘乐观’策略(如Best-of-N)易受奖励欺骗,而‘悲观’正则化方法常抑制发现优质回复所需的探索。本文从后悔最小化视角形式化这一权衡,证明最优策略取决于奖励分布的尾部行为。理论表明,在轻尾情形下应采用乐观策略以发掘高质量异常值,而在重尾情形下需依赖悲观策略防范极端处的奖励校准偏差。受此启发,提出自适应推理时对齐框架Best-of-Tails(BoT),使用Tsallis散度作为可调正则项,实现两种策略间的精细插值。BoT利用Hill估计器按提示粒度刻画奖励尾部密度,并动态调整选择规则以平衡探索收益与对齐误差。在数学、多选推理及人类偏好评估中,相比固定策略基线,BoT在多种参考模型与奖励模型配置下均提升了对齐性能。

原文摘要 · Abstract (English)

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.

推理对齐策略优化奖励建模

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