arXiv:2608.23664cs.CVcs.LG2026-08

用速度匹配方法简化扩散模型强化学习,降低训练成本且效果更优。

Scaling Reinforcement Learning for Diffusion Models via Velocity Matching

论文配图:Scaling Reinforcement Learning for Diffusion Models via Velocity Matching
图 1 · 摘自论文原文
  • 直接在速度场上做奖励驱动更新,无需复杂似然计算。
  • 训练成本大幅降低,在多个任务上性能超越或媲美现有方法。
  • 适合追求高效微调的扩散模型研究者与应用开发者。

奖励微调正成为适应扩散模型人类偏好与特定任务目标的重要工具,但现有方法大多沿用大语言模型的策略梯度框架。不同于自回归模型,扩散模型无法对生成样本提供可计算的似然。因此,当前方法要么从随机去噪过程构造轨迹似然,要么用证据下界近似终点似然,引入额外计算与算法复杂性。本文证明,有效扩散模型奖励微调并不需要基于似然的机制。我们提出基于奖励的速度匹配(RVM),一种无需轨迹的简单更新方法,直接作用于速度场。RVM强化高奖励生成方向,抑制低奖励方向,并包含可选锚点项控制与参考速度的偏移。值得注意的是,该框架具有通用性,可涵盖近期微调方法如RAM和DiffusionNFT作为特例。在多种大规模扩散模型奖励微调任务中,RVM在显著降低训练成本的同时,表现竞争力甚至更优。进一步发现,速度更新简化后,具体损失形式影响较小,关键在于奖励与锚点设计。对于视频生成,标准偏好奖励易导致视觉清晰但近乎静态的输出;引入新的动态追踪奖励后,运动质量显著提升,整体VBench得分也提高。结果表明,扩散模型的可扩展奖励微调应以原生速度表示为基准,而非基于似然的策略优化。

原文摘要 · Abstract (English)

Reward fine-tuning is becoming an important tool for adapting diffusion models to human preferences and task-specific objectives, but existing methods largely inherit policy-gradient machinery from large language models. Unlike autoregressive models, diffusion models do not provide tractable likelihoods for generated samples. As a result, current approaches either construct trajectory likelihoods from stochastic denoising transitions or approximate endpoint likelihoods with evidence lower bound, introducing additional computation and algorithmic complexity. We demonstrate that this likelihood-based machinery is not necessary for effective diffusion reward fine-tuning. We propose reward-based velocity matching (RVM), a simple trajectory-free update that acts directly on the velocity field. RVM reinforces directions associated with high-reward generations, suppresses those with low reward, and involves an optional anchor term controlling drift from a reference velocity. Notably, it provides a general framework that recovers recent fine-tuning methods, including RAM and DiffusionNFT, as special cases. Across various large-scale diffusion models reward fine-tuning tasks, RVM is competitive with or outperforms trajectory-based policy-gradient methods under substantially reduced training cost. We further find that, once the velocity update is simplified, the particular loss variant matters less than reward and anchor design. For video generation, standard preference rewards can favor visually clean but nearly static outputs; introducing a new dynamic-tracking reward that substantially improve motions while improving overall VBench performance. These results suggest that scalable reward fine-tuning for diffusion models is better posed in the native velocity representation than as likelihood-based policy optimization.

扩散模型强化学习速度匹配微调

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