Robin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser, Björn Ommer
By decomposing the image formation process into a sequential application of denoising autoencoders, diffusion models (DMs) achieve state-of-the-art synthesis results on image data and beyond. Additionally, their formulation allows for a guiding mechanism to control the image generation process without retraining. However, since these models typically operate directly in pixel space, optimization of powerful DMs often consumes hundreds of GPU days and inference is expensive due to sequential evaluations. To enable DM training on limited computational resources while retaining their quality and flexibility, we apply them in the latent space of powerful pretrained autoencoders. In contrast to previous work, training diffusion models on such a representation allows for the first time to reach a near-optimal point between complexity reduction and detail preservation, greatly boosting visual fidelity. By introducing cross-attention layers into the model architecture, we turn diffusion models into powerful and flexible generators for general conditioning inputs such as text or bounding boxes and high-resolution synthesis becomes possible in a convolutional manner. Our latent diffusion models (LDMs) achieve a new state of the art for image inpainting and highly competitive performance on various tasks, including unconditional image generation, semantic scene synthesis, and super-resolution, while significantly reducing computational requirements compared to pixel-based DMs. Code is available at https://github.com/CompVis/latent-diffusion .
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Image Generation | CelebA-HQ 256x256 | FID | 5.11 | LDM-4 |
| Image Generation | ImageNet 512x512 | FID | 3.6 | Latent Diffusion (LDM-4-G) |
| Image Generation | ImageNet 512x512 | Inception score | 247.67 | Latent Diffusion (LDM-4-G) |
| Image Generation | COCO (Common Objects in Context) | FID | 12.63 | Latent Diffusion (LDM-KL-8-G) |
| Image Generation | Conceptual Captions | FID | 17.01 | LDM-4 |
| Image Generation | DrawBench | Aesthetics (Laion Aesthtetics Predictor) | 5.4292 | Stable Diffusion 1.5 |
| Image Generation | DrawBench | Human Preference Alignement (HPSv2) | 0.2646 | Stable Diffusion 1.5 |
| Image Generation | DrawBench | Text Alignement (SentenceBERT) | 0.5997 | Stable Diffusion 1.5 |
| Image Generation | LayoutBench | AP | 9.9 | LDM |
| Image Generation | COCO-Stuff 256x256 | FID | 40.96 | LDM-4 (200steps) |
| Image Generation | COCO-Stuff 256x256 | FID | 42.06 | LDM-8 (100steps) |
| Image Reconstruction | Ultra-High Resolution Image Reconstruction Benchmark | PSNR | 26.86 | SD-VAE (16x16) |
| Image Reconstruction | Ultra-High Resolution Image Reconstruction Benchmark | rFID | 1.07 | SD-VAE (16x16) |
| Text-to-Image Generation | COCO (Common Objects in Context) | FID | 12.63 | Latent Diffusion (LDM-KL-8-G) |
| Text-to-Image Generation | Conceptual Captions | FID | 17.01 | LDM-4 |
| Text-to-Image Generation | DrawBench | Aesthetics (Laion Aesthtetics Predictor) | 5.4292 | Stable Diffusion 1.5 |
| Text-to-Image Generation | DrawBench | Human Preference Alignement (HPSv2) | 0.2646 | Stable Diffusion 1.5 |
| Text-to-Image Generation | DrawBench | Text Alignement (SentenceBERT) | 0.5997 | Stable Diffusion 1.5 |
| 10-shot image generation | COCO (Common Objects in Context) | FID | 12.63 | Latent Diffusion (LDM-KL-8-G) |
| 10-shot image generation | Conceptual Captions | FID | 17.01 | LDM-4 |
| 10-shot image generation | DrawBench | Aesthetics (Laion Aesthtetics Predictor) | 5.4292 | Stable Diffusion 1.5 |
| 10-shot image generation | DrawBench | Human Preference Alignement (HPSv2) | 0.2646 | Stable Diffusion 1.5 |
| 10-shot image generation | DrawBench | Text Alignement (SentenceBERT) | 0.5997 | Stable Diffusion 1.5 |
| 1 Image, 2*2 Stitchi | COCO (Common Objects in Context) | FID | 12.63 | Latent Diffusion (LDM-KL-8-G) |
| 1 Image, 2*2 Stitchi | Conceptual Captions | FID | 17.01 | LDM-4 |
| 1 Image, 2*2 Stitchi | DrawBench | Aesthetics (Laion Aesthtetics Predictor) | 5.4292 | Stable Diffusion 1.5 |
| 1 Image, 2*2 Stitchi | DrawBench | Human Preference Alignement (HPSv2) | 0.2646 | Stable Diffusion 1.5 |
| 1 Image, 2*2 Stitchi | DrawBench | Text Alignement (SentenceBERT) | 0.5997 | Stable Diffusion 1.5 |