arXiv:2410.16794cs.CVcs.AI2024-10NeurIPS被引 79

将扩散模型压缩为单步生成器,效果几乎无损且无需训练数据。

One-Step Diffusion Distillation through Score Implicit Matching

  • 通过隐式得分匹配,直接优化生成器与扩散模型的得分差异。
  • CIFAR10上无条件/有条件生成FID分别达2.06和1.96,接近原始模型性能。
  • 可应用于文本到图像生成,单步结果超越主流方案,适合工业部署。

尽管扩散模型在众多生成任务中表现优异,但需大量采样步骤才能生成真实样本。为此,研究者致力于将预训练扩散模型蒸馏为更高效的模型,但现有方法通常仍需多步推理或性能显著下降。本文提出得分隐式匹配(Score Implicit Matching, SIM),一种将预训练扩散模型蒸馏为单步生成器的新方法,保持接近原模型的生成能力,且无需训练数据。该方法基于一个关键观察:在特定条件下,可高效计算生成器与扩散模型之间广泛得分差异的梯度。实验表明,SIM在单步生成上表现强劲:在CIFAR10数据集上,无条件生成FID为2.06,类条件生成FID为1.96。进一步将SIM应用于领先的基于Transformer的扩散模型,蒸馏出一个单步文本到图像生成器,在美学评分上达到6.42,与原始多步模型持平,明显优于其他单步方案(SDXL-TURBO: 5.33,SDXL-LIGHTNING: 5.34,HYPER-SDXL: 5.85)。论文将发布此具备工业应用潜力的单步Transformer文本到图像生成器。

原文摘要 · Abstract (English)

Despite their strong performances on many generative tasks, diffusion models require a large number of sampling steps in order to generate realistic samples. This has motivated the community to develop effective methods to distill pre-trained diffusion models into more efficient models, but these methods still typically require few-step inference or perform substantially worse than the underlying model. In this paper, we present Score Implicit Matching (SIM) a new approach to distilling pre-trained diffusion models into single-step generator models, while maintaining almost the same sample generation ability as the original model as well as being data-free with no need of training samples for distillation. The method rests upon the fact that, although the traditional score-based loss is intractable to minimize for generator models, under certain conditions we can efficiently compute the gradients for a wide class of score-based divergences between a diffusion model and a generator. SIM shows strong empirical performances for one-step generators: on the CIFAR10 dataset, it achieves an FID of 2.06 for unconditional generation and 1.96 for class-conditional generation. Moreover, by applying SIM to a leading transformer-based diffusion model, we distill a single-step generator for text-to-image (T2I) generation that attains an aesthetic score of 6.42 with no performance decline over the original multi-step counterpart, clearly outperforming the other one-step generators including SDXL-TURBO of 5.33, SDXL-LIGHTNING of 5.34 and HYPER-SDXL of 5.85. We will release this industry-ready one-step transformer-based T2I generator along with this paper.

扩散模型模型蒸馏单步生成文本生成图像

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