Data-driven Distributionally Robust Optimization Using the Wasserstein Metric: Performance Guarantees and Tractable Reformulations
Not on the map. Why?
Abstract (arXiv)
We consider stochastic programs where the distribution of the uncertain parameters is only observable through a finite training dataset. Using the Wasserstein metric, we construct a ball in the space of (multivariate and non-discrete) probability distributions centered at the uniform distribution on the training samples, and we seek decisions that perform best in view of the worst-case distribution within this Wasserstein ball. The state-of-the-art methods for solving the resulting distributionally robust optimization problems rely on global optimization techniques, which quickly become computationally excruciating. In this paper we demonstrate that, under mild assumptions, the distributionally robust optimization problems over Wasserstein balls can in fact be reformulated as finite convex programs---in many interesting cases even as tractable linear programs. Leveraging recent measure concentration results, we also show that their solutions enjoy powerful finite-sample performance guarantees. Our theoretical results are exemplified in mean-risk portfolio optimization as well as uncertainty quantification.
Key facts. Cited by 291 papers in the map (data through 14 Oct 2026). First posted on arXiv 19 May 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 52; 2022: 42; 2023: 43; 2024: 48; 2025: 53; 2026: 53.
291 papers.
Sort:
By area:
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2021Unbalanced minibatch Optimal Transport; applications to Domain Adaptationcited by 125
- 2021Edge Artificial Intelligence for 6G: Vision, Enabling Technologies, and Applicationscited by 97
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2021Heterogeneous Risk Minimizationcited by 90
- 2021Distributionally Robust Learningcited by 44
- 2022Can Push-forward Generative Models Fit Multimodal Distributions?cited by 41
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2023A Finite Sample Complexity Bound for Distributionally Robust Q-learningcited by 36
- 2021Doubly-Valid/Doubly-Sharp Sensitivity Analysis for Causal Inference with Unmeasured Confoundingcited by 28
- 2021Statistical Analysis of Wasserstein Distributionally Robust Estimatorscited by 27
- 2021Outlier-Robust Optimal Transport: Duality, Structure, and Statistical Analysiscited by 27
38 more
- 2021Disentangled Recurrent Wasserstein Autoencodercited by 22
- 2021Kernelized Heterogeneous Risk Minimizationcited by 21
- 2023Understanding Contrastive Learning via Distributionally Robust Optimizationcited by 21
- 2023Estimate-Then-Optimize versus Integrated-Estimation-Optimization versus Sample Average Approximation: A Stochastic Dominance Perspectivecited by 20
- 2021Fast Distributionally Robust Learning with Variance Reduced Min-Max Optimizationcited by 19
- 2021Robust Risk-Aware Reinforcement Learningcited by 18
- 2022Holistic Robust Data-Driven Decisionscited by 18
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimizationcited by 18
- 2021Distributionally Robust Policy Learning via Adversarial Environment Generationcited by 17
- 2021Learning Representations that Support Robust Transfer of Predictorscited by 17
- 2022Partial Identification of Treatment Effects with Implicit Generative Modelscited by 17
- 2022Near-Optimal Algorithms for Group Distributionally Robust Optimization and Beyondcited by 17
- 2023Robust Control for Dynamical Systems With Non-Gaussian Noise via Formal Abstractionscited by 16
- 2023Wasserstein distributional robustness of neural networkscited by 16
- 2023Outlier-Robust Wasserstein DROcited by 16
- 2021Smooth p-Wasserstein Distance: Structure, Empirical Approximation, and Statistical Applicationscited by 15
- 2021Unified Group Fairness on Federated Learningcited by 15
- 2024A Short Survey on Importance Weighting for Machine Learningcited by 15
- 2024Score-based generative models are provably robust: an uncertainty quantification perspectivecited by 15
- 2021Sequential Domain Adaptation by Synthesizing Distributionally Robust Expertscited by 14
- 2021Distributionally Robust Learning with Stable Adversarial Trainingcited by 14
- 2021Poisoning Attack against Estimating from Pairwise Comparisonscited by 14
- 2022Distributionally Robust Optimization via Ball Oracle Accelerationcited by 14
- 2022Data-Driven Sample Average Approximation with Covariate Informationcited by 14
- 2022Learning from a Biased Samplecited by 14
- 2022An Exponentially Converging Particle Method for the Mixed Nash Equilibrium of Continuous Gamescited by 14
- 2023Distributionally Robust Optimization with Probabilistic Groupcited by 14
- 2021Semi-Discrete Optimal Transport: Hardness, Regularization and Numerical Solutioncited by 13
- 2021Sampling-Based Robust Control of Autonomous Systems with Non-Gaussian Noisecited 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
- 2021Individually Fair Gradient Boostingcited by 12
- 2021Risk-averse autonomous systems: A brief history and recent developments from the perspective of optimal controlcited by 12
- 2022Treatment Effect Risk: Bounds and Inferencecited by 12
- 2022A Short and General Duality Proof for Wasserstein Distributionally Robust Optimizationcited by 12
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coveragecited by 12
50 most cited of 291 All 291 (JSON)
- 2026Private and Distributionally Robust Federated Multiclass Classification9 Oct 2026
- 2026Largest Rashomon sets of decision trees for robust contextual optimization3 Oct 2026
- 2026Distributionally Robust Survival Models under Subpopulation Shift and Outlier Contamination2 Oct 2026
- 2026Distributionally Robust Schr"odinger Bridge1 Oct 2026
- 2026Parameter-Efficient Distributionally Robust Adaptation of Tabular Foundation Models under Subpopulation Shift1 Oct 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026Learning Risk Scores Robust to Unobserved Confounders22 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
- 2026Certified Safety Radii in Forecast-Error Space for Wasserstein Distributionally Robust Small Signal Stability-Constrained AC Optimal Power Flow via Lifted Spectrahedral Containment31 Aug 2026
- 2026A Unified Perspective on Conformal Prediction and Wasserstein Distributionally Robust Optimization for Uncertainty Quantification30 Aug 2026
- 2026Generative Neural Networks for Sinkhorn Distributionally Robust Hypothesis Testing24 Aug 2026
38 more
- 2026Model-Based Reinforcement Learning for Heterogeneous Multi-Robot Task Assignment Under Distribution Shifts21 Aug 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
- 2026Quantifying Risk Under Evolving Uncertainty: Belief-Dependent Robustness for Safe Sequential Decision Making18 Aug 2026
- 2026Decision-Driven Regularization: A Blended Model for Learning and Optimization15 Aug 2026
- 2026Closing the loop in learning with missing data10 Aug 2026
- 2026Uncovering expert objectives in production planning via inverse optimization: An industrial case study7 Aug 2026
- 2026Generalization Bounds on Optimal Control for Transformer Training and Wasserstein Distributional Robustness30 Jul 2026
- 2026Generative Distributionally Robust Optimization27 Jul 2026
- 2026Robust Metaheuristics under Uncertainty for Berth Allocation and Quay Crane Assignment: A Review6 Jul 2026
- 2026A Distributionally Robust Framework for Learned Reconstructions in Inverse Problems29 Jun 2026
- 2026RECAST: Model Reconstruction via Counterfactual-Aware Wasserstein Geometry under Limited Data26 Jun 2026
- 2026Policy Gradient Learning for Distributionally Robust Markov Decision Processes under Wasserstein Ambiguity25 Jun 2026
- 2026Generative Robust Optimisation21 Jun 2026
- 2026Robust Q-learning for mean-field control under Wasserstein uncertainty in common noise18 Jun 2026
- 2026Conservative neural posterior estimation via distributionally robust training27 May 2026
- 2026Distributionally Robust Transfer Learning with Structurally Missing Covariates, with Application to Cross-National Cardiac Arrest Prediction22 May 2026
- 2026Bounded-Rationality, Hedging, and Generalization14 May 2026
- 2026Regret Equals Covariance: A Closed-Form Characterization for Stochastic Optimization13 May 2026
- 2026Wasserstein Distributionally Robust Regret Optimization for Reinforcement Learning from Human Feedback30 Apr 2026
- 2026Data-driven Reachable Set Estimation with Tunable Adversarial and Wasserstein Distributional Guarantees14 Apr 2026
- 2026Deep Learning for Sequential Decision Making under Uncertainty: Foundations, Frameworks, and Frontiers13 Apr 2026
- 2026Distributionally Robust PAC-Bayesian Control12 Apr 2026
- 2026Generative models for decision-making under distributional shift6 Apr 2026
- 2026Safe Distributionally Robust Feature Selection under Covariate Shift17 Mar 2026
- 2026Evaluating Black-Box Vulnerabilities with Wasserstein-Constrained Data Perturbations16 Mar 2026
- 2026Learning efficient representations of complex constraints for scalable optimization9 Mar 2026
- 2026Oracle-Robust Online Alignment for Large Language Models24 Feb 2026
- 2026Risk-Averse Wasserstein Distributionally Robust Online Learning23 Feb 2026
- 2026Distributionally Robust Cooperative Multi-Agent Reinforcement Learning via Robust Value Factorization11 Feb 2026
- 2026Distributionally Robust Optimization via Generative Ambiguity Modeling9 Feb 2026
- 2026Efficient Distribution Learning with Error Bounds in Wasserstein Distance8 Feb 2026
- 2026Certified Learning under Distribution Shift: Sound Verification and Identifiable Structure6 Feb 2026
- 20263D-Learning: Diffusion-Augmented Distributionally Robust Decision-Focused Learning3 Feb 2026
- 2026On the Fragility of AI-Based Channel Decoders under Small Channel Perturbations2 Feb 2026
- 2026Robust Generalization with Adaptive Optimal Transport Priors for Decision-Focused Learning1 Feb 2026
- 2026Bulk-Calibrated Credal Ambiguity Sets: Fast, Tractable Decision Making under Out-of-Sample Contamination29 Jan 2026
- 2026Distributionally Robust Classification for Multi-source Unsupervised Domain Adaptation29 Jan 2026
50 newest of 291 All 291 (JSON)
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimizationcited by 18
- 2025Distributionally Robust Reinforcement Learning with Human Feedbackcited by 12
- 2025Stackelberg Self-Annotation: A Robust Approach to Data-Efficient LLM Alignmentcited by 10
- 2026Oracle-Robust Online Alignment for Large Language Modelscited by 1
- 2026Wasserstein Distributionally Robust Regret Optimization for Reinforcement Learning from Human Feedbackcited by 0
- 2026Wasserstein Distributionally Robust Regret Optimization for Reinforcement Learning from Human Feedback30 Apr 2026
- 2026Oracle-Robust Online Alignment for Large Language Models24 Feb 2026
- 2025Distributionally Robust Reinforcement Learning with Human Feedback1 Mar 2025
- 2025Stackelberg Self-Annotation: A Robust Approach to Data-Efficient LLM Alignment25 Feb 2025
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimization4 Feb 2025
- 2024Sequential Decision-Making under Uncertainty: A Robust MDPs reviewcited by 2
- 2024On Tractability, Complexity, and Mixed-Integer Convex Programming Representability of Distributionally Favorable Optimizationcited by 1
- 2025ADARL: Adaptive Low-Rank Structures for Robust Policy Learning under Uncertaintycited by 0
- 2026Policy Gradient Learning for Distributionally Robust Markov Decision Processes under Wasserstein Ambiguitycited by 0
- 2026Model-Based Reinforcement Learning for Heterogeneous Multi-Robot Task Assignment Under Distribution Shiftscited by 0
- 2026Model-Based Reinforcement Learning for Heterogeneous Multi-Robot Task Assignment Under Distribution Shifts21 Aug 2026
- 2026Policy Gradient Learning for Distributionally Robust Markov Decision Processes under Wasserstein Ambiguity25 Jun 2026
- 2025ADARL: Adaptive Low-Rank Structures for Robust Policy Learning under Uncertainty13 Oct 2025
- 2024Sequential Decision-Making under Uncertainty: A Robust MDPs review1 Apr 2024
- 2024On Tractability, Complexity, and Mixed-Integer Convex Programming Representability of Distributionally Favorable Optimization31 Jan 2024
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2021Unbalanced minibatch Optimal Transport; applications to Domain Adaptationcited by 125
- 2021Edge Artificial Intelligence for 6G: Vision, Enabling Technologies, and Applicationscited by 97
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2021Heterogeneous Risk Minimizationcited by 90
- 2021Distributionally Robust Learningcited by 44
- 2022Can Push-forward Generative Models Fit Multimodal Distributions?cited by 41
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2023A Finite Sample Complexity Bound for Distributionally Robust Q-learningcited by 36
- 2021Doubly-Valid/Doubly-Sharp Sensitivity Analysis for Causal Inference with Unmeasured Confoundingcited by 28
- 2021Statistical Analysis of Wasserstein Distributionally Robust Estimatorscited by 27
- 2021Outlier-Robust Optimal Transport: Duality, Structure, and Statistical Analysiscited by 27
12 most cited of 275 All 275 (JSON)
- 2026Private and Distributionally Robust Federated Multiclass Classification9 Oct 2026
- 2026Largest Rashomon sets of decision trees for robust contextual optimization3 Oct 2026
- 2026Distributionally Robust Survival Models under Subpopulation Shift and Outlier Contamination2 Oct 2026
- 2026Distributionally Robust Schr"odinger Bridge1 Oct 2026
- 2026Parameter-Efficient Distributionally Robust Adaptation of Tabular Foundation Models under Subpopulation Shift1 Oct 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026Learning Risk Scores Robust to Unobserved Confounders22 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
- 2026Certified Safety Radii in Forecast-Error Space for Wasserstein Distributionally Robust Small Signal Stability-Constrained AC Optimal Power Flow via Lifted Spectrahedral Containment31 Aug 2026
- 2026A Unified Perspective on Conformal Prediction and Wasserstein Distributionally Robust Optimization for Uncertainty Quantification30 Aug 2026
- 2026Generative Neural Networks for Sinkhorn Distributionally Robust Hypothesis Testing24 Aug 2026
12 newest of 275 All 275 (JSON)