Understanding DDPM Latent Codes Through Optimal Transport

02/14/2022
by   Valentin Khrulkov, et al.
0

Diffusion models have recently outperformed alternative approaches to model the distribution of natural images, such as GANs. Such diffusion models allow for deterministic sampling via the probability flow ODE, giving rise to a latent space and an encoder map. While having important practical applications, such as estimation of the likelihood, the theoretical properties of this map are not yet fully understood. In the present work, we partially address this question for the popular case of the VP SDE (DDPM) approach. We show that, perhaps surprisingly, the DDPM encoder map coincides with the optimal transport map for common distributions; we support this claim theoretically and by extensive numerical experiments.

READ FULL TEXT

page 7

page 8

research
07/29/2020

Generalization Properties of Optimal Transport GANs with Latent Distribution Learning

The Generative Adversarial Networks (GAN) framework is a well-establishe...
research
05/27/2022

Maximum Likelihood Training of Implicit Nonlinear Diffusion Models

Whereas diverse variations of diffusion models exist, expanding the line...
research
02/16/2022

GAN Estimation of Lipschitz Optimal Transport Maps

This paper introduces the first statistically consistent estimator of th...
research
10/02/2018

Sinkhorn AutoEncoders

Optimal Transport offers an alternative to maximum likelihood for learni...
research
07/21/2023

DPM-OT: A New Diffusion Probabilistic Model Based on Optimal Transport

Sampling from diffusion probabilistic models (DPMs) can be viewed as a p...
research
02/19/2019

2-Wasserstein Approximation via Restricted Convex Potentials with Application to Improved Training for GANs

We provide a framework to approximate the 2-Wasserstein distance and the...
research
05/24/2023

Debias Coarsely, Sample Conditionally: Statistical Downscaling through Optimal Transport and Probabilistic Diffusion Models

We introduce a two-stage probabilistic framework for statistical downsca...

Please sign up or login with your details

Forgot password? Click here to reset