Wasserstein Distributionally Robust Optimization: Theory and Applications in Machine Learning
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
Citing papers per year: 2021: 26; 2022: 32; 2023: 33; 2024: 30; 2025: 26; 2026: 23.
170 papers.
Sort:
By area:
- 2021Unbalanced minibatch Optimal Transport; applications to Domain Adaptationcited by 125
- 2023Recent Advances in Optimal Transport for Machine Learningcited by 66
- 2021Twice regularized MDPs and the equivalence between robustness and regularizationcited by 47
- 2021Distributionally Robust Learningcited by 44
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 39
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2022A Unified Wasserstein Distributional Robustness Framework for Adversarial Trainingcited by 36
- 2022Minimax Regret Optimization for Robust Machine Learning under Distribution Shiftcited by 31
- 2022Learning to Solve Routing Problems via Distributionally Robust Optimizationcited by 31
- 2021Statistical Analysis of Wasserstein Distributionally Robust Estimatorscited by 27
- 2021Noisy Feature Mixupcited by 27
- 2023Algorithms for mean-field variational inference via polyhedral optimization in the Wasserstein spacecited by 20
38 more
- 2024Adaptive Preference Scaling for Reinforcement Learning with Human Feedbackcited by 20
- 2022Counterfactual Plans under Distributional Ambiguitycited by 19
- 2022Sobolev Transport: A Scalable Metric for Probability Measures with Graph Metricscited by 18
- 2022An improved central limit theorem and fast convergence rates for entropic transportation costscited by 18
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimizationcited by 18
- 2023AdvST: Revisiting Data Augmentations for Single Domain Generalizationcited by 16
- 2021Sequential Domain Adaptation by Synthesizing Distributionally Robust Expertscited by 14
- 2022Data-Driven Sample Average Approximation with Covariate Informationcited by 14
- 2022Distributed Distributionally Robust Optimization with Non-Convex Objectivescited by 14
- 2021Algorithmic Bias and Data Bias: Understanding the Relation between Distributionally Robust Optimization and Data Curationcited by 13
- 2022Instance-Dependent Generalization Bounds via Optimal Transportcited by 13
- 2023Unifying Distributionally Robust Optimization via Optimal Transport Theorycited by 13
- 2023Flow-based Distributionally Robust Optimizationcited by 13
- 2021Robustifying Conditional Portfolio Decisions via Optimal Transportcited by 12
- 2022A Short and General Duality Proof for Wasserstein Distributionally Robust Optimizationcited by 12
- 2022An Optimal Transport Approach to Personalized Federated Learningcited by 12
- 2023Distributionally Robust Recourse Actioncited by 12
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coveragecited by 12
- 2022Global-Local Regularization Via Distributional Robustnesscited by 11
- 2021Wasserstein Robust Classification with Fairness Constraintscited by 10
- 2023Wasserstein Distributionally Robust Policy Evaluation and Learning for Contextual Banditscited by 10
- 2024Scoring Rules and Calibration for Imprecise Probabilitiescited by 10
- 2021Generalization Bounds with Minimal Dependency on Hypothesis Class via Distributionally Robust Optimizationcited by 9
- 2022Distributionally Robust Fair Principal Components via Geodesic Descentscited by 9
- 2022On the Generalization of Wasserstein Robust Federated Learningcited by 9
- 2023Exact Generalization Guarantees for (Regularized) Wasserstein Distributionally Robust Modelscited by 9
- 2021On the regularized risk of distributionally robust learning over deep neural networkscited by 8
- 2022Trust Region Policy Optimization with Optimal Transport Discrepancies: Duality and Algorithm for Continuous Actionscited by 8
- 2025Prompt Disentanglement via Language Guidance and Representation Alignment for Domain Generalizationcited by 8
- 2025Mitigating Spurious Correlation via Distributionally Robust Learning with Hierarchical Ambiguity Setscited by 8
- 2021Learning and Decision-Making with Data: Optimal Formulations and Phase Transitionscited by 7
- 2022Certifying Out-of-Domain Generalization for Blackbox Functionscited by 7
- 2022Group Distributionally Robust Reinforcement Learning with Hierarchical Latent Variablescited by 7
- 2022Stochastic Optimization for Spectral Risk Measurescited by 7
- 2023Smoothed f-Divergence Distributionally Robust Optimizationcited by 7
- 2024Sampling from the Mean-Field Stationary Distributioncited by 7
- 2024On the Benefits of Over-parameterization for Out-of-Distribution Generalizationcited by 7
- 2021Robust Generalization despite Distribution Shift via Minimum Discriminating Informationcited by 6
50 most cited of 170 All 170 (JSON)
- 2026Distributionally Robust Survival Models under Subpopulation Shift and Outlier Contamination2 Oct 2026
- 2026Optimal Transport Reweighting for Robust Learning under Spurious Correlations and Label Noise1 Oct 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026Distributionally Robust Federated Learning with Multi-Source Data17 Sep 2026
- 2026Statistical Inference for Adversarial Training: Central Limit Theorems via Optimal Transport4 Sep 2026
- 2026Multi-Source Wasserstein Distributionally Robust Graph Learning20 Aug 2026
- 2026Robust Risk Under Evolving Uncertainty: A Wasserstein Counterpart of the Entropic Value-at-Risk19 Aug 2026
- 2026Learning Predictive Ambiguity Sets for Decision-Focused Distributionally Robust Optimization10 Jul 2026
- 2026PeTeR: Post-Training Robustification of Probabilistic Circuits8 Jul 2026
- 2026Generative Robust Optimisation21 Jun 2026
- 2026Unification and Optimization of Robust Supervised Learning27 May 2026
- 2026Convergence of empirical subgradients for optimal transport-based objectives27 May 2026
38 more
- 2026Regret Equals Covariance: A Closed-Form Characterization for Stochastic Optimization13 May 2026
- 2026Beyond the Independence Assumption: Finite-Sample Guarantees for Deep Q-Learning under τ-Mixing7 May 2026
- 2026Wasserstein Distributionally Robust Risk-Sensitive Estimation via Conditional Value-at-Risk20 Apr 2026
- 2026Distributionally Robust K-Means Clustering13 Apr 2026
- 2026Distributionally Robust PAC-Bayesian Control12 Apr 2026
- 2026Generative models for decision-making under distributional shift6 Apr 2026
- 2026Oracle-Robust Online Alignment for Large Language Models24 Feb 2026
- 2026Distributionally Robust Optimization via Generative Ambiguity Modeling9 Feb 2026
- 2026Robust Policy Optimization to Prevent Catastrophic Forgetting9 Feb 2026
- 2026Efficient Distribution Learning with Error Bounds in Wasserstein Distance8 Feb 2026
- 2026Semantic-aware Wasserstein Policy Regularization for Large Language Model Alignment2 Feb 2026
- 2025Assured autonomy: How operations research powers and orchestrates generative AI systems30 Dec 2025
- 2025Natural Geometry of Robust Data Attribution: From Convex Models to Deep Networks9 Dec 2025
- 2025A Novel Wasserstein Quaternion Generative Adversarial Network for Color Image Generation9 Dec 2025
- 2025RoME: Domain-Robust Mixture-of-Experts for MILP Solution Prediction across Domains4 Nov 2025
- 2025Robust Decision Making with Partially Calibrated Forecasts27 Oct 2025
- 2025Distributionally Robust Optimization via Diffusion Ambiguity Modeling26 Oct 2025
- 2025Tight Robustness Certificates and Wasserstein Distributional Attacks for Deep Neural Networks11 Oct 2025
- 2025Distributionally Robust Causal Abstractions6 Oct 2025
- 2025Optimal Regularization Under Uncertainty: Distributional Robustness and Convexity Constraints3 Oct 2025
- 2025Mitigating Spurious Correlation via Distributionally Robust Learning with Hierarchical Ambiguity Sets3 Oct 2025
- 2025Predicting Effects, Missing Distributions: Evaluating LLMs as Human Behavior Simulators in Operations Management30 Sep 2025
- 2025Multi-Agent Inverse Reinforcement Learning for Identifying Pareto-Efficient Coordination -- A Distributionally Robust Approach10 Sep 2025
- 2025Group Distributionally Robust Machine Learning under Group Level Distributional Uncertainty10 Sep 2025
- 2025Distributional Adversarial Attacks and Training in Deep Hedging20 Aug 2025
- 2025Prompt Disentanglement via Language Guidance and Representation Alignment for Domain Generalization3 Jul 2025
- 2025Unregularized limit of stochastic gradient method for Wasserstein distributionally robust optimization5 Jun 2025
- 2025Distributionally Robust Wireless Semantic Communication with Large AI Models28 May 2025
- 2025Causality-Inspired Robustness for Nonlinear Models via Representation Learning19 May 2025
- 2025Decision Making under Model Misspecification: DRO with Robust Bayesian Ambiguity Sets6 May 2025
- 2025Temporal Robustness in Discrete Time Linear Dynamical Systems5 May 2025
- 2025Wasserstein Distributionally Robust Regret Optimization15 Apr 2025
- 2025Provable Robust Overfitting Mitigation in Wasserstein Distributionally Robust Optimization6 Mar 2025
- 2025Enhancing Distributional Robustness in Principal Component Analysis by Wasserstein Distances4 Mar 2025
- 2025Flow-based generative models as iterative algorithms in probability space19 Feb 2025
- 2025Achievable distributional robustness when the robust risk is only partially identified4 Feb 2025
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimization4 Feb 2025
- 2024Decision Making under the Exponential Family: Distributionally Robust Optimisation with Bayesian Ambiguity Sets25 Nov 2024
50 newest of 170 All 170 (JSON)
- 2024Adaptive Preference Scaling for Reinforcement Learning with Human Feedbackcited by 20
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimizationcited by 18
- 2026Robust Policy Optimization to Prevent Catastrophic Forgettingcited by 3
- 2026Semantic-aware Wasserstein Policy Regularization for Large Language Model Alignmentcited by 2
- 2026Oracle-Robust Online Alignment for Large Language Modelscited by 1
- 2026Oracle-Robust Online Alignment for Large Language Models24 Feb 2026
- 2026Robust Policy Optimization to Prevent Catastrophic Forgetting9 Feb 2026
- 2026Semantic-aware Wasserstein Policy Regularization for Large Language Model Alignment2 Feb 2026
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimization4 Feb 2025
- 2024Adaptive Preference Scaling for Reinforcement Learning with Human Feedback4 Jun 2024
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coveragecited by 12
- 2022On Practical Reinforcement Learning: Provable Robustness, Scalability, and Statistical Efficiencycited by 0
- 2025Multi-Agent Inverse Reinforcement Learning for Identifying Pareto-Efficient Coordination -- A Distributionally Robust Approachcited by 0
- 2025Multi-Agent Inverse Reinforcement Learning for Identifying Pareto-Efficient Coordination -- A Distributionally Robust Approach10 Sep 2025
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coverage27 Oct 2023
- 2022On Practical Reinforcement Learning: Provable Robustness, Scalability, and Statistical Efficiency3 Mar 2022
- 2021Unbalanced minibatch Optimal Transport; applications to Domain Adaptationcited by 125
- 2023Recent Advances in Optimal Transport for Machine Learningcited by 66
- 2021Twice regularized MDPs and the equivalence between robustness and regularizationcited by 47
- 2021Distributionally Robust Learningcited by 44
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 39
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2022A Unified Wasserstein Distributional Robustness Framework for Adversarial Trainingcited by 36
- 2022Minimax Regret Optimization for Robust Machine Learning under Distribution Shiftcited by 31
- 2022Learning to Solve Routing Problems via Distributionally Robust Optimizationcited by 31
- 2021Statistical Analysis of Wasserstein Distributionally Robust Estimatorscited by 27
- 2021Noisy Feature Mixupcited by 27
- 2023Algorithms for mean-field variational inference via polyhedral optimization in the Wasserstein spacecited by 20
12 most cited of 157 All 157 (JSON)
- 2026Distributionally Robust Survival Models under Subpopulation Shift and Outlier Contamination2 Oct 2026
- 2026Optimal Transport Reweighting for Robust Learning under Spurious Correlations and Label Noise1 Oct 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026Distributionally Robust Federated Learning with Multi-Source Data17 Sep 2026
- 2026Statistical Inference for Adversarial Training: Central Limit Theorems via Optimal Transport4 Sep 2026
- 2026Multi-Source Wasserstein Distributionally Robust Graph Learning20 Aug 2026
- 2026Robust Risk Under Evolving Uncertainty: A Wasserstein Counterpart of the Entropic Value-at-Risk19 Aug 2026
- 2026Learning Predictive Ambiguity Sets for Decision-Focused Distributionally Robust Optimization10 Jul 2026
- 2026PeTeR: Post-Training Robustification of Probabilistic Circuits8 Jul 2026
- 2026Generative Robust Optimisation21 Jun 2026
- 2026Unification and Optimization of Robust Supervised Learning27 May 2026
- 2026Convergence of empirical subgradients for optimal transport-based objectives27 May 2026
12 newest of 157 All 157 (JSON)