Manifold Aware Denoising Score Matching (MAD)

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Levy-Jurgenson, Alona, Prat, Alvaro, Cuin, James, Teh, Yee Whye
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917308421636096
author Levy-Jurgenson, Alona
Prat, Alvaro
Cuin, James
Teh, Yee Whye
author_facet Levy-Jurgenson, Alona
Prat, Alvaro
Cuin, James
Teh, Yee Whye
contents A major focus in designing methods for learning distributions defined on manifolds is to alleviate the need to implicitly learn the manifold so that learning can concentrate on the data distribution within the manifold. However, accomplishing this often leads to compute-intensive solutions. In this work, we propose a simple modification to denoising score-matching in the ambient space to implicitly account for the manifold, thereby reducing the burden of learning the manifold while maintaining computational efficiency. Specifically, we propose a simple decomposition of the score function into a known component $s^{base}$ and a remainder component $s-s^{base}$ (the learning target), with the former implicitly including information on where the data manifold resides. We derive known components $s^{base}$ in analytical form for several important cases, including distributions over rotation matrices and discrete distributions, and use them to demonstrate the utility of this approach in those cases.
format Preprint
id arxiv_https___arxiv_org_abs_2603_02452
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Manifold Aware Denoising Score Matching (MAD)
Levy-Jurgenson, Alona
Prat, Alvaro
Cuin, James
Teh, Yee Whye
Machine Learning
Artificial Intelligence
A major focus in designing methods for learning distributions defined on manifolds is to alleviate the need to implicitly learn the manifold so that learning can concentrate on the data distribution within the manifold. However, accomplishing this often leads to compute-intensive solutions. In this work, we propose a simple modification to denoising score-matching in the ambient space to implicitly account for the manifold, thereby reducing the burden of learning the manifold while maintaining computational efficiency. Specifically, we propose a simple decomposition of the score function into a known component $s^{base}$ and a remainder component $s-s^{base}$ (the learning target), with the former implicitly including information on where the data manifold resides. We derive known components $s^{base}$ in analytical form for several important cases, including distributions over rotation matrices and discrete distributions, and use them to demonstrate the utility of this approach in those cases.
title Manifold Aware Denoising Score Matching (MAD)
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2603.02452