When Scores Learn Geometry: Rate Separations Under the Manifold Hypothesis
Abstract
Score-based methods, such as diffusion models and Bayesian inverse problems, are often interpreted as learning the data distribution in the low-noise limit ($\sigma \to 0$). In this work, we propose an alternative perspective: their success arises from implicitly learning the data manifold rather than the full distribution. Our claim is based on a novel analysis of scores in the small-$\sigma$ regime that reveals a sharp separation of scales: information about the data manifold is $\Theta(\sigma^{-2})$ stronger than information about the distribution. We argue that this insight suggests a paradigm shift from the less practical goal of distributional learning to the more attainable task of geometric learning, which provably tolerates $O(\sigma^{-2})$ larger errors in score approximation. We illustrate this perspective through three consequences: i) in diffusion models, concentration on data support can be achieved with a score error of $o(\sigma^{-2})$, whereas recovering the specific data distribution requires a much stricter $o(1)$ error; ii) more surprisingly, learning the uniform distribution on the manifold—an especially structured and useful object—is also $O(\sigma^{-2})$ easier; and iii) in Bayesian inverse problems, the maximum entropy prior is $O(\sigma^{-2})$ more robust to score errors than generic priors. Finally, we validate our theoretical findings with preliminary experiments on large-scale models, including Stable Diffusion.
Cite
Text
Li et al. "When Scores Learn Geometry: Rate Separations Under the Manifold Hypothesis." International Conference on Learning Representations, 2026.Markdown
[Li et al. "When Scores Learn Geometry: Rate Separations Under the Manifold Hypothesis." International Conference on Learning Representations, 2026.](https://mlanthology.org/iclr/2026/li2026iclr-scores/)BibTeX
@inproceedings{li2026iclr-scores,
title = {{When Scores Learn Geometry: Rate Separations Under the Manifold Hypothesis}},
author = {Li, Xiang and Shen, Zebang and Hsieh, Ya-Ping and He, Niao},
booktitle = {International Conference on Learning Representations},
year = {2026},
url = {https://mlanthology.org/iclr/2026/li2026iclr-scores/}
}