跳到正文
arXiv · Machine Learning· Paribesh Regmi, Rui Li·· 8 小时前AI 评分45

用于单步扩散的累积边缘匹配(CMM)

Cumulative Marginal Matching for One-Step Diffusion

AI 导读

研究提出累积边缘匹配(CMM)目标,用于少步和单步扩散生成,直接匹配基础模型与少步模型在整个生成过程中的中间边缘分布。CMM以 Wasserstein-2 距离刻画累积差异,并将其转化为可利用流匹配中间边缘样本优化的替代目标。标准基准实验显示,在相同训练预算下,CMM的生成质量优于 Shortcut-CSL 及其他先进单步、少步基线。

来源:arXiv · Machine Learning · arxiv.org