arXiv Machine Learning

Light Entropic Optimal Transport on Riemannian Manifolds

Light Entropic Optimal Transport on Riemannian Manifolds introduces ManifoldLightOT, a lightweight method for learning kernel‑induced entropic optimal transport couplings directly on common manifolds such as spheres, tori, SO(3), and SE(3). By constructing geometry‑specific Gibbs kernels and compatible potential parameterizations, the approach yields closed‑form normalization and directly sampleable conditional distributions. The formulation extends naturally to products of manifolds and optimizes potentials from samples using Monte Carlo estimates, demonstrating superior performance over existing manifold OT methods in synthetic and real‑world experiments.

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
Oct 2

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
Jun 18

Riemannian MeanFlow for One-Step Generation on Manifolds

arXiv:2603. 10718v3 Announce Type: replace Abstract: Flow Matching enables simulation-free training of generative models on Riemannian manifolds, yet sampling typically still relies on numerically integrating a probability-flow ODE.

By Zichen Zhong, Haoliang Sun, Yukun Zhao, Yongshun Gong, Yilong Yin
arXiv Machine Learning
Jun 5

Variational Entropic Optimal Transport

arXiv:2602. 02241v2 Announce Type: replace Abstract: Entropic optimal transport (EOT) in continuous spaces with quadratic cost is a classical tool for solving the domain translation problem.

By Roman Dyachenko, Nikita Gushchin, Kirill Sokolov, Petr Mokrov, Evgeny Burnaev, Alexander Korotin
arXiv Machine Learning
Oct 9

Barron Optimal Transport I: Generative Modeling

The paper introduces a new optimal transport framework that uses the Barron energy—an intrinsic neural network complexity measure—instead of the traditional kinetic energy. By modeling transport as deep residual networks that map a reference distribution to a target distribution, the authors define a new metric on probability measures and analyze its properties for generative modeling. They demonstrate that diffusion generative models are suboptimal in this Barron geometry and explore how adaptivity can improve alternative generative approaches.

By Evan Dogariu, Joan Bruna
arXiv AI
Jun 16

Optimal Transport for Machine Learners

arXiv:2505. 06589v2 Announce Type: replace-cross Abstract: Modern machine learning repeatedly manipulates probability measures: empirical datasets, generated samples, latent distributions, class-conditional laws, particle systems, weights of wide networks and attention patterns.

By Gabriel Peyr\'e
arXiv Machine Learning
5d ago

Learning Normal Coordinates on Riemannian Manifolds with Flow Charts

The paper introduces flow charts, a conditional invertible neural network designed to approximate normal coordinates on Riemannian manifolds, enabling fast Euclidean approximations of exponential and logarithmic maps. It presents a novel nonnegative loss function that is zero exactly when the chart is normal, and demonstrates that flow charts match or surpass specialized neural geodesic interpolators across sixteen synthetic and data‑derived geometries. When pretrained, flow charts achieve a median 60–70× speedup over numerical baselines while maintaining comparable logarithmic map estimation error and improving numerical geodesic solver accuracy.

By Edmond Cunningham