mlsanityData to 14 Oct 2026

Wasserstein Distributionally Robust Optimization: Theory and Applications in Machine Learning

arXiv 1908.08729 ↗First posted 23 Aug 2019stat.ML cs.LG math.OCcited by 170

Not on the map. Why?

Abstract (arXiv)

Many decision problems in science, engineering and economics are affected by uncertain parameters whose distribution is only indirectly observable through samples. The goal of data-driven decision-making is to learn a decision from finitely many training samples that will perform well on unseen test samples. This learning task is difficult even if all training and test samples are drawn from the same distribution -- especially if the dimension of the uncertainty is large relative to the training sample size. Wasserstein distributionally robust optimization seeks data-driven decisions that perform well under the most adverse distribution within a certain Wasserstein distance from a nominal distribution constructed from the training samples. In this tutorial we will argue that this approach has many conceptual and computational benefits. Most prominently, the optimal decisions can often be computed by solving tractable convex optimization problems, and they enjoy rigorous out-of-sample and asymptotic consistency guarantees. We will also show that Wasserstein distributionally robust optimization has interesting ramifications for statistical learning and motivates new approaches for fundamental learning tasks such as classification, regression, maximum likelihood estimation or minimum mean square error estimation, among others.

Key facts. Cited by 170 papers in the map (data through 14 Oct 2026). First posted on arXiv 23 Aug 2019.

Builds on

No reference list yet.

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2021: 26; 2022: 32; 2023: 33; 2024: 30; 2025: 26; 2026: 23.

170 papers.

Sort:

By area:

  1. 2021Unbalanced minibatch Optimal Transport; applications to Domain Adaptationcited by 125
  2. 2023Recent Advances in Optimal Transport for Machine Learningcited by 66
  3. 2021Twice regularized MDPs and the equivalence between robustness and regularizationcited by 47
  4. 2021Distributionally Robust Learningcited by 44
  5. 2023Diagnosing Model Performance Under Distribution Shiftcited by 39
  6. 2021Sinkhorn Distributionally Robust Optimizationcited by 38
  7. 2022A Unified Wasserstein Distributional Robustness Framework for Adversarial Trainingcited by 36
  8. 2022Minimax Regret Optimization for Robust Machine Learning under Distribution Shiftcited by 31
  9. 2022Learning to Solve Routing Problems via Distributionally Robust Optimizationcited by 31
  10. 2021Statistical Analysis of Wasserstein Distributionally Robust Estimatorscited by 27
  11. 2021Noisy Feature Mixupcited by 27
  12. 2023Algorithms for mean-field variational inference via polyhedral optimization in the Wasserstein spacecited by 20
38 more
  1. 2024Adaptive Preference Scaling for Reinforcement Learning with Human Feedbackcited by 20
  2. 2022Counterfactual Plans under Distributional Ambiguitycited by 19
  3. 2022Sobolev Transport: A Scalable Metric for Probability Measures with Graph Metricscited by 18
  4. 2022An improved central limit theorem and fast convergence rates for entropic transportation costscited by 18
  5. 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimizationcited by 18
  6. 2023AdvST: Revisiting Data Augmentations for Single Domain Generalizationcited by 16
  7. 2021Sequential Domain Adaptation by Synthesizing Distributionally Robust Expertscited by 14
  8. 2022Data-Driven Sample Average Approximation with Covariate Informationcited by 14
  9. 2022Distributed Distributionally Robust Optimization with Non-Convex Objectivescited by 14
  10. 2021Algorithmic Bias and Data Bias: Understanding the Relation between Distributionally Robust Optimization and Data Curationcited by 13
  11. 2022Instance-Dependent Generalization Bounds via Optimal Transportcited by 13
  12. 2023Unifying Distributionally Robust Optimization via Optimal Transport Theorycited by 13
  13. 2023Flow-based Distributionally Robust Optimizationcited by 13
  14. 2021Robustifying Conditional Portfolio Decisions via Optimal Transportcited by 12
  15. 2022A Short and General Duality Proof for Wasserstein Distributionally Robust Optimizationcited by 12
  16. 2022An Optimal Transport Approach to Personalized Federated Learningcited by 12
  17. 2023Distributionally Robust Recourse Actioncited by 12
  18. 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coveragecited by 12
  19. 2022Global-Local Regularization Via Distributional Robustnesscited by 11
  20. 2021Wasserstein Robust Classification with Fairness Constraintscited by 10
  21. 2023Wasserstein Distributionally Robust Policy Evaluation and Learning for Contextual Banditscited by 10
  22. 2024Scoring Rules and Calibration for Imprecise Probabilitiescited by 10
  23. 2021Generalization Bounds with Minimal Dependency on Hypothesis Class via Distributionally Robust Optimizationcited by 9
  24. 2022Distributionally Robust Fair Principal Components via Geodesic Descentscited by 9
  25. 2022On the Generalization of Wasserstein Robust Federated Learningcited by 9
  26. 2023Exact Generalization Guarantees for (Regularized) Wasserstein Distributionally Robust Modelscited by 9
  27. 2021On the regularized risk of distributionally robust learning over deep neural networkscited by 8
  28. 2022Trust Region Policy Optimization with Optimal Transport Discrepancies: Duality and Algorithm for Continuous Actionscited by 8
  29. 2025Prompt Disentanglement via Language Guidance and Representation Alignment for Domain Generalizationcited by 8
  30. 2025Mitigating Spurious Correlation via Distributionally Robust Learning with Hierarchical Ambiguity Setscited by 8
  31. 2021Learning and Decision-Making with Data: Optimal Formulations and Phase Transitionscited by 7
  32. 2022Certifying Out-of-Domain Generalization for Blackbox Functionscited by 7
  33. 2022Group Distributionally Robust Reinforcement Learning with Hierarchical Latent Variablescited by 7
  34. 2022Stochastic Optimization for Spectral Risk Measurescited by 7
  35. 2023Smoothed f-Divergence Distributionally Robust Optimizationcited by 7
  36. 2024Sampling from the Mean-Field Stationary Distributioncited by 7
  37. 2024On the Benefits of Over-parameterization for Out-of-Distribution Generalizationcited by 7
  38. 2021Robust Generalization despite Distribution Shift via Minimum Discriminating Informationcited by 6

50 most cited of 170 All 170 (JSON)