arXiv:2507.20067cs.AIcs.CL2025-07被引 4

PITA让大模型生成更符合用户偏好的内容,无需额外训练或奖励模型。

PITA: Preference-Guided Inference-Time Alignment for LLM Post-Training

  • 用小模型在推理时直接调整生成概率,不依赖预训练奖励模型。
  • 通过迭代优化偏好引导策略,在数学推理和情感分类上显著提升对齐效果。
  • 适合需要快速适配用户偏好的场景,如对话系统与个性化生成。

推理时对齐技术使大语言模型在不进行再训练的情况下生成符合用户偏好的输出。现有后训练方法通常使用小型引导模型在推理过程中修改令牌生成,其优化目标为以原始LLM作为参考策略的KL正则化奖励函数。然而,该方法严重依赖预训练的奖励模型,而奖励模型需拟合人类偏好反馈——这一过程可能不稳定。相比之下,我们提出PITA,一种将偏好反馈直接融入LLM令牌生成的新框架,无需奖励模型。PITA学习一个小型基于偏好的引导策略,在推理时修改令牌概率,避免了对LLM的微调,降低计算开销并摆脱奖励模型依赖。该问题被建模为识别潜在偏好分布,通过随机搜索与引导模型的迭代精炼求解。我们在多种任务(包括数学推理和情感分类)上评估PITA,证明其能有效对齐用户偏好。

原文摘要 · Abstract (English)

Inference-time alignment enables large language models (LLMs) to generate outputs aligned with end-user preferences without further training. Recent post-training methods achieve this by using small guidance models to modify token generation during inference. These methods typically optimize a reward function KL-regularized by the original LLM taken as the reference policy. A critical limitation, however, is their dependence on a pre-trained reward model, which requires fitting to human preference feedback--a potentially unstable process. In contrast, we introduce PITA, a novel framework that integrates preference feedback directly into the LLM's token generation, eliminating the need for a reward model. PITA learns a small preference-based guidance policy to modify token probabilities at inference time without LLM fine-tuning, reducing computational cost and bypassing the pre-trained reward model dependency. The problem is framed as identifying an underlying preference distribution, solved through stochastic search and iterative refinement of the preference-based guidance model. We evaluate PITA across diverse tasks, including mathematical reasoning and sentiment classification, demonstrating its effectiveness in aligning LLM outputs with user preferences.

推理对齐偏好学习大模型

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