Mathematical analysis of singularities in the diffusion model under the submanifold assumption
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
This paper concerns the mathematical analyses of the diffusion model in machine learning. The drift term of the backward sampling process is represented as a conditional expectation involving the data distribution and the forward diffusion. The training process aims to find such a drift function by minimizing the mean-squared residue related to the conditional expectation. Using small-time approximations of the Green's function of the forward diffusion, we show that the analytical mean drift function in DDPM and the score function in SGM asymptotically blow up in the final stages of the sampling process for singular data distributions such as those concentrated on lower-dimensional manifolds, and are therefore difficult to approximate by a network. To overcome this difficulty, we derive a new target function and associated loss, which remains bounded even for singular data distributions. We validate the theoretical findings with several numerical examples.
Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 21 of its 34 references that are in the map. First posted on arXiv 19 Jan 2023.
Builds on
21 of 34 references.
Closest ancestors · 11
Same direction, newest first
- 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Models2 Nov
- 2022Diffusion Models: A Comprehensive Survey of Methods and Applications2 Sep
- 2022Understanding Diffusion Models: A Unified Perspective25 Aug
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2022Fast Sampling of Diffusion Models with Exponential Integrator29 Apr
- 2022Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Models17 Jan
- 2021Score-Based Generative Modeling with Critically-Damped Langevin Diffusion14 Dec
- 2021Structured Denoising Diffusion Models in Discrete State-Spaces7 Jul
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 10
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2018Neural Ordinary Differential Equationscited by 3,733
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
- 2020Generative Modeling with Denoising Auto-Encoders and Langevin Samplingcited by 118
- 2021Deep Generative Learning via Schr"odinger Bridgecited by 103
- 2021Deep Generative Modeling for Protein Designcited by 17
- 2021Interpreting diffusion score matching using normalizing flowcited by 9
- 2021DeepParticle: learning invariant measure by a deep neural network minimizing Wasserstein distance on data generated from an interacting particle methodcited by 9
Built on by
Citing papers per year: 2024: 7; 2025: 8; 2026: 5.
20 papers.
Sort:
By area:
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connectionscited by 47
- 2024A Geometric View of Data Complexity: Efficient Local Intrinsic Dimension Estimation with Diffusion Modelscited by 41
- 2024A Geometric Framework for Understanding Memorization in Generative Modelscited by 40
- 2025Memorization and Regularization in Generative Diffusion Modelscited by 39
- 2024A Geometric Explanation of the Likelihood OOD Detection Paradoxcited by 22
- 2025Improving the Euclidean Diffusion Generation of Manifold Data by Mitigating Score Function Singularitycited by 14
- 2024Tackling the Singularities at the Endpoints of Time Intervals in Diffusion Modelscited by 13
- 2025When Scores Learn Geometry: Rate Separations under the Manifold Hypothesiscited by 13
- 2024Masked Completion via Structured Diffusion with White-Box Transformerscited by 9
- 2025Resolving Memorization in Empirical Diffusion Model for Manifold Data in High-Dimensional Spacescited by 8
- 2025Anomaly Detection and Generation with Diffusion Models: A Surveycited by 8
- 2025When and how can inexact generative models still sample from the data manifold?cited by 4
8 more
- 2025Pathway to O(√d) Complexity bound under Wasserstein metric of flow-based modelscited by 4
- 2025Optimal Stopping in Latent Diffusion Modelscited by 1
- 2026Preconditioned One-Step Generative Modeling for Bayesian Inverse Problems in Function Spacescited by 1
- 2026Intrinsic Wasserstein Rates for Score-Based Generative Models on Smooth Manifoldscited by 1
- 2024Inconsistencies In Consistency Models: Better ODE Solving Does Not Imply Better Samplescited by 0
- 2026SSI-DM: Singularity Skipping Inversion of Diffusion Modelscited by 0
- 2026ML-based approach to classification and generation of structured light propagation in turbulent mediacited by 0
- 2026One Inverse Step is a Convex Program: Bayes-Limit Calibration of Diffusion Inversioncited by 0
- 2026One Inverse Step is a Convex Program: Bayes-Limit Calibration of Diffusion Inversion24 Aug 2026
- 2026Intrinsic Wasserstein Rates for Score-Based Generative Models on Smooth Manifolds15 May 2026
- 2026ML-based approach to classification and generation of structured light propagation in turbulent media4 Apr 2026
- 2026Preconditioned One-Step Generative Modeling for Bayesian Inverse Problems in Function Spaces16 Mar 2026
- 2026SSI-DM: Singularity Skipping Inversion of Diffusion Models2 Feb 2026
- 2025Pathway to O(√d) Complexity bound under Wasserstein metric of flow-based models7 Dec 2025
- 2025Optimal Stopping in Latent Diffusion Models9 Oct 2025
- 2025When Scores Learn Geometry: Rate Separations under the Manifold Hypothesis29 Sep 2025
- 2025When and how can inexact generative models still sample from the data manifold?11 Aug 2025
- 2025Anomaly Detection and Generation with Diffusion Models: A Survey11 Jun 2025
- 2025Improving the Euclidean Diffusion Generation of Manifold Data by Mitigating Score Function Singularity15 May 2025
- 2025Resolving Memorization in Empirical Diffusion Model for Manifold Data in High-Dimensional Spaces5 May 2025
8 more
- 2025Memorization and Regularization in Generative Diffusion Models27 Jan 2025
- 2024Inconsistencies In Consistency Models: Better ODE Solving Does Not Imply Better Samples13 Nov 2024
- 2024A Geometric Framework for Understanding Memorization in Generative Models31 Oct 2024
- 2024A Geometric View of Data Complexity: Efficient Local Intrinsic Dimension Estimation with Diffusion Models5 Jun 2024
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connections3 Apr 2024
- 2024Masked Completion via Structured Diffusion with White-Box Transformers3 Apr 2024
- 2024A Geometric Explanation of the Likelihood OOD Detection Paradox27 Mar 2024
- 2024Tackling the Singularities at the Endpoints of Time Intervals in Diffusion Models13 Mar 2024
- 2024Deep Generative Models through the Lens of the Manifold Hypothesis: A Survey and New Connectionscited by 47
- 2024A Geometric View of Data Complexity: Efficient Local Intrinsic Dimension Estimation with Diffusion Modelscited by 41
- 2025Memorization and Regularization in Generative Diffusion Modelscited by 39
- 2025Improving the Euclidean Diffusion Generation of Manifold Data by Mitigating Score Function Singularitycited by 14
- 2024Tackling the Singularities at the Endpoints of Time Intervals in Diffusion Modelscited by 13
- 2025When Scores Learn Geometry: Rate Separations under the Manifold Hypothesiscited by 13
- 2025Resolving Memorization in Empirical Diffusion Model for Manifold Data in High-Dimensional Spacescited by 8
- 2025Anomaly Detection and Generation with Diffusion Models: A Surveycited by 8
- 2025When and how can inexact generative models still sample from the data manifold?cited by 4
- 2025Optimal Stopping in Latent Diffusion Modelscited by 1
- 2026Intrinsic Wasserstein Rates for Score-Based Generative Models on Smooth Manifoldscited by 1
- 2024Inconsistencies In Consistency Models: Better ODE Solving Does Not Imply Better Samplescited by 0
12 most cited of 15 All 15 (JSON)
- 2026One Inverse Step is a Convex Program: Bayes-Limit Calibration of Diffusion Inversion24 Aug 2026
- 2026Intrinsic Wasserstein Rates for Score-Based Generative Models on Smooth Manifolds15 May 2026
- 2026ML-based approach to classification and generation of structured light propagation in turbulent media4 Apr 2026
- 2026SSI-DM: Singularity Skipping Inversion of Diffusion Models2 Feb 2026
- 2025Optimal Stopping in Latent Diffusion Models9 Oct 2025
- 2025When Scores Learn Geometry: Rate Separations under the Manifold Hypothesis29 Sep 2025
- 2025When and how can inexact generative models still sample from the data manifold?11 Aug 2025
- 2025Anomaly Detection and Generation with Diffusion Models: A Survey11 Jun 2025
- 2025Improving the Euclidean Diffusion Generation of Manifold Data by Mitigating Score Function Singularity15 May 2025
- 2025Resolving Memorization in Empirical Diffusion Model for Manifold Data in High-Dimensional Spaces5 May 2025
- 2025Memorization and Regularization in Generative Diffusion Models27 Jan 2025
- 2024Inconsistencies In Consistency Models: Better ODE Solving Does Not Imply Better Samples13 Nov 2024
12 newest of 15 All 15 (JSON)
- 2024A Geometric Framework for Understanding Memorization in Generative Modelscited by 40
- 2024A Geometric Explanation of the Likelihood OOD Detection Paradoxcited by 22
- 2024Masked Completion via Structured Diffusion with White-Box Transformerscited by 9
- 2025Pathway to O(√d) Complexity bound under Wasserstein metric of flow-based modelscited by 4
- 2026Preconditioned One-Step Generative Modeling for Bayesian Inverse Problems in Function Spacescited by 1
- 2026Preconditioned One-Step Generative Modeling for Bayesian Inverse Problems in Function Spaces16 Mar 2026
- 2025Pathway to O(√d) Complexity bound under Wasserstein metric of flow-based models7 Dec 2025
- 2024A Geometric Framework for Understanding Memorization in Generative Models31 Oct 2024
- 2024Masked Completion via Structured Diffusion with White-Box Transformers3 Apr 2024
- 2024A Geometric Explanation of the Likelihood OOD Detection Paradox27 Mar 2024
Similar, not cited
Close in wording, no citation either way
- 2023Diffusion Models are Minimax Optimal Distribution Estimators
- 2024Convergence of Diffusion Models Under the Manifold Hypothesis in High-Dimensions
- 2026Diffusion Models Are Statistically Optimal for Learning Low-Dimensional Multi-Modal Distributions
- 2024Non-asymptotic bounds for forward processes in denoising diffusions: Ornstein-Uhlenbeck is hard to beat
- 2023Fast Diffusion Probabilistic Model Sampling through the lens of Backward Error Analysis
- 2023Score-based Diffusion Models in Function Space
- 2023Restoration-Degradation Beyond Linear Diffusions: A Non-Asymptotic Analysis For DDIM-Type Samplers
- 2025Information-Theoretic Proofs for Diffusion Sampling