arXiv:2410.07679cs.CV2024-10中稿 · ACM MM 2024 Oral被引 36

提出关系扩散蒸馏方法,让小模型生成图像更快更准。

Relational Diffusion Distillation for Efficient Image Generation

  • 通过跨样本关系交互,提升教师模型知识传递效率。
  • 1步采样下FID降低1.47,比现有方法快256倍。
  • 适合资源受限设备上的高效图像生成应用。

尽管扩散模型在图像生成领域表现卓越,但其高推理延迟限制了在计算资源有限的边缘设备上的广泛应用。为此,众多无需训练的采样方法被提出以减少扩散模型所需的采样步数,但在极小步数下表现不佳。得益于知识蒸馏技术的发展,现有训练方案在极低步数下已取得优异效果。然而,当前方法多聚焦于设计新颖的扩散模型采样策略,而如何更有效地从教师模型中转移扩散知识这一更具价值的问题却少有研究。为此,我们提出专为扩散模型设计的关系扩散蒸馏(RDD)方法。不同于以往仅在像素级或特征分布上对齐师生模型的方法,我们的方法在蒸馏过程中引入跨样本关系交互,并缓解多样本交互带来的内存压力。所提RDD显著提升了扩散模型中渐进式蒸馏框架的有效性。在CIFAR-10和ImageNet等多个数据集上的大量实验表明,相较于最先进的扩散蒸馏方法,本方法在1步采样下实现1.47的FID下降,且相比DDIM策略获得256倍加速。代码已开源:https://github.com/cantbebetter2/RDD。

原文摘要 · Abstract (English)

Although the diffusion model has achieved remarkable performance in the field of image generation, its high inference delay hinders its wide application in edge devices with scarce computing resources. Therefore, many training-free sampling methods have been proposed to reduce the number of sampling steps required for diffusion models. However, they perform poorly under a very small number of sampling steps. Thanks to the emergence of knowledge distillation technology, the existing training scheme methods have achieved excellent results at very low step numbers. However, the current methods mainly focus on designing novel diffusion model sampling methods with knowledge distillation. How to transfer better diffusion knowledge from teacher models is a more valuable problem but rarely studied. Therefore, we propose Relational Diffusion Distillation (RDD), a novel distillation method tailored specifically for distilling diffusion models. Unlike existing methods that simply align teacher and student models at pixel level or feature distributions, our method introduces cross-sample relationship interaction during the distillation process and alleviates the memory constraints induced by multiple sample interactions. Our RDD significantly enhances the effectiveness of the progressive distillation framework within the diffusion model. Extensive experiments on several datasets (e.g., CIFAR-10 and ImageNet) demonstrate that our proposed RDD leads to 1.47 FID decrease under 1 sampling step compared to state-of-the-art diffusion distillation methods and achieving 256x speed-up compared to DDIM strategy. Code is available at https://github.com/cantbebetter2/RDD.

扩散模型知识蒸馏图像生成高效推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。