arXiv Machine Learning

A Theory of Speciation in Generative Diffusion Models on Compact Riemannian Manifolds

arXiv Machine Learning
2d ago

RW-Flow: One-Step Generation on Compact Manifolds via Wasserstein Gradient Flows

RW-Flow presents a new one‑step generative framework for data on compact manifolds, leveraging Wasserstein gradient flows. The authors derive a necessary and sufficient identifiability condition for velocity fields on compact, connected Riemannian manifolds, showing that a symmetric, Lipschitz‑continuous cost function yields identifiability iff its Gibbs kernel is nondegenerate. Experiments on geospatial events, protein and RNA torsion angles, and discretized manifolds demonstrate that RW‑Flow surpasses existing one‑step methods across most benchmark settings.

By Ualibyek Nurgulan, Seungwoo Yoo, Prin Phunyaphibarn, Minhyuk Sung
arXiv Machine Learning
Jul 14

Riemannian Denoising Diffusion Probabilistic Models

arXiv:2505. 04338v3 Announce Type: replace Abstract: We propose Riemannian Denoising Diffusion Probabilistic Models (RDDPMs) for learning distributions on submanifolds of Euclidean space that are level sets of functions, including most of the manifolds relevant to applications.

By Zichen Liu, Wei Zhang, Christof Sch\"utte, Tiejun Li
arXiv Machine Learning
Sep 21

Riemannian Neural Hamiltonian Flows: Geodesic Symplectic Transport and Interpretability

The paper introduces Riemannian Neural Hamiltonian Flows, a generative model that extends Hamiltonian normalizing flows to Riemannian manifolds by combining a fixed kinetic energy, a learned scalar potential, and a geodesic leapfrog integrator. It provides an analysis showing how the learned Hamiltonian can be interpreted through an implicit profile and a matched potential, with special cases such as isotropic Gaussian and local harmonic analysis around modes. Experiments on Euclidean, hyperbolic, and spherical spaces demonstrate competitive sample quality and computational efficiency compared to a Riemannian continuous normalizing flow, while confirming the interpretability of the learned potential.

By Vincent Souveton
arXiv Machine Learning
Aug 10

Convergence of Diffusion Models Under the Manifold Hypothesis in High-Dimensions

arXiv:2409. 18804v3 Announce Type: replace-cross Abstract: Denoising Diffusion Probabilistic Models (DDPM) are powerful state-of-the-art methods used to generate synthetic data from high-dimensional data distributions and are widely used for image, audio, and video generation as well as many more applications in science and beyond.

By Iskander Azangulov, George Deligiannidis, Judith Rousseau