Awesome Optimal Transport
A list of awesome papers and cool resources on optimal transport (OT) and its applications in general! As you will notice, this list is currently mostly focused on optimal transport for machine learning topics.
Table of Contents
Tutorials and Blogs
Libraries
- POT: Python Optimal Transport (Python Optimal Transport library)
- Geomloss (Pytorch library of regularized OT loss variants) | GitHub Repo
Books
Papers
Papers are ordered by theme and inside each theme by publication date (accepted papers only).
Fast approximation Optimal Transport
Fast Optimal Transport problems
Approximating Optimal Transport
Fast approximation of Optimal Transport problems
- Learning with minibatch Wasserstein : asymptotic and gradient properties (AISTATS 2020) | Code
- Hierarchical Optimal Transport for Multimodal Distribution Alignment (NeurIPS 2019)
- Optimal Transport: Fast Probabilistic Approximation with Exact Solvers (JMLR 2019)
- Multiscale Strategies for Computing Optimal Transport (JMLR 2017)
Regularized Optimal Transport
Regularized Optimal Transport and solvers
- Stochastic Optimization for Regularized Wasserstein Estimators (ICML 2020)
- Regularized Optimal Transport is Ground Cost Adversarial (ICML 2020)
- Interpolating between Optimal Transport and MMD using Sinkhorn Divergences (AISTATS 2019) | Code
- Differential Properties of Sinkhorn Approximation for Learning with Wasserstein Distance (NeurIPS 2018)
- Smooth and Sparse Optimal Transport (AISTATS 2018) | Code
- Near-linear time approximation algorithms for optimal transport via Sinkhorn iteration (NeurIPS 2017) | Code
- Stochastic Optimization for Large-scale Optimal Transport (NeurIPS 2016) | Code
- Regularized Discrete Optimal Transport (SIAM Journal on Imaging Sciences 2014)
- Sinkhorn Distances: Lightspeed Computation of Optimal Transport (NeurIPS 2013)
Sliced Optimal Transport
Sliced Optimal Transport (averaged of 1D projections)
- On Projection Robust Optimal Transport: Sample Complexity and Model Misspecification (AISTATS 2021)
- Distributional Sliced-Wasserstein and Applications to Generative Modeling (ICLR 2021)
- Sliced Gromov-Wasserstein (NeurIPS 2019) | Code
- Generalized Sliced Wasserstein Distances (NeurIPS 2019)
- Max-Sliced Wasserstein Distance and Its Use for GANs (CVPR 2019)
- Sliced Wasserstein kernels for probability distributions (CVPR 2016)
Relaxed Optimal Transport
Unbalanced and Partial Optimal Transport
Unbalanced and Partial Optimal Transport variants
- Partial Gromov-Wasserstein with applications on Positive-Unlabeled Learning (NeurIPS 2020) | Code
- On Unbalanced Optimal Transport: An Analysis of Sinkhorn Algorithm (ICML 2020)
- Scaling Algorithms for Unbalanced Transport Problems (Mathematics of Computation 2018)
- Unbalanced Optimal Transport: Dynamic and Kantorovich formulations (Journal of Functional Analysis 2018) | Code
Robust Optimal Transport
Robust Optimal Transport variants
- When OT meets MoM: Robust estimation of Wasserstein Distance (AISTATS 2021)
- Robust Optimal Transport with Applications in Generative Modeling and Domain Adaptation (NeurIPS 2020) | Code
Curse of dimension
Theoretical insights and Optimal Transport variants to tackle the curse of dimension.
- Gaussian-Smoothed Optimal Transport: Metric Structure and Statistical Efficiency (AISTATS 2020)
- Statistical bounds for entropic optimal transport: sample complexity and the central limit theorem (NeurIPS2019)
- Subspace Detours: Building Transport Plans that are Optimal on Subspace Projections (NeurIPS 2019)
- Subspace Robust Wasserstein Distances (ICML 2019) | Code
- Sample Complexity of Sinkhorn Divergences (AISTATS 2019)
- Sharp asymptotic and finite-sample rates of convergence of empirical measures in Wasserstein distance (Bernoulli 2019)
Wasserstein barycenters
Wasserstein barycenters applications
- Improved Complexity Bounds in Wasserstein Barycenter Problem (AISTATS 2021)
- Sinkhorn Barycenters with Free Support via Frank-Wolfe Algorithm (NeurIPS 2019)
- On the Complexity of Approximating Wasserstein Barycenters (ICML 2019)
- Decentralize and Randomize: Faster Algorithm for Wasserstein Barycenters (NeurIPS 2018)
- Stochastic Wasserstein Barycenters (ICML 2018)
- Parallel Streaming Wasserstein Barycenters (NeurIPS 2017)
- Fast Computation of Wasserstein Barycenters (ICML 2014)
- Wasserstein Barycenter and Its Application to Texture Mixing (SSVM 2011)
Generative models
Optimal Transport for Generative Models
- Disentangled Recurrent Wasserstein Autoencoder (ICLR 2021)
- Wasserstein-2 Generative Networks (ICLR 2021)
- Learning Generative Models across Incomparable Spaces (ICML 2019) | Code
- Sliced-Wasserstein Flows: Nonparametric Generative Modeling via Optimal Transport and Diffusions (ICML 2019) | Code
- Sliced Wasserstein Generative Models (CVPR 2019)
- OT-GAN: Improving GANs Using Optimal Transport (ICLR 2018) | Code
- Learning Generative Models with Sinkhorn Divergences (AISTATS 2018)
- WGAN-GP: Improved Training of Wasserstein GANs (NeurIPS 2017)
- WGAN: Wasserstein Generative Adversarial Networks (NeurIPS 2017)
Domain adaptation
Optimal Transport for Domain Adaptation
- Unbalanced minibatch Optimal Transport; applications to Domain Adaptation (ICML 2021)
- Margin-aware Adversarial Domain Adaptation with Optimal Transport (ICML 2020)
- Joint Partial Optimal Transport for Open Set Domain Adaptation (IJCAI 2020)
- Sliced Wasserstein Discrepancy for Unsupervised Domain Adaptation (CVPR 2019)
- Optimal Transport for Multi-source Domain Adaptation under Target Shift (AISTATS 2019)
- Differentially Private Optimal Transport: Application to Domain Adaptation (IJCAI 2019)
- DeepJDOT: Deep Joint Distribution Optimal Transport for Unsupervised Domain Adaptation (ECCV 2018) | Code
- Wasserstein Distance Guided Representation Learning for Domain Adaptation (AAAI 2018) | Code
- Joint distribution optimal transportation for domain adaptation (NeurIPS2017) | Code
- Optimal Transport for Domain Adaptation (TPAMI 2017)
Adversarial robustness
Papers on Adversarial robustness using Wasserstein distance