English

Rao-Blackwellized Score Matching on Manifolds

Machine Learning 2026-05-28 v2 Machine Learning

Abstract

We study denoising score matching (DSM) when the latent distribution is supported on a smooth embedded manifold MRDM \subset \mathbb{R}^D. Under ambient Gaussian corruption, the tangent denoising target contains a singular normal-fiber noise channel whose variance diverges as d/σ2d/\sigma^2 as σ0+\sigma \to 0^+. We show that conditioning on the nearest-point projection π(X)\pi(X) canonically removes this singularity: the resulting conditional expectation is the unique L2L^2-optimal Rao-Blackwellized predictor of the tangent DSM target among all estimators depending only on the projected observation π(X)\pi(X). We then compute the small-noise expansion of this canonical target and show that it equals the intrinsic Riemannian score up to an explicit order-σ2\sigma^2 correction that decomposes into an intrinsic Tweedie term and an extrinsic curvature term involving the Weingarten and Ricci operators. In the flat case, the construction reduces exactly to ordinary lower-dimensional Gaussian DSM, while on SdS^d the extrinsic correction simplifies to the scalar factor (1d/2)Mlogq(1-d/2)\nabla_M \log q; this extrinsic σ2\sigma^2 correction cancels identically on S2S^2, though the intrinsic Tweedie term remains.

Keywords

Cite

@article{arxiv.2605.25567,
  title  = {Rao-Blackwellized Score Matching on Manifolds},
  author = {Divit Rawal},
  journal= {arXiv preprint arXiv:2605.25567},
  year   = {2026}
}

Comments

22 pages, 3 figures; SPIGM @ ICML 2026