[2411.01629] Denoising Diffusions with Optimal Transport: Localization, Curvature, and Multi-Scale Complexity
Summary
This paper explores denoising diffusions using optimal transport, focusing on localization, curvature, and multi-scale complexity in generative models.
Why It Matters
Understanding the mechanics of denoising in diffusion models is crucial for advancements in machine learning. This research provides insights into the complexities of denoising processes, which can enhance model performance in various applications, from image generation to data reconstruction.
Key Takeaways
- Denoising in diffusion models is influenced by the curvature function, affecting localization uncertainty.
- The paper introduces a multi-scale curvature complexity that quantifies denoising difficulty based on signal-to-noise ratios.
- The diffuse-then-denoise process is characterized by the net contraction of the forward diffusion chain.
- Denoising is easier when the integrated tail function of curvature is light, indicating favorable conditions for reconstruction.
- Non-log-concave examples illustrate the practical implications of multi-scale complexity in denoising tasks.
Statistics > Machine Learning arXiv:2411.01629 (stat) [Submitted on 3 Nov 2024 (v1), last revised 14 Feb 2026 (this version, v2)] Title:Denoising Diffusions with Optimal Transport: Localization, Curvature, and Multi-Scale Complexity Authors:Tengyuan Liang, Kulunu Dharmakeerthi, Takuya Koriyama View a PDF of the paper titled Denoising Diffusions with Optimal Transport: Localization, Curvature, and Multi-Scale Complexity, by Tengyuan Liang and 2 other authors View PDF HTML (experimental) Abstract:Adding noise is easy; what about denoising? Diffusion is easy; what about reverting a diffusion? Diffusion-based generative models aim to denoise a Langevin diffusion chain, moving from a log-concave equilibrium measure $\nu$, say an isotropic Gaussian, back to a complex, possibly non-log-concave initial measure $\mu$. The score function performs denoising, moving backward in time, and predicting the conditional mean of the past location given the current one. We show that score denoising is the optimal backward map in transportation cost. What is its localization uncertainty? We show that the curvature function determines this localization uncertainty, measured as the conditional variance of the past location given the current. We study in this paper the effectiveness of the diffuse-then-denoise process: the contraction of the forward diffusion chain, offset by the possible expansion of the backward denoising chain, governs the denoising difficulty. For any initial measure $\mu$, w...