分布反向追踪构建更快收敛轨迹的单步扩散蒸馏

💡 原文中文,约400字,阅读约需1分钟。
📝

内容提要

本文介绍了一种名为Score identity Distillation(SiD)的无数据方法,通过将预训练扩散模型的生成能力提炼到一个单步生成器中,实现了快速的Fréchet inception distance (FID) 减小。SiD算法在蒸馏过程中显示出高迭代效率,并超越了其他蒸馏方法,无论是一步还是几步,无数据还是依赖于训练数据。这一成就不仅重新定义了扩散蒸馏中效率和效果的基准,还在更广泛的扩散生成领域中也有着重要的意义。

🎯

关键要点

  • 引入了Score identity Distillation(SiD),一种无数据方法。

  • SiD将预训练扩散模型的生成能力提炼到单步生成器中。

  • 通过重构前向扩散过程为半隐式分布,创造了创新的损失机制。

  • 该机制使用合成图像训练生成器,消除了对真实数据的需求。

  • SiD实现了快速的Fréchet inception distance (FID) 减小。

  • 在四个基准数据集上评估,SiD显示出高迭代效率,超越其他蒸馏方法。

  • 这一成就重新定义了扩散蒸馏中的效率和效果基准。

  • SiD在更广泛的扩散生成领域中具有重要意义。

  • 我们的PyTorch实现将在GitHub上公开获取。

➡️

继续阅读