TensorX
返回文献探索

Paper · arXiv 2403.12015

Fast High-Resolution Image Synthesis with Latent Adversarial Diffusion Distillation

Axel Sauer, Frederic Boesel, Tim Dockhorn, Andreas Blattmann, Patrick Esser, Robin Rombach

71 upvotesMarch 18, 2024arXiv 预印本
AI 摘要

Latent Adversarial Diffusion Distillation (LADD) simplifies the training and improves the performance of diffusion models by using generative features from latent diffusion models, enabling fast and high-resolution image synthesis.

diffusion modelsadversarial diffusion distillation (ADD)pretrained DINOv2 discriminatorLatent Adversarial Diffusion Distillation (LADD)pixel-based ADDlatent diffusion modelshigh-resolution multi-aspect ratio image synthesisStable Diffusion 3 (8B)SD3-Turboimage editingimage inpainting

Abstract

Diffusion models are the main driver of progress in image and video synthesis, but suffer from slow inference speed. Distillation methods, like the recently introduced adversarial diffusion distillation (ADD) aim to shift the model from many-shot to single-step inference, albeit at the cost of expensive and difficult optimization due to its reliance on a fixed pretrained DINOv2 discriminator. We introduce Latent Adversarial Diffusion Distillation (LADD), a novel distillation approach overcoming the limitations of ADD. In contrast to pixel-based ADD, LADD utilizes generative features from pretrained latent diffusion models. This approach simplifies training and enhances performance, enabling high-resolution multi-aspect ratio image synthesis. We apply LADD to Stable Diffusion 3 (8B) to obtain SD3-Turbo, a fast model that matches the performance of state-of-the-art text-to-image generators using only four unguided sampling steps. Moreover, we systematically investigate its scaling behavior and demonstrate LADD's effectiveness in various applications such as image editing and inpainting.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号