对抗性扩散蒸馏

Adversarial Diffusion Distillation

罗宾·罗姆巴赫 Robin Rombach · Stability AI · 2023-11-28 · arXiv:2311.17042 ↗ · 被引 807

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了对抗性扩散蒸馏(ADD),一种新颖的训练方法,能够仅用 1-4 步高效采样大规模基础图像扩散模型,同时保持高图像质量。我们利用分数蒸馏,将大规模现成图像扩散模型作为教师信号,并结合对抗性损失,确保即使在一步或两步的低步数情况下也能保持高图像保真度。我们的分析表明,我们的模型在单步中明显优于现有的少步方法(GAN、潜在一致性模型),并在仅四步中达到最先进扩散模型(SDXL)的性能。ADD 是首个解锁基于基础模型的单步实时图像合成的方法。代码和权重可在 https://github.com/Stability-AI/generative-models 和 https://huggingface.co/stabilityai/ 获取。

We introduce Adversarial Diffusion Distillation (ADD), a novel training approach that efficiently samples large-scale foundational image diffusion models in just 1-4 steps while maintaining high image quality. We use score distillation to leverage large-scale off-the-shelf image diffusion models as a teacher signal in combination with an adversarial loss to ensure high image fidelity even in the low-step regime of one or two sampling steps. Our analyses show that our model clearly outperforms existing few-step methods (GANs, Latent Consistency Models) in a single step and reaches the performance of state-of-the-art diffusion models (SDXL) in only four steps. ADD is the first method to unlock single-step, real-time image synthesis with foundation models. Code and weights available under https://github.com/Stability-AI/generative-models and https://huggingface.co/stabilityai/ .

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 13)

阅读逐段中英对照全文 →