Scalable Optimal Transport Methods in Machine Learning: A Contemporary Survey
Not on the map. Why?
Abstract (arXiv)
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 questions. The last decade has been a witness to the remarkable contributions of this classical optimization problem to machine learning. This paper is about where and how optimal transport is used in machine learning with a focus on the question of scalable optimal transport. We provide a comprehensive survey of optimal transport while ensuring an accessible presentation as permitted by the nature of the topic and the context. First, we explain the optimal transport background and introduce different flavors (i.e., mathematical formulations), properties, and notable applications. We then address the fundamental question of how to scale optimal transport to cope with the current demands of big and high dimensional data. We conduct a systematic analysis of the methods used in the literature for scaling OT and present the findings in a unified taxonomy. We conclude with presenting some open challenges and discussing potential future research directions. A live repository of related OT research papers is maintained in https://github.com/abdelwahed/OT_for_big_data.git
Key facts. Cited by 27 papers in the map (data through 7 Oct 2026). Builds on 200 of its 317 references that are in the map. First posted on arXiv 8 May 2023.
Builds on
200 of 317 references.
References · 200
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2021Convex optimizationcited by 2,479
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,277
- 2017Improved Training of Wasserstein GANscited by 2,223
- 2019Federated Learning: Challenges, Methods, and Future Directionscited by 1,531
- 2019A Comprehensive Survey on Transfer Learningcited by 1,154
- 2021LoFTR: Detector-Free Local Feature Matching with Transformerscited by 974
38 more
- 2021Generalized Out-of-Distribution Detection: A Surveycited by 796
- 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
- 2017Maximum Classifier Discrepancy for Unsupervised Domain Adaptationcited by 682
- 2021Maximum Likelihood Training of Score-Based Diffusion Modelscited by 670
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 631
- 2019Deep Equilibrium Modelscited by 603
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 594
- 2019Self-labelling via simultaneous clustering and representation learningcited by 525
- 2015Optimal Transport for Domain Adaptationcited by 457
- 2018On the Global Convergence of Gradient Descent for Over-parameterized Models using Optimal Transportcited by 422
- 2016Input Convex Neural Networkscited by 407
- 2019A Survey on Federated Learning Systems: Vision, Hype and Reality for Data Privacy and Protectioncited by 371
- 2017Learning Generative Models with Sinkhorn Divergencescited by 310
- 2020Learning to Generate Novel Domains for Domain Generalizationcited by 296
- 2017Joint Distribution Optimal Transportation for Domain Adaptationcited by 280
- 2018Generalizing Across Domains via Cross-Gradient Trainingcited by 268
- 2017AttGAN: Facial Attribute Editing by Only Changing What You Wantcited by 241
- 2019Sliced Wasserstein Discrepancy for Unsupervised Domain Adaptationcited by 235
- 2017Near-linear time approximation algorithms for optimal transport via Sinkhorn iterationcited by 219
- 2019Model Fusion via Optimal Transportcited by 217
- 2013Fast Computation of Wasserstein Barycenterscited by 216
- 2018DeepJDOT: Deep Joint Distribution Optimal Transport for Unsupervised Domain Adaptationcited by 212
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2019Generalized Sliced Wasserstein Distancescited by 205
- 2019Optimal transport mapping via input convex neural networkscited by 202
- 2021Efficient and Modular Implicit Differentiationcited by 199
- 2016Stochastic Optimization for Large-scale Optimal Transportcited by 188
- 2020LEEP: A New Measure to Evaluate Transferability of Learned Representationscited by 182
- 2021OTA: Optimal Transport Assignment for Object Detectioncited by 177
- 2018Learning Latent Permutations with Gumbel-Sinkhorn Networkscited by 172
- 2018Optimal Transport for structured data with application on graphscited by 170
- 2020Geometric Dataset Distances via Optimal Transportcited by 157
- 2015Optimal Entropy-Transport problems and a new Hellinger-Kantorovich distance between positive measurescited by 138
- 2021Sinkformers: Transformers with Doubly Stochastic Attentioncited by 136
- 2020FLOT: Scene Flow on Point Clouds Guided by Optimal Transportcited by 129
- 2019Gromov-Wasserstein Learning for Graph Matching and Node Embeddingcited by 125
- 2019Transferability and Hardness of Supervised Classification Taskscited by 125
- 2021Unbalanced minibatch Optimal Transport; applications to Domain Adaptationcited by 123
50 most cited of 200 All 200 (JSON)
Built on by
Citing papers per year: 2023: 2; 2024: 3; 2025: 12; 2026: 10.
27 papers.
Sort:
By area:
- 2023Training neural operators to preserve invariant measures of chaotic attractorscited by 43
- 2024AWT: Transferring Vision-Language Models via Augmentation, Weighting, and Transportationcited by 27
- 2023Building the Bridge of Schr"odinger: A Continuous Entropic Optimal Transport Benchmarkcited by 19
- 2025Joint Self-Supervised Video Alignment and Action Segmentationcited by 11
- 2025OTSurv: A Novel Multiple Instance Learning Framework for Survival Prediction with Heterogeneity-aware Optimal Transportcited by 8
- 2024ENOT: Expectile Regularization for Fast and Accurate Training of Neural Optimal Transportcited by 6
- 2025Efficient Transferable Optimal Transport via Min-Sliced Transport Planscited by 4
- 2025Wasserstein Barycenter Gaussian Process based Bayesian Optimizationcited by 2
- 2025Domain-aware Category-level Geometry Learning Segmentation for 3D Point Cloudscited by 2
- 2025Scalable Learning from Probability Measures with Mean Measure Quantizationcited by 1
- 2025SE-Merging: A Self-Enhanced Approach for Dynamic Model Mergingcited by 1
- 2024TPCL: Task Progressive Curriculum Learning for Robust Visual Question Answeringcited by 0
15 more
- 2025Relation-Aware Slicing in Cross-Domain Alignmentcited by 0
- 2025Gromov-Wasserstein and optimal transport: from assignment problems to probabilistic numericcited by 0
- 2025Gradient Flow Convergence Guarantee for General Neural Network Architecturescited by 0
- 2025Designing Ambiguity Sets for Distributionally Robust Optimization Using Structural Causal Optimal Transportcited by 0
- 2025Simplifying Optimal Transport through Schatten-p Regularizationcited by 0
- 2026Accelerated Sinkhorn Algorithms for Partial Optimal Transportcited by 0
- 2026PCA of probability measures: Sparse and Dense sampling regimescited by 0
- 2026Gromov Wasserstein Optimal Transport for Semantic Correspondencescited by 0
- 2026Weighted Wasserstein Barycenter of Gaussian Processes for exotic Bayesian Optimization taskscited by 0
- 2026Move What Matters: Parameter-Efficient Domain Adaptation via Optimal Transport Flow for Collaborative Perceptioncited by 0
- 2026ViCLIP-OT: The First Foundation Vision-Language Model for Vietnamese Image-Text Retrieval with Optimal Transportcited by 0
- 2026Region-Graph Optimal Transport Routing for Mixture-of-Experts Whole-Slide Image Classificationcited by 0
- 2026UG-UMRE: Uncertainty-Guided Modality Augmentation and Distributional Calibration for Unified Multimodal Relation Extractioncited by 0
- 2026SinkSLOT: Sinkhorn via Sparse Lifted Optimal Transportcited by 0
- 2026Optimal transport meets speech: a tutorial reviewcited by 0
- 2026Optimal transport meets speech: a tutorial review24 Sep 2026
- 2026SinkSLOT: Sinkhorn via Sparse Lifted Optimal Transport28 Aug 2026
- 2026UG-UMRE: Uncertainty-Guided Modality Augmentation and Distributional Calibration for Unified Multimodal Relation Extraction5 Aug 2026
- 2026Region-Graph Optimal Transport Routing for Mixture-of-Experts Whole-Slide Image Classification8 Apr 2026
- 2026ViCLIP-OT: The First Foundation Vision-Language Model for Vietnamese Image-Text Retrieval with Optimal Transport26 Feb 2026
- 2026Move What Matters: Parameter-Efficient Domain Adaptation via Optimal Transport Flow for Collaborative Perception12 Feb 2026
- 2026Weighted Wasserstein Barycenter of Gaussian Processes for exotic Bayesian Optimization tasks9 Feb 2026
- 2026Gromov Wasserstein Optimal Transport for Semantic Correspondences3 Feb 2026
- 2026PCA of probability measures: Sparse and Dense sampling regimes2 Feb 2026
- 2026Accelerated Sinkhorn Algorithms for Partial Optimal Transport23 Jan 2026
- 2025Efficient Transferable Optimal Transport via Min-Sliced Transport Plans24 Nov 2025
- 2025Simplifying Optimal Transport through Schatten-p Regularization13 Oct 2025
15 more
- 2025Designing Ambiguity Sets for Distributionally Robust Optimization Using Structural Causal Optimal Transport1 Oct 2025
- 2025Gradient Flow Convergence Guarantee for General Neural Network Architectures28 Sep 2025
- 2025Gromov-Wasserstein and optimal transport: from assignment problems to probabilistic numeric4 Sep 2025
- 2025Domain-aware Category-level Geometry Learning Segmentation for 3D Point Clouds15 Aug 2025
- 2025Relation-Aware Slicing in Cross-Domain Alignment17 Jul 2025
- 2025OTSurv: A Novel Multiple Instance Learning Framework for Survival Prediction with Heterogeneity-aware Optimal Transport25 Jun 2025
- 2025SE-Merging: A Self-Enhanced Approach for Dynamic Model Merging22 Jun 2025
- 2025Wasserstein Barycenter Gaussian Process based Bayesian Optimization18 May 2025
- 2025Joint Self-Supervised Video Alignment and Action Segmentation21 Mar 2025
- 2025Scalable Learning from Probability Measures with Mean Measure Quantization7 Feb 2025
- 2024TPCL: Task Progressive Curriculum Learning for Robust Visual Question Answering26 Nov 2024
- 2024AWT: Transferring Vision-Language Models via Augmentation, Weighting, and Transportation5 Jul 2024
- 2024ENOT: Expectile Regularization for Fast and Accurate Training of Neural Optimal Transport6 Mar 2024
- 2023Building the Bridge of Schr"odinger: A Continuous Entropic Optimal Transport Benchmark16 Jun 2023
- 2023Training neural operators to preserve invariant measures of chaotic attractors1 Jun 2023
- 2023Training neural operators to preserve invariant measures of chaotic attractorscited by 43
- 2023Building the Bridge of Schr"odinger: A Continuous Entropic Optimal Transport Benchmarkcited by 19
- 2025Joint Self-Supervised Video Alignment and Action Segmentationcited by 11
- 2025OTSurv: A Novel Multiple Instance Learning Framework for Survival Prediction with Heterogeneity-aware Optimal Transportcited by 8
- 2024ENOT: Expectile Regularization for Fast and Accurate Training of Neural Optimal Transportcited by 6
- 2025Efficient Transferable Optimal Transport via Min-Sliced Transport Planscited by 4
- 2025Wasserstein Barycenter Gaussian Process based Bayesian Optimizationcited by 2
- 2025Domain-aware Category-level Geometry Learning Segmentation for 3D Point Cloudscited by 2
- 2025Scalable Learning from Probability Measures with Mean Measure Quantizationcited by 1
- 2025SE-Merging: A Self-Enhanced Approach for Dynamic Model Mergingcited by 1
- 2024TPCL: Task Progressive Curriculum Learning for Robust Visual Question Answeringcited by 0
- 2025Relation-Aware Slicing in Cross-Domain Alignmentcited by 0
12 most cited of 23 All 23 (JSON)
- 2026Optimal transport meets speech: a tutorial review24 Sep 2026
- 2026SinkSLOT: Sinkhorn via Sparse Lifted Optimal Transport28 Aug 2026
- 2026UG-UMRE: Uncertainty-Guided Modality Augmentation and Distributional Calibration for Unified Multimodal Relation Extraction5 Aug 2026
- 2026Move What Matters: Parameter-Efficient Domain Adaptation via Optimal Transport Flow for Collaborative Perception12 Feb 2026
- 2026Weighted Wasserstein Barycenter of Gaussian Processes for exotic Bayesian Optimization tasks9 Feb 2026
- 2026PCA of probability measures: Sparse and Dense sampling regimes2 Feb 2026
- 2026Accelerated Sinkhorn Algorithms for Partial Optimal Transport23 Jan 2026
- 2025Efficient Transferable Optimal Transport via Min-Sliced Transport Plans24 Nov 2025
- 2025Simplifying Optimal Transport through Schatten-p Regularization13 Oct 2025
- 2025Designing Ambiguity Sets for Distributionally Robust Optimization Using Structural Causal Optimal Transport1 Oct 2025
- 2025Gradient Flow Convergence Guarantee for General Neural Network Architectures28 Sep 2025
- 2025Gromov-Wasserstein and optimal transport: from assignment problems to probabilistic numeric4 Sep 2025
12 newest of 23 All 23 (JSON)