Rao-Blackwellized Score Matching on Manifolds
Abstract
We study denoising score matching (DSM) when the latent distribution is supported on a smooth embedded manifold . Under ambient Gaussian corruption, the tangent denoising target contains a singular normal-fiber noise channel whose variance diverges as as . We show that conditioning on the nearest-point projection canonically removes this singularity: the resulting conditional expectation is the unique -optimal Rao-Blackwellized predictor of the tangent DSM target among all estimators depending only on the projected observation . 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- 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 the extrinsic correction simplifies to the scalar factor ; this extrinsic correction cancels identically on , though the intrinsic Tweedie term remains.
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