Large-Scale Methods for Distributionally Robust Optimization
Not on the map. Why?
Abstract (arXiv)
We propose and analyze algorithms for distributionally robust optimization of convex losses with conditional value at risk (CVaR) and chi2 divergence uncertainty sets. We prove that our algorithms require a number of gradient evaluations independent of training set size and number of parameters, making them suitable for large-scale applications. For chi2 uncertainty sets these are the first such guarantees in the literature, and for CVaR our guarantees scale linearly in the uncertainty level rather than quadratically as in previous work. We also provide lower bounds proving the worst-case optimality of our algorithms for CVaR and a penalized version of the chi2 problem. Our primary technical contributions are novel bounds on the bias of batch robust risk estimation and the variance of a multilevel Monte Carlo gradient estimator due to [Blanchet & Glynn, 2015]. Experiments on MNIST and ImageNet confirm the theoretical scaling of our algorithms, which are 9--36 times more efficient than full-batch methods.
Key facts. Cited by 179 papers in the map (data through 14 Oct 2026). First posted on arXiv 12 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 17; 2022: 38; 2023: 37; 2024: 39; 2025: 25; 2026: 23.
179 papers.
Sort:
By area:
- 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 523
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2022Challenges and Strategies in Cross-Cultural NLPcited by 218
- 2022Correct-N-Contrast: A Contrastive Approach for Improving Robustness to Spurious Correlationscited by 172
- 2024RL on Incorrect Synthetic Data Scales the Efficiency of LLM Math Reasoning by Eight-Foldcited by 99
- 2022Spread Spurious Attribute: Improving Worst-group Accuracy with Spurious Attribute Estimationcited by 88
- 2023DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedulecited by 85
- 2023Discover and Cure: Concept-aware Mitigation of Spurious Correlationcited by 71
- 2024Preserving Fairness Generalization in Deepfake Detectioncited by 70
- 2022MaskTune: Mitigating Spurious Correlations by Forcing to Explorecited by 66
- 2023Benchmarking Distribution Shift in Tabular Data with TableShiftcited by 66
- 2021Risk-Averse Offline Reinforcement Learningcited by 65
38 more
- 2022Diversify and Disambiguate: Learning From Underspecified Datacited by 56
- 2022Learning Debiased Classifier with Biased Committeecited by 51
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 44
- 2021On Tilted Losses in Machine Learning: Theory and Applicationscited by 43
- 2024The Clever Hans Mirage: A Comprehensive Survey on Spurious Correlations in Machine Learningcited by 41
- 2023A Guide Through the Zoo of Biased SGDcited by 40
- 2023Improving Fairness in Deepfake Detectioncited by 40
- 2021Non-convex Distributionally Robust Optimization: Non-asymptotic Analysiscited by 39
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2022Adapting to Mixing Time in Stochastic Optimization with Markovian Datacited by 36
- 2025Representation Learning for Tabular Data: A Comprehensive Surveycited by 36
- 2022UMIX: Improving Importance Weighting for Subpopulation Shift via Uncertainty-Aware Mixupcited by 34
- 2022Calibrated Selective Classificationcited by 33
- 2022Learning Fair Representation via Distributional Contrastive Disentanglementcited by 31
- 2022When AUC meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence Guaranteecited by 29
- 2024AI-Face: A Million-Scale Demographically Annotated AI-Generated Face Dataset and Fairness Benchmarkcited by 28
- 2022Rank-based Decomposable Losses in Machine Learning: A Surveycited by 26
- 2021Stochastic Bias-Reduced Gradient Methodscited by 25
- 2021BARACK: Partially Supervised Group Robustness With Guaranteescited by 25
- 2023Not All Semantics are Created Equal: Contrastive Self-supervised Learning with Automatic Temperature Individualizationcited by 25
- 2023Improving Generalization of Alignment with Human Preferences through Group Invariant Learningcited by 25
- 2021Distributionally Robust Multilingual Machine Translationcited 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
- 2022Learning to Split for Automatic Bias Detectioncited by 16
- 2022Stochastic Constrained DRO with a Complexity Independent of Sample Sizecited by 16
- 2025Upweighting Easy Samples in Fine-Tuning Mitigates Forgettingcited by 16
- 2024A Short Survey on Importance Weighting for Machine Learningcited by 15
- 2022Distributionally Robust Optimization via Ball Oracle Accelerationcited by 14
- 2022Learning from a Biased Samplecited by 14
- 2022Distributed Distributionally Robust Optimization with Non-Convex Objectivescited by 14
- 2023Distributionally Robust Optimization with Probabilistic Groupcited by 14
- 2025Sign Operator for Coping with Heavy-Tailed Noise in Non-Convex Optimization: High Probability Bounds Under (L0, L1)-Smoothnesscited by 13
- 2021Constrained Instance and Class Reweighting for Robust Learning under Label Noisecited by 12
- 2022Nonsmooth Nonconvex-Nonconcave Minimax Optimization: Primal-Dual Balancing and Iteration Complexity Analysiscited by 12
- 2022Universal Gradient Descent Ascent Method for Nonconvex-Nonconcave Minimax Optimizationcited by 12
50 most cited of 179 All 179 (JSON)
- 2026Parameter-Efficient Distributionally Robust Adaptation of Tabular Foundation Models under Subpopulation Shift1 Oct 2026
- 2026Scalable Cox Regression via Grouped Risk Sets and Sharper LogSumExp Rates30 Sep 2026
- 2026FedeRage: Provably Convergent Agnostic Federated Learning under General Client Drift17 Sep 2026
- 2026Beyond the Performance Illusion: Structure-Aware Stratified Partitioning and Curriculum Distributionally Robust Optimization for Spatially Correlated Domains2 Jul 2026
- 2026Automated Background Swapping for Robustness against Spurious Backgrounds30 Jun 2026
- 2026Toward Calibrated Mixture-of-Experts Under Distribution Shift18 Jun 2026
- 2026Unification and Optimization of Robust Supervised Learning27 May 2026
- 2026Deep Attention Reweighting: Post-Hoc Attention-Based Feature Aggregation in CNNs for Disentangling Core and Spurious Features under Spurious Correlations20 May 2026
- 2026From Dual Tracking to Clipping: Provably Faster Distributionally Robust Multi-Objective Optimization7 May 2026
- 2026Anatomy of a failure: When, how, and why deep vision fails in scientific domains5 May 2026
- 2026Deciphering Shortcut Learning from an Evolutionary Game Theory Perspective4 May 2026
- 2026Robust Fair Disease Diagnosis in CT Images8 Apr 2026
38 more
- 2026RAMPAGE: RAndomized Mid-Point for debiAsed Gradient Extrapolation23 Mar 2026
- 2026Stochastic set-valued optimization and its application to robust learning18 Mar 2026
- 2026Unbiased and Biased Variance-Reduced Forward-Reflected-Backward Splitting Methods for Stochastic Composite Inclusions16 Mar 2026
- 2026DARC: Disagreement-Aware Alignment via Risk-Constrained Decoding9 Mar 2026
- 2026Differentially Private Non-convex Distributionally Robust Optimization18 Feb 2026
- 2026Adaptive Sampling and Clipping for Private Worst-Case Group Optimization11 Feb 2026
- 2026REMIND: Rethinking Medical High-Modality Learning under Missingness--A Long-Tailed Distribution Perspective9 Feb 2026
- 2026Revisiting Robustness for LLM Safety Alignment via Selective Geometry Control7 Feb 2026
- 2026A Geometry-Aware Efficient Algorithm for Compositional Entropic Risk Minimization2 Feb 2026
- 2026Analyzing and Improving Diffusion Models for Time-Series Data Imputation: A Proximal Recursion Perspective1 Feb 2026
- 2026RobustDebias: Debiasing Language Models using Distributionally Robust Optimization30 Jan 2026
- 2025Certifying the Right to Be Forgotten: Primal-Dual Optimization for Sample and Label Unlearning in Vertical Federated Learning29 Dec 2025
- 2025Shortcut Invariance: Targeted Jacobian Regularization in Disentangled Latent Space24 Nov 2025
- 2025FAST-CAD: A Fairness-Aware Framework for Non-Contact Stroke Diagnosis12 Nov 2025
- 2025Gradient Flow Sampler-based Distributionally Robust Optimization29 Oct 2025
- 2025Improved Stochastic Optimization of LogSumExp29 Sep 2025
- 2025Distributionally Robust Federated Learning with Outlier Resilience29 Sep 2025
- 2025Superclass-Guided Representation Disentanglement for Spurious Correlation Mitigation12 Aug 2025
- 2025Revisiting Convergence: Shuffling Complexity Beyond Lipschitz Smoothness11 Jul 2025
- 2025Duality and Policy Evaluation in Distributionally Robust Bayesian Diffusion Control24 Jun 2025
- 2025Bridging Distribution Shift and AI Safety: Conceptual and Methodological Synergies28 May 2025
- 2025When Shift Happens - Confounding Is to Blame27 May 2025
- 2025Preserving AUC Fairness in Learning with Noisy Protected Groups24 May 2025
- 2025Conformalized Decision Risk Assessment19 May 2025
- 2025Model Steering: Learning with a Reference Model Improves Generalization Bounds and Scaling Laws10 May 2025
- 2025Representation Learning for Tabular Data: A Comprehensive Survey17 Apr 2025
- 2025Nested Stochastic Algorithm for Generalized Sinkhorn distance-Regularized Distributionally Robust Optimization29 Mar 2025
- 2025Improved Rates of Differentially Private Nonconvex-Strongly-Concave Minimax Optimization24 Mar 2025
- 2025pFedFair: Towards Optimal Group Fairness-Accuracy Trade-off in Heterogeneous Federated Learning19 Mar 2025
- 2025AIM-Fair: Advancing Algorithmic Fairness via Selectively Fine-Tuning Biased Models with Contextual Synthetic Data7 Mar 2025
- 2025Elliptic Loss Regularization4 Mar 2025
- 2025Distributionally Robust Reinforcement Learning with Human Feedback1 Mar 2025
- 2025A Critical Review of Predominant Bias in Neural Networks16 Feb 2025
- 2025Sign Operator for Coping with Heavy-Tailed Noise in Non-Convex Optimization: High Probability Bounds Under (L0, L1)-Smoothness11 Feb 2025
- 2025Upweighting Easy Samples in Fine-Tuning Mitigates Forgetting5 Feb 2025
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimization4 Feb 2025
- 2024Re-evaluating Group Robustness via Adaptive Class-Specific Scaling19 Dec 2024
- 2024DRPruning: Efficient Large Language Model Pruning through Distributionally Robust Optimization21 Nov 2024
50 newest of 179 All 179 (JSON)
- 2023Improving Generalization of Alignment with Human Preferences through Group Invariant Learningcited by 25
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimizationcited by 18
- 2025Distributionally Robust Reinforcement Learning with Human Feedbackcited by 12
- 2026Revisiting Robustness for LLM Safety Alignment via Selective Geometry Controlcited by 4
- 2026DARC: Disagreement-Aware Alignment via Risk-Constrained Decodingcited by 0
- 2026DARC: Disagreement-Aware Alignment via Risk-Constrained Decoding9 Mar 2026
- 2026Revisiting Robustness for LLM Safety Alignment via Selective Geometry Control7 Feb 2026
- 2025Distributionally Robust Reinforcement Learning with Human Feedback1 Mar 2025
- 2025Robust LLM Alignment via Distributionally Robust Direct Preference Optimization4 Feb 2025
- 2023Improving Generalization of Alignment with Human Preferences through Group Invariant Learning18 Oct 2023
- 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 523
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
- 2022Challenges and Strategies in Cross-Cultural NLPcited by 218
- 2022Correct-N-Contrast: A Contrastive Approach for Improving Robustness to Spurious Correlationscited by 172
- 2022Spread Spurious Attribute: Improving Worst-group Accuracy with Spurious Attribute Estimationcited by 88
- 2023DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedulecited by 85
- 2023Discover and Cure: Concept-aware Mitigation of Spurious Correlationcited by 71
- 2024Preserving Fairness Generalization in Deepfake Detectioncited by 70
- 2022MaskTune: Mitigating Spurious Correlations by Forcing to Explorecited by 66
- 2023Benchmarking Distribution Shift in Tabular Data with TableShiftcited by 66
- 2022Diversify and Disambiguate: Learning From Underspecified Datacited by 56
- 2022Learning Debiased Classifier with Biased Committeecited by 51
12 most cited of 164 All 164 (JSON)
- 2026Parameter-Efficient Distributionally Robust Adaptation of Tabular Foundation Models under Subpopulation Shift1 Oct 2026
- 2026Scalable Cox Regression via Grouped Risk Sets and Sharper LogSumExp Rates30 Sep 2026
- 2026FedeRage: Provably Convergent Agnostic Federated Learning under General Client Drift17 Sep 2026
- 2026Beyond the Performance Illusion: Structure-Aware Stratified Partitioning and Curriculum Distributionally Robust Optimization for Spatially Correlated Domains2 Jul 2026
- 2026Automated Background Swapping for Robustness against Spurious Backgrounds30 Jun 2026
- 2026Unification and Optimization of Robust Supervised Learning27 May 2026
- 2026Deep Attention Reweighting: Post-Hoc Attention-Based Feature Aggregation in CNNs for Disentangling Core and Spurious Features under Spurious Correlations20 May 2026
- 2026From Dual Tracking to Clipping: Provably Faster Distributionally Robust Multi-Objective Optimization7 May 2026
- 2026Anatomy of a failure: When, how, and why deep vision fails in scientific domains5 May 2026
- 2026Deciphering Shortcut Learning from an Evolutionary Game Theory Perspective4 May 2026
- 2026Robust Fair Disease Diagnosis in CT Images8 Apr 2026
- 2026RAMPAGE: RAndomized Mid-Point for debiAsed Gradient Extrapolation23 Mar 2026
12 newest of 164 All 164 (JSON)