Optimal transport for single-cell and spatial omics

C Bunne, G Schiebinger, A Krause, A Regev… - Nature Reviews …, 2024 - nature.com
High-throughput single-cell profiling provides an unprecedented ability to uncover the
molecular states of millions of cells. These technologies are, however, destructive to cells …

Multisample flow matching: Straightening flows with minibatch couplings

AA Pooladian, H Ben-Hamu, C Domingo-Enrich… - arxiv preprint arxiv …, 2023 - arxiv.org
Simulation-free methods for training continuous-time generative models construct probability
paths that go between noise distributions and individual data samples. Recent works, such …

Learning single-cell perturbation responses using neural optimal transport

C Bunne, SG Stark, G Gut, JS Del Castillo… - Nature …, 2023 - nature.com
Understanding and predicting molecular responses in single cells upon chemical, genetic or
mechanical perturbations is a core question in biology. Obtaining single-cell measurements …

Action matching: Learning stochastic dynamics from samples

K Neklyudov, R Brekelmans… - … on machine learning, 2023 - proceedings.mlr.press
Learning the continuous dynamics of a system from snapshots of its temporal marginals is a
problem which appears throughout natural sciences and machine learning, including in …

Minimax estimation of discontinuous optimal transport maps: The semi-discrete case

AA Pooladian, V Divol… - … Conference on Machine …, 2023 - proceedings.mlr.press
We consider the problem of estimating the optimal transport map between two probability
distributions, $ P $ and $ Q $ in $\mathbb {R}^ d $, on the basis of iid samples. All existing …

Generative modeling through the semi-dual formulation of unbalanced optimal transport

J Choi, J Choi, M Kang - Advances in Neural Information …, 2024 - proceedings.neurips.cc
Optimal Transport (OT) problem investigates a transport map that bridges two distributions
while minimizing a given cost function. In this regard, OT between tractable prior distribution …

On amortizing convex conjugates for optimal transport

B Amos - arxiv preprint arxiv:2210.12153, 2022 - arxiv.org
This paper focuses on computing the convex conjugate operation that arises when solving
Euclidean Wasserstein-2 optimal transport problems. This conjugation, which is also …

Neural optimal transport with lagrangian costs

AA Pooladian, C Domingo-Enrich, RTQ Chen… - arxiv preprint arxiv …, 2024 - arxiv.org
We investigate the optimal transport problem between probability measures when the
underlying cost function is understood to satisfy a least action principle, also known as a …

Extremal domain translation with neural optimal transport

M Gazdieva, A Korotin… - Advances in Neural …, 2024 - proceedings.neurips.cc
In many unpaired image domain translation problems, eg, style transfer or super-resolution,
it is important to keep the translated image similar to its respective input image. We propose …

Unbalanced Diffusion Schr\" odinger Bridge

M Pariset, YP Hsieh, C Bunne, A Krause… - arxiv preprint arxiv …, 2023 - arxiv.org
Schr\" odinger bridges (SBs) provide an elegant framework for modeling the temporal
evolution of populations in physical, chemical, or biological systems. Such natural processes …