arXiv:2506.14574cs.LGcs.AI2025-06ICML被引 9

用词级奖励指导优化大模型对齐,性能显著提升。

TGDPO: Harnessing Token-Level Reward Guidance for Enhancing Direct Preference Optimization

  • 将序列级优化拆解为词级优化,引入词级奖励指导
  • 在MT-Bench等数据集上最高提升7.5个百分点胜率
  • 适合关注大模型对齐与强化学习融合的研究者

最近的基于人类反馈的强化学习进展表明,利用细粒度的词级奖励模型可显著提升近端策略优化(PPO)在对齐大语言模型方面的表现。然而,由于直接偏好优化(DPO)被建模为序列级老虎机问题,难以有效利用词级奖励进行指导。为此,本文将序列级PPO分解为一系列词级近端策略优化问题,并构建了带词级奖励引导的词级PPO框架,从中推导出闭式最优词级策略及对应词级奖励。结合得到的奖励与Bradley-Terry模型,本文建立了一套可计算的、含词级奖励引导的DPO损失函数,并提出一种基于诱导出的DPO奖励的实际奖励引导方法。该方法使不同词根据其奖励值表现出不同程度的参考策略偏离。实验结果表明,本方法在MT-Bench上胜率提升最高达7.5点,在AlpacaEval 2上提升6.2点,在Arena-Hard上提升4.3点。代码已公开于https://github.com/dvlab-research/TGDPO。

原文摘要 · Abstract (English)

Recent advancements in reinforcement learning from human feedback have shown that utilizing fine-grained token-level reward models can substantially enhance the performance of Proximal Policy Optimization (PPO) in aligning large language models. However, it is challenging to leverage such token-level reward as guidance for Direct Preference Optimization (DPO), since DPO is formulated as a sequence-level bandit problem. To address this challenge, this work decomposes the sequence-level PPO into a sequence of token-level proximal policy optimization problems and then frames the problem of token-level PPO with token-level reward guidance, from which closed-form optimal token-level policy and the corresponding token-level reward can be derived. Using the obtained reward and Bradley-Terry model, this work establishes a framework of computable loss functions with token-level reward guidance for DPO, and proposes a practical reward guidance based on the induced DPO reward. This formulation enables different tokens to exhibit varying degrees of deviation from reference policy based on their respective rewards. Experiment results demonstrate that our method achieves substantial performance improvements over DPO, with win rate gains of up to 7.5 points on MT-Bench, 6.2 points on AlpacaEval 2, and 4.3 points on Arena-Hard. Code is available at https://github.com/dvlab-research/TGDPO.

大模型对齐强化学习奖励引导

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