arXiv Machine Learning

Schr\"odinger-Bass Bridge Barycenter Models for Measure-Valued Data

The paper introduces Schr"odinger-Bass Bridge (SBB) barycenter models—SBB-encoder, SBB-decoder, SBB-symmetric, and SBB-Markovian—for measure-valued data, leveraging directional asymmetry and dual drift-diffusion control. It proves existence of SBB-barycenters under mild assumptions, shows that for Gaussian inputs the decoder admits a Gaussian minimizer simplifying computation, and develops a hybrid Wasserstein gradient particle descent for general measures. Experiments on spatial images and single-cell RNA sequencing demonstrate that the SBB-decoder and Markovian barycenter outperform existing models in distributional interpolation and inference.

arXiv Machine Learning
Jul 21

Twisted Schr\"odinger Bridge Matching

arXiv:2607. 16987v1 Announce Type: cross Abstract: Over the past few years, diffusion-based Schr\"odinger bridge models have been proposed to approximate optimal transport dynamics between two prescribed boundary distributions, with successful applications to generative modeling.

By Maxence Noble, Marie Scheid, Yazid Janati, Eric Moulines, Alain Durmus
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 Statistics ML
Aug 27

Barycentric Weak Inner-Product Gromov-Wasserstein

The paper introduces a weak Gromov-Wasserstein (wGW) framework that compares source relations with relations between target conditional laws, focusing on inner-product relations and preserving conditional means. It defines the barycentric weak inner-product GW (wIGW) distance, proves existence of minimizers under finite second moments, and presents a ridge-regularized dual formulation leading to an iterative algorithm for finitely supported measures. Experiments on point clouds, graphs, and a PBMC multiome study demonstrate that mean-preserving target refinements can incur zero cost and improve atlas-based cell type transfer.

By Youssef Mroueh
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
Jun 30

Learning from samples: inverse problems over measures

arXiv:2505. 07124v3 Announce Type: replace Abstract: We study inverse problems where an unknown potential is observed only through samples from the measure it induces by a convex variational principle.

By Francisco Andrade, Gabriel Peyr\'e, Clarice Poon
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 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