Heteroscedasticity of Denoising Score Matching with Generalised Smooth Noise
Not on the map. Why?
Abstract (arXiv)
Score Matching (SM) is a powerful framework for estimating the log-density derivatives of a distribution without calculating its normalizing constants. This capability has made it a cornerstone across multiple domains, from classical sta- tistical estimation and energy-based models to modern diffusion-based generative models. In practice, these models rely almost exclusively on Denoising Score Matching (DSM) as a tractable proxy for score matching. This ubiquity naturally raises a fundamental question: Is DSM truly "score matching for free"? In this work, we demonstrate that DSM is not a perfect substitute. We prove that the denoising objective is inherently heteroscedastic, the variance of model parame- ters fluctuates unpredictably based on both noise levels and the underlying data geometry. This instability is baked into the mathematical structure of the DSM. To address this, we derive an ideal weighting function that equalizes this variance, yielding a homoscedastic generalization of DSM. Since the ideal weights are of- ten empirically inaccessible, we show that a practical approximation weighting function via Taylor expansion reduces gradient variance during training, at the cost of statistical optimality. Notably, this provides a theoretical justification for an existing heuristic weight used in Isotropic Gaussian Diffusion. We validate our theory across different perturbed distributions and for higher-order scores.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 19 of its 28 references that are in the map. First posted on arXiv 3 Aug 2025.
Builds on
19 of 28 references.
References · 19
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
- 2021Variational Diffusion Modelscited by 1,153
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 913
- 2021Maximum Likelihood Training of Score-Based Diffusion Modelscited by 672
- 2023Understanding Diffusion Objectives as the ELBO with Simple Data Augmentationcited by 304
- 2021How to Train Your Energy-Based Modelscited by 274
- 2023Efficient Diffusion Training via Min-SNR Weighting Strategycited by 246
- 2022GENIE: Higher-Order Denoising Diffusion Solverscited by 127
- 2022Maximum Likelihood Training for Score-Based Diffusion ODEs by High-Order Denoising Score Matchingcited by 80
7 more
- 2021Estimating High Order Gradients of the Data Distribution by Denoisingcited by 64
- 2023Stable Target Field for Reduced Variance Score Estimation in Diffusion Modelscited by 44
- 2024Heavy-Tailed Diffusion Modelscited by 40
- 2023On the Posterior Distribution in Denoising: Application to Uncertainty Quantificationcited by 33
- 2021Heavy-tailed denoising score matchingcited by 18
- 2020One Size Fits All: Can We Train One Denoiser for All Noise Levels?cited by 10
- 2025Loss Functions in Diffusion Models: A Comparative Studycited by 3
Built on by
Citing papers per year: 2025: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2025Denoising Score Matching with Random Features: Insights on Diffusion Models from Precise Learning Curves
- 2025Implicit score matching meets denoising score matching: improved rates of convergence and log-density Hessian estimation
- 2022From Denoising Diffusions to Denoising Markov Models
- 2026Denoising Score Matching in Diffusion Models: The Effect of Rank Constraints on Generalization
- 2025DDPM Score Matching and Distribution Learning
- 2024What's the score? Automated Denoising Score Matching for Nonlinear Diffusions
- 2026The Loss Floor of Denoising Score Matching: Fisher Geometry from Schr"odinger Bridges
- 2025Generalization through variance: how noise shapes inductive biases in diffusion models