← 最新论文
🤖 AI

Continuous-Time Distribution Matching for Few-Step Diffusion Distillation

本文介绍了连续时间分布匹配(CDM),这是一种新颖的蒸馏框架,它通过将稀疏的离散监督替换为动态的连续时间优化和轨迹外对齐来增强少步扩散模型,从而在不依赖复杂辅助模块的情况下实现高视觉保真度。

原作者: Tao Liu, Hao Yan, Mengting Chen, Taihang Hu, Zhengrong Yue, Zihao Pan, Jinsong Lan, Xiaoyong Zhu, Ming-Ming Cheng, Bo Zheng, Yaxing Wang

发布于 2026-05-08
📖 1 分钟阅读☕ 轻松阅读

原作者: Tao Liu, Hao Yan, Mengting Chen, Taihang Hu, Zhengrong Yue, Zihao Pan, Jinsong Lan, Xiaoyong Zhu, Ming-Ming Cheng, Bo Zheng, Yaxing Wang

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

想象一下,你正试图教导一位才华横溢但动作缓慢的艺术家(即教师),如何仅用寥寥几笔快速 strokes 完成一幅杰作,而非其惯常的数百层缓慢而细致的涂抹。这正是扩散蒸馏(Diffusion Distillation)所面临的挑战:如何在保持质量的同时,让 AI 图像生成器变得快速。

本文介绍了一种名为**CDM(连续时间分布匹配,Continuous-Time Distribution Matching)**的新方法,它解决了以往教学方法中的一个特定问题。以下通过简单的类比进行拆解:

问题: “走走停停”式的教师

以往的方法(如 DMD2)试图通过在少数几个特定的、固定的检查点来检查学生的工作,以此教导快速的学生。

  • 类比:想象一位驾驶教练,只在上午 10:00、10:15 和 10:30 这几个确切时刻检查你的车辆位置。他们会说:"10:15 时你做得完美”,但他们并不关心你在这些时间点之间做了什么。
  • 结果:由于学生仅从这些稀疏的“快照”中学习,他们在这些时间点之间往往驾驶得 erratic(不稳定)。当他们试图快速驾驶(即在 4 步内生成图像)时,容易偏离道路,导致生成模糊、"过度平滑”的图像,缺乏发丝或织物纹理等精细细节。为了解决这个问题,旧方法不得不添加复杂的“安全网”(如 GAN 或奖励模型),这些工具运行起来既笨重又昂贵。

解决方案: “平滑高速公路”(CDM)

作者提出了CDM,将教学风格从“走走停停”转变为“平滑高速公路”。他们通过两个主要技巧实现这一点:

1. 动态调度(不再使用固定检查点)

教师不再仅在固定时间检查学生,而是在整个旅程中的随机时刻进行检查。

  • 类比:驾驶教练现在会在随机时间跳上车——有时是 10:03,有时是 10:27,有时是 10:41。他们会检查学生在道路上的任何位置,而不仅仅是预定的停靠点。
  • 益处:学生学会了在任何地方都平稳驾驶,而不仅仅是在特定的检查点。这防止了导致图像模糊的“抖动”式驾驶。

2. “越野”测试(基于速度的外推)

这是本文最独特的创新。当学生迈出一大步(因为他们试图加快速度)时,可能会越过理想路径。CDM 会主动检查如果学生真的越过了会发生什么。

  • 类比:想象学生正在快速驾驶。教练说:“好吧,你迈出了一大步。现在,让我们假设你根据当前的速度和方向,再迈出一步。”然后,教练检查那个“想象”中的位置是否仍在道路上。如果学生的速度有误,那个想象的位置就会严重偏离道路。教练会在学生真正犯错之前,就纠正他们的速度。
  • 益处:这就像一个自我修正的 GPS。它迫使学生学习平稳、一致的速度(velocity),这样即使他们迈出巨大而快速的步伐,也不会撞车或偏离路径。这无需额外的“安全网”即可保留锐利的细节。

结果:快速且锐利

本文在强大的图像生成器(SD3-Medium 和 Longcat-Image)上测试了这种新方法。

  • 速度:新学生仅需4 步即可生成高质量图像(非常快)。
  • 质量:与以往快速方法相比,图像更锐利,拥有更多细粒度的细节(如逼真的纹理)。
  • 简洁性:与其他快速方法不同,CDM 不需要复杂的额外工具(如 GAN 或奖励模型)来修复模糊问题。它通过“平滑高速公路”式的教学方法自行修复。

总结

简而言之,这篇论文指出:“要让 AI 图像生成变快,就停止仅在固定时间检查学生。相反,要在随机时间检查他们,并在‘想象’的越野路径上测试他们的速度。这教会他们平稳且快速地驾驶,从而在不依赖复杂额外设备的情况下,生成清晰、细节丰富的图像。”

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →