用医生病人模型实现高效精准的LLM测试时对齐
LLMdoctor: Token-Level Flow-Guided Preference Optimization for Efficient Test-Time Alignment of Large Language Models
- 通过逐标记级信号引导小模型优化大模型生成
- 在多个数据集上超越全量微调效果,且保持生成多样性
- 适合需要快速适配且不改主模型的场景
大型语言模型(LLM)与人类偏好对齐至关重要,但传统微调方法计算成本高且灵活性差。测试时对齐虽具潜力,但现有方法常依赖扭曲的轨迹级信号或低效采样,严重限制性能并破坏生成多样性。本文提出LLMdoctor,一种基于患者-医生范式的高效测试时对齐框架。它将标记级奖励获取与标记级流引导偏好优化(TFPO)结合,利用小而专精的医生模型,指导冻结的大患者模型生成。不同于依赖轨迹级奖励的传统方法,LLMdoctor首先从患者模型的行为差异中提取细粒度的标记级偏好信号,再通过TFPO确保所有子轨迹间的流一致性,实现逐标记精准对齐,同时自然保留生成多样性。大量实验表明,该方法显著优于现有测试时对齐方法,甚至超越全量微调技术如DPO。
原文摘要 · Abstract (English)
Aligning Large Language Models (LLMs) with human preferences is critical, yet traditional fine-tuning methods are computationally expensive and inflexible. While test-time alignment offers a promising alternative, existing approaches often rely on distorted trajectory-level signals or inefficient sampling, fundamentally capping performance and failing to preserve the generative diversity of the base model. This paper introduces LLMdoctor, a novel framework for efficient test-time alignment that operates via a patient-doctor paradigm. It integrates token-level reward acquisition with token-level flow-guided preference optimization (TFPO) to steer a large, frozen patient LLM with a smaller, specialized doctor model. Unlike conventional methods that rely on trajectory-level rewards, LLMdoctor first extracts fine-grained, token-level preference signals from the patient model's behavioral variations. These signals then guide the training of the doctor model via TFPO, which establishes flow consistency across all subtrajectories, enabling precise token-by-token alignment while inherently preserving generation diversity. Extensive experiments demonstrate that LLMdoctor significantly outperforms existing test-time alignment methods and even surpasses the performance of full fine-tuning approaches like DPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。