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
Loading...
Thumbnail Image
Name

Covariance Mismatch in Diffusion Models.pdf

Type

Main Document

Version

http://purl.org/coar/version/c_71e4c1898caa6e32

Access type

openaccess

License Condition

N/A

Size

44.74 MB

Format

Adobe PDF

Checksum (MD5)

0c75962c4e31b95c3ee1624dae045db8

Loading...
Thumbnail Image
Name

Covariance Mismatch in Diffusion Models (with Appendix).pdf

Type

Supplementary Material/information

Version

http://purl.org/coar/version/c_71e4c1898caa6e32

Access type

openaccess

License Condition

N/A

Size

139.37 MB

Format

Adobe PDF

Checksum (MD5)

ddd859a67da10bade71d3c9f4b20e3df

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