arXiv:2606.30339cs.CLcs.LG2026-06中稿 · ICML

让大模型在测试时自动调整偏好,无需重新训练。

REAR: Test-time Preference Realignment through Reward Decomposition

论文配图:REAR: Test-time Preference Realignment through Reward Decomposition
图 1 · 摘自论文原文
  • 将奖励函数拆分为问题和偏好两部分,动态调节权重。
  • 在多个任务上实现优于基线的偏好对齐效果。
  • 适合需要快速适配用户偏好的场景,如对话系统优化。

将大语言模型与多样化的用户偏好对齐是一项关键但具挑战性的任务。尽管后训练方法可适应特定需求,但通常需要昂贵的数据整理和额外训练。测试时扩展(TTS)提供了一种无需训练的高效替代方案,但其应用主要局限于数学和编程等可验证领域。为将TTS扩展至偏好对齐,我们提出一种新框架,将该任务视为重对齐问题,因为基础模型往往无法充分匹配用户声明的偏好。核心思想是将底层奖励函数分解为与问题相关和与偏好信息相关的两部分,从而推导出一个可选择性重缩放这两项奖励比例的重构奖励(REAR)。我们证明,REAR可表示为词元级策略对数概率的线性组合,计算高效且易于集成到多种TTS算法中,如best-of-$N$采样和树搜索。实验表明,相比其他测试时基线,REAR不仅实现了在多样化用户需求下可扩展的测试时对齐,还适用于数学和视觉任务,在适当偏好设置下表现出良好泛化能力。

原文摘要 · Abstract (English)

Aligning large language models (LLMs) with diverse user preferences is a critical yet challenging task. While post-training methods can adapt models to specific needs, they often require costly data curation and additional training. Test-time scaling (TTS) presents an efficient, training-free alternative, but its application has been largely limited to verifiable domains like mathematics and coding, where response correctness is easily judged. To extend TTS to preference alignment, we introduce a novel framework that models the task as a realignment problem, since the base model often fails to sufficiently align with the stated preference. Our key insight is to decompose the underlying reward function into two components: one related to the question and the other to preference information. This allows us to derive a REAlignment Reward (REAR) that selectively rescales the proportions of these two reward terms. We then show that REAR can be formulated as a linear combination of token-level policy log-probabilities, making it computationally efficient and easy to integrate with various TTS algorithms such as best-of-$N$ sampling and tree search. Experiments show that compared to other test-time baselines, REAR not only enables scalable test-time realignment for preference alignment tasks under diverse user requirements, but also generalizes to mathematical and visual tasks under appropriate preference settings.

大模型对齐测试时扩展偏好学习

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