通过得分隐式匹配实现一步扩散蒸馏

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

内容提要

我们提出了一种无数据的创新方法——Score identity Distillation(SiD),将预训练扩散模型的生成能力提炼到单步生成器中。通过重构前向扩散过程并利用与分数相关的身份,SiD实现了高效生成,显著缩短了生成时间,并在多个基准数据集上超越了其他蒸馏方法。我们的PyTorch实现将公开在GitHub上。

🎯

关键要点

  • 引入了Score identity Distillation(SiD),一种无数据的创新方法。
  • SiD将预训练扩散模型的生成能力提炼到单步生成器中。
  • 通过重构前向扩散过程并利用与分数相关的身份,创造了一种创新的损失机制。
  • 该机制使用自己合成的图像训练生成器,消除了对真实数据的需求。
  • SiD实现了快速的Fréchet inception distance (FID) 减小,显著缩短生成时间。
  • 在四个基准数据集上评估后,SiD算法显示出高迭代效率,超越了其他蒸馏方法。
  • 这一成就重新定义了扩散蒸馏中的效率和效果基准。
  • SiD在更广泛的扩散生成领域中具有重要意义。
  • 我们的PyTorch实现将在GitHub上公开获取。
➡️

继续阅读