arXiv:2503.02832cs.CLcs.AI2025-03ACL被引 14

让大模型生成更精准,通过逐词优化奖励提升对齐效果

AlignDistil: Token-Level Language Model Alignment as Adaptive Policy Distillation

  • 用逐词奖励替代整体反馈,让每部分输出都精准优化
  • 实验显示收敛速度更快,性能优于现有方法
  • 适合追求高效高质生成的模型开发者和研究者

在现代大语言模型中,对齐至关重要,通常通过人类反馈强化学习(RLHF)和直接偏好优化(DPO)实现。然而,现有方法多使用稀疏的响应级奖励或偏好标注,忽略了逐词奖励,可能错误惩罚高质量词汇或鼓励低质量词汇,导致性能不佳且收敛慢。为此,我们提出AlignDistil,一种等效于RLHF的逐词奖励优化蒸馏方法。具体而言,我们将DPO学习到的奖励引入RLHF目标,并理论证明该目标与逐词蒸馏过程等价,其中教师分布线性结合DPO模型和参考模型的logits。为进一步缩小DPO模型奖励与纯奖励模型之间的差距,我们构建了包含正向与反向DPO模型的对比奖励机制。同时,为避免不同词语的过优或欠优优化,设计了逐词自适应logit外推机制,为每个词语构建合适的教师分布。实验表明,AlignDistil在多项指标上优于现有方法,得益于其逐词分布奖励优化,展现出快速收敛特性。

原文摘要 · Abstract (English)

In modern large language models (LLMs), LLM alignment is of crucial importance and is typically achieved through methods such as reinforcement learning from human feedback (RLHF) and direct preference optimization (DPO). However, in most existing methods for LLM alignment, all tokens in the response are optimized using a sparse, response-level reward or preference annotation. The ignorance of token-level rewards may erroneously punish high-quality tokens or encourage low-quality tokens, resulting in suboptimal performance and slow convergence speed. To address this issue, we propose AlignDistil, an RLHF-equivalent distillation method for token-level reward optimization. Specifically, we introduce the reward learned by DPO into the RLHF objective and theoretically prove the equivalence between this objective and a token-level distillation process, where the teacher distribution linearly combines the logits from the DPO model and a reference model. On this basis, we further bridge the accuracy gap between the reward from the DPO model and the pure reward model, by building a contrastive DPO reward with a normal and a reverse DPO model. Moreover, to avoid under- and over-optimization on different tokens, we design a token adaptive logit extrapolation mechanism to construct an appropriate teacher distribution for each token. Experimental results demonstrate the superiority of our AlignDistil over existing methods and showcase fast convergence due to its token-level distributional reward optimization.

大模型对齐逐词优化奖励蒸馏DPO

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