Fast High-Resolution Image Synthesis with Latent Adversarial Diffusion Distillation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sauer, Axel, Boesel, Frederic, Dockhorn, Tim, Blattmann, Andreas, Esser, Patrick, Rombach, Robin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909139900301312
author Sauer, Axel
Boesel, Frederic
Dockhorn, Tim
Blattmann, Andreas
Esser, Patrick
Rombach, Robin
author_facet Sauer, Axel
Boesel, Frederic
Dockhorn, Tim
Blattmann, Andreas
Esser, Patrick
Rombach, Robin
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2403_12015
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Fast High-Resolution Image Synthesis with Latent Adversarial Diffusion Distillation
Sauer, Axel
Boesel, Frederic
Dockhorn, Tim
Blattmann, Andreas
Esser, Patrick
Rombach, Robin
Computer Vision and Pattern Recognition
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.
title Fast High-Resolution Image Synthesis with Latent Adversarial Diffusion Distillation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.12015