Exact Generalization Guarantees for (Regularized) Wasserstein Distributionally Robust Models
Not on the map. Why?
Abstract (arXiv)
Wasserstein distributionally robust estimators have emerged as powerful models for prediction and decision-making under uncertainty. These estimators provide attractive generalization guarantees: the robust objective obtained from the training distribution is an exact upper bound on the true risk with high probability. However, existing guarantees either suffer from the curse of dimensionality, are restricted to specific settings, or lead to spurious error terms. In this paper, we show that these generalization guarantees actually hold on general classes of models, do not suffer from the curse of dimensionality, and can even cover distribution shifts at testing. We also prove that these results carry over to the newly-introduced regularized versions of Wasserstein distributionally robust problems.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 26 of its 49 references that are in the map. First posted on arXiv 26 May 2023.
Builds on
26 of 49 references.
References · 26
- 2017Certifying Some Distributional Robustness with Principled Adversarial Trainingcited by 337
- 2015Data-driven Distributionally Robust Optimization Using the Wasserstein Metric: Performance Guarantees and Tractable Reformulationscited by 291
- 2016Stochastic Optimization for Large-scale Optimal Transportcited by 189
- 2016Distributionally Robust Stochastic Optimization with Wasserstein Distancecited by 185
- 2020Large-Scale Methods for Distributionally Robust Optimizationcited by 179
- 2019Wasserstein Distributionally Robust Optimization: Theory and Applications in Machine Learningcited by 170
- 2015Distributionally Robust Logistic Regressioncited by 126
- 2017Regularization via Mass Transportationcited by 81
- 2019Distributionally Robust Optimization and Generalization in Kernel Methodscited by 73
- 2020Finite-Sample Guarantees for Wasserstein Distributionally Robust Optimization: Breaking the Curse of Dimensionalitycited by 49
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2018Optimal Transport Based Distributionally Robust Optimization: Structural Properties and Iterative Schemescited by 30
14 more
- 2021Statistical Analysis of Wasserstein Distributionally Robust Estimatorscited by 27
- 2022Non-Convex Bilevel Games with Critical Point Selection Mapscited by 25
- 2019Confidence Regions in Wasserstein Distributionally Robust Estimationcited by 24
- 2020Regularized Optimal Transport is Ground Cost Adversarialcited by 17
- 2020Principled learning method for Wasserstein distributionally robust optimization with local perturbationscited by 14
- 2017Semi-supervised Learning based on Distributionally Robust Optimizationcited by 12
- 2020Fast Epigraphical Projection-based Incremental Algorithms for Wasserstein Distributionally Robust Support Vector Machinecited by 10
- 2021Generalization Bounds with Minimal Dependency on Hypothesis Class via Distributionally Robust Optimizationcited by 9
- 2022A Data-Driven Approach to Robust Hypothesis Testing Using Sinkhorn Uncertainty Setscited by 9
- 2022Regularization for Wasserstein Distributionally Robust Optimizationcited by 9
- 2023Statistical Limit Theorems in Distributionally Robust Optimizationcited by 7
- 2020First-order Optimization for Superquantile-based Supervised Learningcited by 4
- 2021Adversarially Robust Kernel Smoothingcited by 4
- 2022Entropy-regularized Wasserstein distributionally robust shape and topology optimizationcited by 4
Built on by
Citing papers per year: 2023: 2; 2024: 2; 2025: 4; 2026: 1.
9 papers.
- 2023Unifying Distributionally Robust Optimization via Optimal Transport Theorycited by 13
- 2024Universal generalization guarantees for Wasserstein distributionally robust modelscited by 5
- 2024textttskwdro: a library for Wasserstein distributionally robust machine learningcited by 4
- 2025Wasserstein Distributionally Robust Nonparametric Regressioncited by 3
- 2023Mixture Data for Training Cannot Ensure Out-of-distribution Generalizationcited by 1
- 2025Provable Robust Overfitting Mitigation in Wasserstein Distributionally Robust Optimizationcited by 0
- 2025Distributionally Robust Wireless Semantic Communication with Large AI Modelscited by 0
- 2025Unregularized limit of stochastic gradient method for Wasserstein distributionally robust optimizationcited by 0
- 2026Statistical Guarantees for Distributionally Robust Optimization with Optimal Transport and OT-Regularized Divergencescited by 0
- 2026Statistical Guarantees for Distributionally Robust Optimization with Optimal Transport and OT-Regularized Divergences29 Mar 2026
- 2025Unregularized limit of stochastic gradient method for Wasserstein distributionally robust optimization5 Jun 2025
- 2025Distributionally Robust Wireless Semantic Communication with Large AI Models28 May 2025
- 2025Wasserstein Distributionally Robust Nonparametric Regression12 May 2025
- 2025Provable Robust Overfitting Mitigation in Wasserstein Distributionally Robust Optimization6 Mar 2025
- 2024textttskwdro: a library for Wasserstein distributionally robust machine learning28 Oct 2024
- 2024Universal generalization guarantees for Wasserstein distributionally robust models19 Feb 2024
- 2023Mixture Data for Training Cannot Ensure Out-of-distribution Generalization25 Dec 2023
- 2023Unifying Distributionally Robust Optimization via Optimal Transport Theory10 Aug 2023