arXiv Machine Learning

When Riemann flows with Wasserstein: Generative Modeling of Probability Distributions on Manifolds

arXiv AI
Jun 24

The Geometry Behind Diffusion and Flow Matching: Gradient Flows and Geodesics in Wasserstein Space

arXiv:2606. 24157v1 Announce Type: new Abstract: The space $\mathcal{P}_2(\mathbb{R}^d$) of probability measures with finite second moment carries a natural geometry: the quadratic Wasserstein distance W_2 makes it a complete metric space and, following Otto, a (formal) Riemannian manifold whose geodesics are the optimal-transport interpolations.

By Yian Yao, Weiwei Zhang
arXiv Machine Learning
Jun 30

Robustness and Structure Preservation in Flow-Based Generative Models via Wasserstein Path-Space Divergences

arXiv:2410. 01244v2 Announce Type: replace-cross Abstract: We introduce a novel Wasserstein-1 ($W_1$) path-space divergence for stochastic and deterministic dynamics and establish a Wasserstein Uncertainty Propagation (WUP) theorem that bounds the $W_1$ distance between terminal distributions by the proposed divergence, equivalently characterized by a weighted $L^2$ discrepancy between the underlying drifts and the $W_1$ distance between their initial measures.

By Ziyu Chen, Markos A. Katsoulakis, Benjamin J. Zhang
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 AI
Sep 3

Schr\"odinger Bridges on Lie Group Manifolds for Probabilistic Intrinsic Generation

The paper introduces a probabilistic generative framework called Schr"odinger Bridges on Lie Group Manifolds, enabling direct modeling of non‑Euclidean data without flattening or coordinate inconsistencies. It develops two computational realizations—Wrapped‑Kernel Bridge Calibration for compact Abelian groups and Reciprocal Conditional‑Control Bridge Matching for compact non‑Abelian groups—while providing a modular error bound that separates various sources of approximation error. Experiments on protein, RNA torsions, SO(3), U(n), and protein conformational pathways demonstrate the method’s feasibility and consistency.

By Shizhe Zhang, Mingyang Zhao, Lei Ma