mlsanityData to 7 Oct 2026

Rao-Blackwellized Score Matching on Manifolds

arXiv 2605.25567 ↗First posted 25 May 2026stat.ML cs.LGcited by 2

Not on the map. Why?

Abstract (arXiv)

We study denoising score matching (DSM) when data are drawn from an embedded manifold M subset ℝD. We show that under ambient Gaussian corruption, the target has variance that diverges as the noise scale decreases and correct for it by regressing against the conditional expectation given the nearest point projection on the manifold: the L2-optimal Rao-Blackwellized target. We then compute the small-noise expansion of this target and show that it recovers the true intrinsic Riemannian score to first order, with a second-order bias from a Tweedie term and two geometric terms dependent on how the manifold is embedded in ambient space: a curvature operator acting on the intrinsic score, and an additive drift generated by the spatial variation of the embedding's second fundamental form. On hyperspheres, we derive a simplified formula and show that both geometric terms vanish exactly on S2, offering a theoretical explanation for why ambient DSM performs comparably to intrinsic methods on real Earth science spherical data in prior work.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 7 of its 18 references that are in the map. First posted on arXiv 25 May 2026.

Builds on

7 of 18 references.

References · 7

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  3. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
  4. 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptionscited by 440
  5. 2022Convergence of denoising diffusion models under the manifold hypothesiscited by 237
  6. 2023Diffusion Models are Minimax Optimal Distribution Estimatorscited by 159
  7. 2025Generalization error bound for denoising score matching under relaxed manifold assumptioncited by 18

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.