Learning Models with Uniform Performance via Distributionally Robust Optimization
Not on the map. Why?
Abstract (arXiv)
A common goal in statistics and machine learning is to learn models that can perform well against distributional shifts, such as latent heterogeneous subpopulations, unknown covariate shifts, or unmodeled temporal effects. We develop and analyze a distributionally robust stochastic optimization (DRO) framework that learns a model providing good performance against perturbations to the data-generating distribution. We give a convex formulation for the problem, providing several convergence guarantees. We prove finite-sample minimax upper and lower bounds, showing that distributional robustness sometimes comes at a cost in convergence rates. We give limit theorems for the learned parameters, where we fully specify the limiting distribution so that confidence intervals can be computed. On real tasks including generalizing to unknown subpopulations, fine-grained recognition, and providing good tail performance, the distributionally robust approach often exhibits improved performance.
Key facts. Cited by 335 papers in the map (data through 14 Oct 2026). First posted on arXiv 20 Oct 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 41; 2022: 60; 2023: 69; 2024: 58; 2025: 60; 2026: 47.
335 papers.
Sort:
By area:
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2023Change is Hard: A Closer Look at Subpopulation Shiftcited by 120
- 2022Robust Reinforcement Learning using Offline Datacited by 96
- 2021Heterogeneous Risk Minimizationcited by 90
- 2022Distributionally Robust Model-Based Offline Reinforcement Learning with Near-Optimal Sample Complexitycited by 87
- 2022Probable Domain Generalization via Quantile Risk Minimizationcited by 66
- 2022ZIN: When and How to Learn Invariance Without Environment Partition?cited by 58
- 2021DORO: Distributional and Outlier Robust Optimizationcited by 56
- 2024Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimizationcited by 53
- 2023The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative Modelcited by 52
- 2022Out-of-distribution Generalization with Causal Invariant Transformationscited by 45
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 44
38 more
- 2023Improved Sample Complexity Bounds for Distributionally Robust Reinforcement Learningcited by 44
- 2021On Tilted Losses in Machine Learning: Theory and Applicationscited by 43
- 2021Towards Robust and Adaptive Motion Forecasting: A Causal Representation Perspectivecited by 40
- 2023Improving Fairness in Deepfake Detectioncited by 40
- 2021Non-convex Distributionally Robust Optimization: Non-asymptotic Analysiscited by 39
- 2022On-Demand Sampling: Learning Optimally from Multiple Distributionscited by 39
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 39
- 2023A Finite Sample Complexity Bound for Distributionally Robust Q-learningcited by 36
- 2023Natural Actor-Critic for Robust Reinforcement Learning with Function Approximationcited by 36
- 2022Doubly Robust Distributionally Robust Off-Policy Evaluation and Learningcited by 34
- 2022A System-Level View on Out-of-Distribution Data in Roboticscited by 33
- 2022Distributionally Robust Offline Reinforcement Learning with Linear Function Approximationcited by 32
- 2022Minimax Regret Optimization for Robust Machine Learning under Distribution Shiftcited by 31
- 2022Metaphors in Pre-Trained Language Models: Probing and Generalization Across Datasets and Languagescited by 30
- 2021Modeling the Second Player in Distributionally Robust Optimizationcited by 29
- 2021Off-Policy Risk Assessment in Contextual Banditscited by 29
- 2022Understanding Why Generalized Reweighting Does Not Improve Over ERMcited by 29
- 2023Robust Generalization against Photon-Limited Corruptions via Worst-Case Sharpness Minimizationcited by 29
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalizationcited by 29
- 2024AI-Face: A Million-Scale Demographically Annotated AI-Generated Face Dataset and Fairness Benchmarkcited by 28
- 2021Towards Theoretical Understandings of Robust Markov Decision Processes: Sample Complexity and Asymptoticscited by 27
- 2021Focus on the Common Good: Group Distributional Robustness Followscited by 25
- 2024Sample-Efficient Robust Multi-Agent Reinforcement Learning in the Face of Environmental Uncertaintycited by 25
- 2021Kernelized Heterogeneous Risk Minimizationcited by 21
- 2023Understanding Contrastive Learning via Distributionally Robust Optimizationcited by 21
- 2024A Survey on Evaluation of Out-of-Distribution Generalizationcited by 21
- 2023Statistical Inference for Fairness Auditingcited by 20
- 2023Rethinking the Evaluation Protocol of Domain Generalizationcited by 20
- 2021A Theoretical Analysis on Independence-driven Importance Weighting for Covariate-shift Generalizationcited by 19
- 2025AVTrustBench: Assessing and Enhancing Reliability and Robustness in Audio-Visual LLMscited by 19
- 2021Distributionally Robust Multilingual Machine Translationcited by 18
- 2022How unfair is private learning ?cited by 18
- 2023On the Theories Behind Hard Negative Sampling for Recommendationcited by 18
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimizationcited by 18
- 2022Distributionally Robust Models with Parametric Likelihood Ratioscited by 17
- 2022Subgroup Robustness Grows On Trees: An Empirical Baseline Investigationcited by 17
- 2022Near-Optimal Algorithms for Group Distributionally Robust Optimization and Beyondcited by 17
- 2023Distributionally Robust Model-based Reinforcement Learning with Large State Spacescited by 17
50 most cited of 335 All 335 (JSON)
- 2026Distributionally Robust Survival Models under Subpopulation Shift and Outlier Contamination2 Oct 2026
- 2026Parameter-Efficient Distributionally Robust Adaptation of Tabular Foundation Models under Subpopulation Shift1 Oct 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026RSD-Poker: Structure-Adaptive and Shift-Robust Risk-Utility Certification for Residual Policies in Imperfect-Information Games27 Sep 2026
- 2026Beyond Empirical Support: Structured Outlier Generation via Sinkhorn Optimal Transport25 Sep 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026CVaR anchor regression protects against rare shifts22 Sep 2026
- 2026Occupancy-based Quantile Risk Control2 Sep 2026
- 2026Selection-Aware Stress Testing for Interactive Agents31 Aug 2026
- 2026Mitigating Over-Optimization in PRM-Guided Search in Mathematical Reasoning by Optimizing the Guide30 Aug 2026
- 2026Multi-Source Wasserstein Distributionally Robust Graph Learning20 Aug 2026
- 2026The Sample Complexity of Distributionally Robust PAC Learning under Cressie--Read Divergences5 Aug 2026
38 more
- 2026Augmented Inverse Hybrid Weighting: Robust Inference under Deterministic and Random Distribution Shifts1 Aug 2026
- 2026SAGG: Sample-Adaptive Gradient Gating for Robust Multimodal Learning under Heterogeneous Corruption30 Jul 2026
- 2026Learning Predictive Ambiguity Sets for Decision-Focused Distributionally Robust Optimization10 Jul 2026
- 2026RUBRIC: Realism--Utility Balanced Ranking for Imbalanced Classification10 Jul 2026
- 2026Environment-Robust Representation Learning with Empirical Bayes3 Jun 2026
- 2026Shortcut to Nowhere: Demystifying Deep Spurious Regression1 Jun 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
- 2026Penalty-Based First-Order Methods for Bilevel Optimization with Minimax and Constrained Lower-Level Problems8 May 2026
- 2026Agentic AIs Are the Missing Paradigm for Out-of-Distribution Generalization in Foundation Models7 May 2026
- 2026From Dual Tracking to Clipping: Provably Faster Distributionally Robust Multi-Objective Optimization7 May 2026
- 2026Taming the Curses of Multiagency in Robust Markov Games with Large State Space through Linear Function Approximation4 May 2026
- 2026Robust Representation Learning through Explicit Environment Modeling28 Apr 2026
- 2026Deep Learning for Sequential Decision Making under Uncertainty: Foundations, Frameworks, and Frontiers13 Apr 2026
- 2026Statistical Guarantees for Distributionally Robust Optimization with Optimal Transport and OT-Regularized Divergences29 Mar 2026
- 2026Stochastic set-valued optimization and its application to robust learning18 Mar 2026
- 2026Structured Credal Learning14 Mar 2026
- 2026DARC: Disagreement-Aware Alignment via Risk-Constrained Decoding9 Mar 2026
- 2026First-Order Softmax Weighted Switching Gradient Method for Distributed Stochastic Minimax Optimization with Stochastic Constraints6 Mar 2026
- 2026Robust AI Evaluation through Maximal Lotteries24 Feb 2026
- 2026Oracle-Robust Online Alignment for Large Language Models24 Feb 2026
- 2026Agentic Adversarial QA for Improving Domain-Specific LLMs20 Feb 2026
- 2026Omitted Variable Bias in Language Models Under Distribution Shift18 Feb 2026
- 2026Differentially Private Non-convex Distributionally Robust Optimization18 Feb 2026
- 2026Distributionally Robust Cooperative Multi-Agent Reinforcement Learning via Robust Value Factorization11 Feb 2026
- 2026Adaptive Sampling and Clipping for Private Worst-Case Group Optimization11 Feb 2026
- 2026Robust Policy Optimization to Prevent Catastrophic Forgetting9 Feb 2026
- 2026Rank-Constrained Adaptation for Reliable Real-World Performance6 Feb 2026
- 2026f-FUM: Federated Unlearning via min--max and f-divergence5 Feb 2026
- 2026Solving Stochastic Variational Inequalities without the Bounded Variance Assumption5 Feb 2026
- 2026Environment-Conditioned Tail Reweighting for Invariant Learning under Mixed Shifts30 Jan 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
- 2026Robust Learning of a Group DRO Neuron26 Jan 2026
- 2026Adversarial Drift-Aware Predictive Transfer: Toward Durable Clinical AI17 Jan 2026
- 2025Online Robust Reinforcement Learning with General Function Approximation22 Dec 2025
- 2025Maximum Risk Minimization with Random Forests11 Dec 2025
- 2025On the Bayes Inconsistency of Disagreement Discrepancy Surrogates5 Dec 2025
50 newest of 335 All 335 (JSON)
- 2024Mitigating Covariate Shift in Misspecified Regression with Applications to Reinforcement Learningcited by 4
- 2024Model-Free Robust φ-Divergence Reinforcement Learning Using Both Offline and Online Datacited by 4
- 2025Hybrid Cross-domain Robust Reinforcement Learningcited by 3
- 2025Statistical and Algorithmic Foundations of Reinforcement Learningcited by 2
- 2025Online Robust Reinforcement Learning with General Function Approximationcited by 2
- 2025Near-Optimal Sample Complexities of Divergence-based S-rectangular Distributionally Robust Reinforcement Learningcited by 1
- 2025Finite-Time Convergence of Single-Trajectory Chi-Square Robust Q-Learning With Linear Function Approximationcited by 0
- 2025Robust Reinforcement Learning in Finance: Modeling Market Impact with Elliptic Uncertainty Setscited by 0
- 2026Taming the Curses of Multiagency in Robust Markov Games with Large State Space through Linear Function Approximationcited by 0
- 2026Taming the Curses of Multiagency in Robust Markov Games with Large State Space through Linear Function Approximation4 May 2026
- 2025Online Robust Reinforcement Learning with General Function Approximation22 Dec 2025
- 2025Robust Reinforcement Learning in Finance: Modeling Market Impact with Elliptic Uncertainty Sets22 Oct 2025
- 2025Finite-Time Convergence of Single-Trajectory Chi-Square Robust Q-Learning With Linear Function Approximation2 Oct 2025
- 2025Statistical and Algorithmic Foundations of Reinforcement Learning19 Jul 2025
- 2025Hybrid Cross-domain Robust Reinforcement Learning29 May 2025
- 2025Near-Optimal Sample Complexities of Divergence-based S-rectangular Distributionally Robust Reinforcement Learning18 May 2025
- 2024Model-Free Robust φ-Divergence Reinforcement Learning Using Both Offline and Online Data8 May 2024
- 2024Mitigating Covariate Shift in Misspecified Regression with Applications to Reinforcement Learning22 Jan 2024
- 2024Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimizationcited by 53
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimizationcited by 18
- 2025On Symmetric Losses for Robust Policy Optimization with Noisy Preferencescited by 4
- 2026Robust Policy Optimization to Prevent Catastrophic Forgettingcited by 3
- 2025Leveraging Robust Optimization for LLM Alignment under Distribution Shiftscited by 2
- 2025Robust Preference Alignment via Directional Neighborhood Consensuscited by 2
- 2026Oracle-Robust Online Alignment for Large Language Modelscited by 1
- 2026DARC: Disagreement-Aware Alignment via Risk-Constrained Decodingcited by 0
- 2026DARC: Disagreement-Aware Alignment via Risk-Constrained Decoding9 Mar 2026
- 2026Oracle-Robust Online Alignment for Large Language Models24 Feb 2026
- 2026Robust Policy Optimization to Prevent Catastrophic Forgetting9 Feb 2026
- 2025Robust Preference Alignment via Directional Neighborhood Consensus23 Oct 2025
- 2025On Symmetric Losses for Robust Policy Optimization with Noisy Preferences30 May 2025
- 2025Leveraging Robust Optimization for LLM Alignment under Distribution Shifts8 Apr 2025
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimization4 Feb 2025
- 2024Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimization10 Jul 2024
- 2022Distributionally Robust Model-Based Offline Reinforcement Learning with Near-Optimal Sample Complexitycited by 87
- 2022Distributionally Robust Offline Reinforcement Learning with Linear Function Approximationcited by 32
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coveragecited by 12
- 2024DIGIC: Domain Generalizable Imitation Learning by Causal Discoverycited by 0
- 2024DIGIC: Domain Generalizable Imitation Learning by Causal Discovery29 Feb 2024
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coverage27 Oct 2023
- 2022Distributionally Robust Offline Reinforcement Learning with Linear Function Approximation14 Sep 2022
- 2022Distributionally Robust Model-Based Offline Reinforcement Learning with Near-Optimal Sample Complexity11 Aug 2022
- 2023Attacks on fairness in Federated Learningcited by 1
- 2026f-FUM: Federated Unlearning via min--max and f-divergencecited by 0
- 2026f-FUM: Federated Unlearning via min--max and f-divergence5 Feb 2026
- 2023Attacks on fairness in Federated Learning21 Nov 2023
- 2025Knowledge Distillation with Adapted Weightcited by 0
- 2025Knowledge Distillation with Adapted Weight6 Jan 2025
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2023Change is Hard: A Closer Look at Subpopulation Shiftcited by 120
- 2022Robust Reinforcement Learning using Offline Datacited by 96
- 2021Heterogeneous Risk Minimizationcited by 90
- 2022Probable Domain Generalization via Quantile Risk Minimizationcited by 66
- 2022ZIN: When and How to Learn Invariance Without Environment Partition?cited by 58
- 2021DORO: Distributional and Outlier Robust Optimizationcited by 56
- 2023The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative Modelcited by 52
- 2022Out-of-distribution Generalization with Causal Invariant Transformationscited by 45
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 44
- 2023Improved Sample Complexity Bounds for Distributionally Robust Reinforcement Learningcited by 44
- 2021On Tilted Losses in Machine Learning: Theory and Applicationscited by 43
12 most cited of 306 All 306 (JSON)
- 2026Distributionally Robust Survival Models under Subpopulation Shift and Outlier Contamination2 Oct 2026
- 2026Parameter-Efficient Distributionally Robust Adaptation of Tabular Foundation Models under Subpopulation Shift1 Oct 2026
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026RSD-Poker: Structure-Adaptive and Shift-Robust Risk-Utility Certification for Residual Policies in Imperfect-Information Games27 Sep 2026
- 2026Beyond Empirical Support: Structured Outlier Generation via Sinkhorn Optimal Transport25 Sep 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026CVaR anchor regression protects against rare shifts22 Sep 2026
- 2026Occupancy-based Quantile Risk Control2 Sep 2026
- 2026Selection-Aware Stress Testing for Interactive Agents31 Aug 2026
- 2026Multi-Source Wasserstein Distributionally Robust Graph Learning20 Aug 2026
- 2026The Sample Complexity of Distributionally Robust PAC Learning under Cressie--Read Divergences5 Aug 2026
- 2026Augmented Inverse Hybrid Weighting: Robust Inference under Deterministic and Random Distribution Shifts1 Aug 2026
12 newest of 306 All 306 (JSON)