Score-based generative neural networks for large-scale optimal transport

M Daniels, T Maunu, P Hand - Advances in neural …, 2021 - proceedings.neurips.cc
We consider the fundamental problem of sampling the optimal transport coupling between
given source and target distributions. In certain cases, the optimal transport plan takes the …

Curriculum reinforcement learning using optimal transport via gradual domain adaptation

P Huang, M Xu, J Zhu, L Shi… - Advances in Neural …, 2022 - proceedings.neurips.cc
Abstract Curriculum Reinforcement Learning (CRL) aims to create a sequence of tasks,
starting from easy ones and gradually learning towards difficult tasks. In this work, we focus …

Averaging on the Bures-Wasserstein manifold: dimension-free convergence of gradient descent

J Altschuler, S Chewi, PR Gerber… - Advances in Neural …, 2021 - proceedings.neurips.cc
We study first-order optimization algorithms for computing the barycenter of Gaussian
distributions with respect to the optimal transport metric. Although the objective is …

Scalable computations of wasserstein barycenter via input convex neural networks

J Fan, A Taghvaei, Y Chen - arXiv preprint arXiv:2007.04462, 2020 - arxiv.org
Wasserstein Barycenter is a principled approach to represent the weighted mean of a given
set of probability distributions, utilizing the geometry induced by optimal transport. In this …

Continuous wasserstein-2 barycenter estimation without minimax optimization

A Korotin, L Li, J Solomon, E Burnaev - arXiv preprint arXiv:2102.01752, 2021 - arxiv.org
Wasserstein barycenters provide a geometric notion of the weighted average of probability
measures based on optimal transport. In this paper, we present a scalable algorithm to …

Wasserstein iterative networks for barycenter estimation

A Korotin, V Egiazarian, L Li… - Advances in Neural …, 2022 - proceedings.neurips.cc
Wasserstein barycenters have become popular due to their ability to represent the average
of probability measures in a geometrically meaningful way. In this paper, we present an …

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 …

Meta optimal transport

B Amos, S Cohen, G Luise, I Redko - arXiv preprint arXiv:2206.05262, 2022 - arxiv.org
We study the use of amortized optimization to predict optimal transport (OT) maps from the
input measures, which we call Meta OT. This helps repeatedly solve similar OT problems …

On the benefit of optimal transport for curriculum reinforcement learning

P Klink, C D'Eramo, J Peters… - IEEE Transactions on …, 2024 - ieeexplore.ieee.org
Curriculum reinforcement learning (CRL) allows solving complex tasks by generating a
tailored sequence of learning tasks, starting from easy ones and subsequently increasing …

Scalable Optimal Transport Methods in Machine Learning: A Contemporary Survey

A Khamis, R Tsuchida, M Tarek… - … on Pattern Analysis …, 2024 - ieeexplore.ieee.org
Optimal Transport (OT) is a mathematical framework that first emerged in the eighteenth
century and has led to a plethora of methods for answering many theoretical and applied …