Large-Scale Optimal Transport and Mapping Estimation
Not on the map. Why?
Abstract (arXiv)
This paper presents a novel two-step approach for the fundamental problem of learning an optimal map from one distribution to another. First, we learn an optimal transport (OT) plan, which can be thought as a one-to-many map between the two distributions. To that end, we propose a stochastic dual approach of regularized OT, and show empirically that it scales better than a recent related approach when the amount of samples is very large. Second, we estimate a textitMonge map as a deep neural network learned by approximating the barycentric projection of the previously-obtained OT plan. This parameterization allows generalization of the mapping outside the support of the input measure. We prove two theoretical stability results of regularized OT which show that our estimations converge to the OT plan and Monge map between the underlying continuous measures. We showcase our proposed approach on two applications: domain adaptation and generative modeling.
Key facts. Cited by 120 papers in the map (data through 7 Oct 2026). First posted on arXiv 7 Nov 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 0; 2021: 25; 2022: 18; 2023: 29; 2024: 21; 2025: 18; 2026: 8.
120 papers.
Sort:
By area:
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplingscited by 267
- 2022Neural Optimal Transportcited by 123
- 2021Do Neural Optimal Transport Solvers Work? A Continuous Wasserstein-2 Benchmarkcited by 95
- 2021Entropic estimation of optimal transport mapscited by 87
- 2023Simulation-free Schr"odinger bridges via score and flow matchingcited by 81
- 2021Generative Modeling with Optimal Transport Mapscited by 80
- 2021Large-Scale Wasserstein Gradient Flowscited by 76
- 2024Optimal Flow Matching: Learning Straight Trajectories in Just One Stepcited by 74
- 2023Recent Advances in Optimal Transport for Machine Learningcited by 66
- 2021Minibatch optimal transport distances; analysis and applicationscited by 62
- 2019Wasserstein Adversarial Imitation Learningcited by 59
38 more
- 2021Differentiable Particle Filtering via Entropy-Regularized Optimal Transportcited by 59
- 2021Score-based Generative Neural Networks for Large-Scale Optimal Transportcited by 59
- 2021Variational Wasserstein gradient flowcited by 53
- 2021Rates of Estimation of Optimal Transport Maps using Plug-in Estimators via Barycentric Projectionscited by 52
- 2021Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimizationcited by 50
- 2022Entropic Neural Optimal Transport via Diffusion Processescited by 49
- 2024Light and Optimal Schr"odinger Bridge Matchingcited by 45
- 2021Neural Monge Map estimation and its applicationscited by 40
- 2023Generative Modeling through the Semi-dual Formulation of Unbalanced Optimal Transportcited by 39
- 2022Neural Optimal Transport with General Cost Functionalscited by 34
- 2023Optimal Transport for Treatment Effect Estimationcited by 32
- 2022Kernel Neural Optimal Transportcited by 31
- 2022On amortizing convex conjugates for optimal transportcited by 30
- 2022Meta Optimal Transportcited by 28
- 2023Scalable Optimal Transport Methods in Machine Learning: A Contemporary Surveycited by 27
- 2023GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell Genomicscited by 27
- 2023Energy-guided Entropic Neural Optimal Transportcited by 25
- 2022Wasserstein Iterative Networks for Barycenter Estimationcited by 23
- 2023Minimax estimation of discontinuous optimal transport maps: The semi-discrete casecited by 20
- 2022Kantorovich Strikes Back! Wasserstein GANs are not Optimal Transport?cited by 19
- 2023Building the Bridge of Schr"odinger: A Continuous Entropic Optimal Transport Benchmarkcited by 19
- 2023Light Schr"odinger Bridgecited by 18
- 2021Large-scale optimal transport map estimation using projection pursuitcited by 16
- 2024Neural Optimal Transport with Lagrangian Costscited by 16
- 2024Estimating Barycenters of Distributions with Neural Optimal Transportcited by 14
- 2021Semi-Discrete Optimal Transport: Hardness, Regularization and Numerical Solutioncited by 13
- 2023Causal Optimal Transport of Abstractionscited by 13
- 2021Near-optimal estimation of smooth transport maps with kernel sums-of-squarescited by 12
- 2023Extremal Domain Translation with Neural Optimal Transportcited by 12
- 2023Unifying Distributionally Robust Optimization via Optimal Transport Theorycited by 12
- 2023Energy-Guided Continuous Entropic Barycenter Estimation for General Costscited by 12
- 2024Adversarial Schr"odinger Bridge Matchingcited by 12
- 2023Minimum intrinsic dimension scaling for entropic optimal transportcited by 11
- 2023Optimal Transport-Guided Conditional Score-Based Diffusion Modelscited by 11
- 2021Learning High Dimensional Wasserstein Geodesicscited by 10
- 2024Graph Edit Distance with General Costs Using Neural Set Divergencecited by 10
- 2024Neural Topic Modeling with Large Language Models in the Loopcited by 10
- 2021Fairness with Continuous Optimal Transportcited by 9
50 most cited of 120 All 120 (JSON)
- 2026Gromov-Wasserstein Distillation for Inductive Multi-View Embedding30 Sep 2026
- 2026Deep Barycentric Regression for Optimal Transport Map Estimation and its Statistical Optimality6 Sep 2026
- 2026Quadratically Regularized Optimal Transport: Localization Bounds and Affine Case Analysis23 May 2026
- 2026Implicit Neural Optimal Transport via Fixed-Point Optimization11 May 2026
- 2026Hyperparameter Trajectory Inference with Conditional Lagrangian Optimal Transport2 Mar 2026
- 2026Neural solver for Wasserstein Geodesics and optimal transport dynamics25 Feb 2026
- 2026Fast and Large-Scale Unbalanced Optimal Transport via its Semi-Dual and Adaptive Gradient Methods11 Feb 2026
- 2026Variational Entropic Optimal Transport2 Feb 2026
- 2025Estimation of Stochastic Optimal Transport Maps10 Dec 2025
- 2025Decreasing Entropic Regularization Averaged Gradient for Semi-Discrete Optimal Transport31 Oct 2025
- 2025Exponential Convergence Guarantees for Iterative Markovian Fitting23 Oct 2025
- 2025Expert or not? assessing data quality in offline reinforcement learning14 Oct 2025
38 more
- 2025Neural Hamilton--Jacobi Characteristic Flows for Optimal Transport30 Sep 2025
- 2025Neural Optimal Transport Meets Multivariate Conformal Prediction29 Sep 2025
- 2025Improved Stochastic Optimization of LogSumExp29 Sep 2025
- 2025Incorporating Pre-trained Diffusion Models in Solving the Schr"odinger Bridge Problem25 Aug 2025
- 2025FairPOT: Balancing AUC Performance and Fairness with Proportional Optimal Transport5 Aug 2025
- 2025DPOT: A DeepParticle method for Computation of Optimal Transport with convergence guarantee29 Jun 2025
- 2025Neural Estimation for Scaling Entropic Multimarginal Optimal Transport31 May 2025
- 2025Distances for Markov chains from sample streams23 May 2025
- 2025Unsupervised Learning for Optimal Transport plan prediction between unbalanced graphs21 May 2025
- 2025Fair Clustering via Alignment14 May 2025
- 2025A Physics-preserved Transfer Learning Method for Differential Equations2 May 2025
- 2025Hierarchical Refinement: Optimal Transport to Infinity and Beyond4 Mar 2025
- 2025Score-Based Diffusion Policy Compatible with Reinforcement Learning via Optimal Transport18 Feb 2025
- 2025Joint Learning of Energy-based Models and their Partition Function30 Jan 2025
- 2024Sum-of-Squares Programming for Ma-Trudinger-Wang Regularity of Optimal Transport Maps17 Dec 2024
- 2024Neural Topic Modeling with Large Language Models in the Loop13 Nov 2024
- 2024Linear Spherical Sliced Optimal Transport: A Fast Metric for Comparing Spherical Data9 Nov 2024
- 2024Robust Barycenter Estimation using Semi-Unbalanced Neural Optimal Transport4 Oct 2024
- 2024Graph Edit Distance with General Costs Using Neural Set Divergence26 Sep 2024
- 2024Localized Schr"odinger Bridge Sampler12 Sep 2024
- 2024HOTS3D: Hyper-Spherical Optimal Transport for Semantic Alignment of Text-to-3D Generation19 Jul 2024
- 2024Generative Modeling by Minimizing the Wasserstein-2 Loss19 Jun 2024
- 2024Differentiable Cost-Parameterized Monge Map Estimators12 Jun 2024
- 2024Neural Optimal Transport with Lagrangian Costs1 Jun 2024
- 2024A Differential Equation Approach for Wasserstein GANs and Beyond25 May 2024
- 2024Semi-Discrete Optimal Transport: Nearly Minimax Estimation With Stochastic Gradient Descent and Adaptive Entropic Regularization23 May 2024
- 2024Adversarial Schr"odinger Bridge Matching23 May 2024
- 2024Optimal Aggregation of Prediction Intervals under Unsupervised Domain Shift16 May 2024
- 2024Optimal Flow Matching: Learning Straight Trajectories in Just One Step19 Mar 2024
- 2024Optimal Transport for Domain Adaptation through Gaussian Mixture Models18 Mar 2024
- 2024Estimating Barycenters of Distributions with Neural Optimal Transport6 Feb 2024
- 2024Light and Optimal Schr"odinger Bridge Matching5 Feb 2024
- 2024Continuous Unsupervised Domain Adaptation Using Stabilized Representations and Experience Replay31 Jan 2024
- 2024Accelerating Sinkhorn Algorithm with Sparse Newton Iterations20 Jan 2024
- 2024Stable generative modeling using Schr"odinger bridges9 Jan 2024
- 2023Causal Optimal Transport of Abstractions13 Dec 2023
- 2023Optimal Transport-Guided Conditional Score-Based Diffusion Models2 Nov 2023
- 2023Optimal Transport for Treatment Effect Estimation27 Oct 2023
50 newest of 120 All 120 (JSON)
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplingscited by 267
- 2023Simulation-free Schr"odinger bridges via score and flow matchingcited by 81
- 2024Optimal Flow Matching: Learning Straight Trajectories in Just One Stepcited by 74
- 2024Light and Optimal Schr"odinger Bridge Matchingcited by 45
- 2023GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell Genomicscited by 27
- 2024Adversarial Schr"odinger Bridge Matchingcited by 12
- 2023Optimal Transport-Guided Conditional Score-Based Diffusion Modelscited by 11
- 2025Score-Based Diffusion Policy Compatible with Reinforcement Learning via Optimal Transportcited by 6
- 2025Incorporating Pre-trained Diffusion Models in Solving the Schr"odinger Bridge Problemcited by 1
- 2025Incorporating Pre-trained Diffusion Models in Solving the Schr"odinger Bridge Problem25 Aug 2025
- 2025Score-Based Diffusion Policy Compatible with Reinforcement Learning via Optimal Transport18 Feb 2025
- 2024Adversarial Schr"odinger Bridge Matching23 May 2024
- 2024Optimal Flow Matching: Learning Straight Trajectories in Just One Step19 Mar 2024
- 2024Light and Optimal Schr"odinger Bridge Matching5 Feb 2024
- 2023Optimal Transport-Guided Conditional Score-Based Diffusion Models2 Nov 2023
- 2023GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell Genomics13 Oct 2023
- 2023Simulation-free Schr"odinger bridges via score and flow matching7 Jul 2023
- 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplings28 Apr 2023
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep 2022
- 2019Wasserstein Adversarial Imitation Learningcited by 59
- 2025Expert or not? assessing data quality in offline reinforcement learningcited by 1
- 2025Expert or not? assessing data quality in offline reinforcement learning14 Oct 2025
- 2019Wasserstein Adversarial Imitation Learning19 Jun 2019
- 2022Neural Optimal Transportcited by 123
- 2021Do Neural Optimal Transport Solvers Work? A Continuous Wasserstein-2 Benchmarkcited by 95
- 2021Entropic estimation of optimal transport mapscited by 87
- 2021Generative Modeling with Optimal Transport Mapscited by 80
- 2021Large-Scale Wasserstein Gradient Flowscited by 76
- 2023Recent Advances in Optimal Transport for Machine Learningcited by 66
- 2021Minibatch optimal transport distances; analysis and applicationscited by 62
- 2021Differentiable Particle Filtering via Entropy-Regularized Optimal Transportcited by 59
- 2021Score-based Generative Neural Networks for Large-Scale Optimal Transportcited by 59
- 2021Variational Wasserstein gradient flowcited by 53
- 2021Rates of Estimation of Optimal Transport Maps using Plug-in Estimators via Barycentric Projectionscited by 52
- 2021Continuous Wasserstein-2 Barycenter Estimation without Minimax Optimizationcited by 50
12 most cited of 108 All 108 (JSON)
- 2026Gromov-Wasserstein Distillation for Inductive Multi-View Embedding30 Sep 2026
- 2026Deep Barycentric Regression for Optimal Transport Map Estimation and its Statistical Optimality6 Sep 2026
- 2026Quadratically Regularized Optimal Transport: Localization Bounds and Affine Case Analysis23 May 2026
- 2026Implicit Neural Optimal Transport via Fixed-Point Optimization11 May 2026
- 2026Hyperparameter Trajectory Inference with Conditional Lagrangian Optimal Transport2 Mar 2026
- 2026Neural solver for Wasserstein Geodesics and optimal transport dynamics25 Feb 2026
- 2026Fast and Large-Scale Unbalanced Optimal Transport via its Semi-Dual and Adaptive Gradient Methods11 Feb 2026
- 2026Variational Entropic Optimal Transport2 Feb 2026
- 2025Estimation of Stochastic Optimal Transport Maps10 Dec 2025
- 2025Decreasing Entropic Regularization Averaged Gradient for Semi-Discrete Optimal Transport31 Oct 2025
- 2025Exponential Convergence Guarantees for Iterative Markovian Fitting23 Oct 2025
- 2025Neural Hamilton--Jacobi Characteristic Flows for Optimal Transport30 Sep 2025
12 newest of 108 All 108 (JSON)