vix.ing · top · new · best · stats · spec

Diffusion models for Gaussian distributions: Exact solutions and Wasserstein errors

2024/05/23 by Emile Pierret, Pierret, Emile, Bruno Galerne +1 · 6 citations
Mathematics · Medicine · #Advanced Neuroimaging Techniques and Applications #FOS: Computer and information sciences #FOS: Electrical engineering #FOS: Mathematics #Image and Video Processing (eess.IV) #MRI in cancer diagnosis #Machine Learning (cs.LG) #Probability (math.PR) #Statistical Methods and Inference #electronic engineering #information engineering

paper · pdf · doi:10.48550/arxiv.2405.14250

openalex publication_date 2024/05/23 · openalex created_date 2025/10/10 · openalex updated_date 2026/08/01

Abstract

Diffusion or score-based models recently showed high performance in image generation. They rely on a forward and a backward stochastic differential equations (SDE). The sampling of a data distribution is achieved by numerically solving the backward SDE or its associated flow ODE. Studying the convergence of these models necessitates to control four different types of error: the initialization error, the truncation error, the discretization error and the score approximation. In this paper, we theoretically study the behavior of diffusion models and their numerical implementation when the data distribution is Gaussian. Our first contribution is to derive the analytical solutions of the backward SDE and the probability flow ODE and to prove that these solutions and their discretizations are all Gaussian processes. Our second contribution is to compute the exact Wasserstein errors between the target and the numerically sampled distributions for any numerical scheme. This allows us to monitor convergence directly in the data space, while experimental works limit their empirical analysis to Inception features. An implementation of our code is available online.

Cited by

Related