Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Preprints and Working Papers
  4. Covariance Mismatch in Diffusion Models
 
preprint

Covariance Mismatch in Diffusion Models

Everaert, Martin Nicolas  
•
Süsstrunk, Sabine  
•
Achanta, Radhakrishna  
November 2024

Diffusion models are usually trained using isotropic noise. Yet, common data distributions are strongly anisotropic. We explain that this covariance mismatch negatively impacts the model in several ways. It leads to the model predicting only the high-variance components at high noise levels and only the low-variance components at low noise levels, requiring a wide range of noise levels during training and inference to model all components accurately. This partition of components across noise levels also prevents smaller timesteps from correcting predictions of larger timesteps, and limits diffusion editing to only low-variance components. We show two approaches to realign the noise and data covariances: whitening the data distribution or coloring the noise distribution. We apply our approach on 2D point distributions and, using a Fourier-based approach, on images. Realigning covariances allows the model to focus more equally on all components, improving editing and enabling fewer noise levels in training. Models trained with realigned covariances offer greater flexibility in the choice of timesteps during inference and can even generate reasonable output while being trained on just a single timestep. Project page: https://ivrl.github.io/covariance-mismatch

  • Files
  • Details
  • Metrics
Type
preprint
Author(s)
Everaert, Martin Nicolas  

EPFL

Süsstrunk, Sabine  

EPFL

Achanta, Radhakrishna  
Date Issued

2024-11

URL

Project page

https://ivrl.github.io/covariance-mismatch
Written at

EPFL

EPFL units
IVRL  
Available on Infoscience
November 27, 2024
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/242173
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés