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