arXiv:2606.15048cs.LGcs.CV2026-06中稿 · ICML

用时序差分优化扩散模型生成一致性,少步采样效果更明显。

Temporal Difference Learning for Diffusion Models

论文配图:Temporal Difference Learning for Diffusion Models
图 1 · 摘自论文原文
  • 将去噪过程建模为马尔可夫奖励过程,引入时序差分损失约束多步一致性。
  • 在少步采样下显著提升生成质量(FID降低),低算力场景优势突出。
  • 适用于离散与连续时间扩散模型,可作为通用插件模块直接替换训练目标。

扩散模型通常采用关注单个时间步或相邻对的局部去噪目标,未能强制去噪轨迹上预测的一致性。这种跨时间不一致会降低性能,尤其在少步采样时更为明显。本文提出一种时序差分(TD)目标,惩罚模型沿去噪路径的多步进展不一致。通过将扩散过程重述为马尔可夫奖励过程,并将去噪视为强化学习中的策略评估问题,我们推导出统一的TD方法,适用于离散与连续时间扩散形式。进一步提出一种基于样本的重加权方法,稳定训练过程。实验表明,使用我们的TD训练能显著提升样本质量(以FID衡量),且在采样步数较少时优势更明显,凸显其在低计算预算场景下的实用性。我们进行了消融实验验证设计选择,包括成对损失重加权、正则化权重和一步步长。总体而言,该TD方法可作为通用插件,增强跨时间一致性,提升各类扩散生成模型的生成质量。

原文摘要 · Abstract (English)

Diffusion models are typically trained with objectives that focus on local denoising targets at individual time steps (or adjacent pairs), which do not enforce consistency between predictions along the denoising trajectory. This lack of cross-time consistency can degrade performance, especially for few-step samplers. We introduce a temporal difference (TD) objective that penalizes inconsistency of the model's multi-step progress along the denoising path. By reformulating the diffusion process as a Markov reward process and casting denoising as a policy evaluation problem in reinforcement learning, we derive a unified TD approach that applies to both discrete- and continuous-time diffusion formulations. We further propose a principled sample-based reweighting method that stabilizes training. Empirically, we show that using our TD training can significantly improve sample quality measured by FID, with stronger advantages when the number of sampling steps is small, highlighting its practical utility under low-computation-budget scenarios. We provide ablation studies to justify our design choices, including pairwise loss reweighting, regularization weight, and one-step stride. Overall, our TD approach can be a general drop-in that enforces cross-time consistency and improves generation quality across different diffusion generative models.

扩散模型时序差分生成质量少步采样

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