跳到正文
arXiv · cs.LG· Jaihoon Kim, Lingxiao Li, Soshi Shimada, Roshni Cooper, Minhyuk Sung, Yiwei Zhao·· 19 小时前

sDMD:用 Sinkhorn 分布匹配蒸馏改进扩散模型少步生成

sDMD: Sinkhorn Distribution Matching Distillation

AI 导读

sDMD 在 DMD 的积分 KL 目标上引入基于有限批次估计的 Sinkhorn 散度最优传输项,并按判别器输出自适应激活该修正,以处理教师密度低区域的离群样本。在 FLUX.1-dev、Qwen-Image-20B 和 Wan2.1-14B 上以 2–4 步采样时,sDMD 在 GenEval、DPG、WISE 和 VBench 上优于蒸馏基线,且样本多样性较 DMD2 提升。

来源:arXiv · cs.LG · arxiv.org