SOURCE-LINKED INTELLIGENCE

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

arXiv · AI, language, vision and robotics · article · Sep 22, 2026 · UTC

Many scientific datasets, such as molecular conformational ensembles or single-cell tissue measurements, are naturally modeled as meta-distributions: distributions over probability measures on non-Euclidean domains. Existing generative methods largely assume Euclidean geometry and fail to capture this structure. We introduce Riemannian Wasserstein Entropic Flow Matching (RWEFM), a generative framework on the Wasserstein space $\mathcal{P}_2(\mathcal{M})$ of a Riemannian manifold $(\mathcal{M},g)$. RWEFM is trained by regressing a neural vector field onto Riemannian optimal transport velocities

Read original source ↗ Open in workspace

recordType

paper

region

Global