Boosting Latent Diffusion with Perceptual Objectives - Inria - Institut national de recherche en sciences et technologies du numérique
Preprints, Working Papers, ... Year : 2024

Boosting Latent Diffusion with Perceptual Objectives

Abstract

Latent diffusion models (LDMs) power state-of-the-art high-resolution generative image models. LDMs learn the data distribution in the latent space of an autoencoder (AE) and produce images by mapping the generated latents into RGB image space using the AE decoder. While this approach allows for efficient model training and sampling, it induces a disconnect between the training of the diffusion model and the decoder, resulting in a loss of detail in the generated images. To remediate this disconnect, we propose to leverage the internal features of the decoder to define a latent perceptual loss (LPL). This loss encourages the models to create sharper and more realistic images. Our loss can be seamlessly integrated with common autoencoders used in latent diffusion models, and can be applied to different generative modeling paradigms such as DDPM with epsilon and velocity prediction, as well as flow matching. Extensive experiments with models trained on three datasets at 256 and 512 resolution show improved quantitative – with boosts between 6% and 20% in FID – and qualitative results when using our perceptual loss.
Fichier principal
Vignette du fichier
2411.04873v1.pdf (55.99 Mo) Télécharger le fichier
Origin Files produced by the author(s)

Dates and versions

hal-04837733 , version 1 (13-12-2024)

Licence

Identifiers

Cite

Tariq Berrada, Pietro Astolfi, Jakob Verbeek, Melissa Hall, Marton Havasi, et al.. Boosting Latent Diffusion with Perceptual Objectives. 2024. ⟨hal-04837733⟩
15 View
6 Download

Altmetric

Share

More