Distributionally Robust Losses for Latent Covariate Mixtures
Not on the map. Why?
Abstract (arXiv)
While modern large-scale datasets often consist of heterogeneous subpopulations -- for example, multiple demographic groups or multiple text corpora -- the standard practice of minimizing average loss fails to guarantee uniformly low losses across all subpopulations. We propose a convex procedure that controls the worst-case performance over all subpopulations of a given size. Our procedure comes with finite-sample (nonparametric) convergence guarantees on the worst-off subpopulation. Empirically, we observe on lexical similarity, wine quality, and recidivism prediction tasks that our worst-case procedure learns models that do well against unseen subpopulations.
Key facts. Cited by 57 papers in the map (data through 7 Oct 2026). First posted on arXiv 28 Jul 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 13; 2022: 12; 2023: 11; 2024: 11; 2025: 8; 2026: 2.
57 papers.
Sort:
By area:
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
- 2022NICO++: Towards Better Benchmarking for Domain Generalizationcited by 86
- 2021The Spotlight: A General Method for Discovering Systematic Errors in Deep Learning Modelscited by 63
- 2021Unsupervised Learning of Debiased Representations with Pseudo-Attributescited by 44
- 2022Perfectly Balanced: Improving Transfer and Robustness of Supervised Contrastive Learningcited by 40
- 2022Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalizationcited by 38
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 37
- 2021An Online Learning Approach to Interpolation and Extrapolation in Domain Generalizationcited by 30
- 2021Simple and near-optimal algorithms for hidden stratification and multi-group learningcited by 26
- 2021Regularizing towards Causal Invariance: Linear Models with Proxiescited by 20
- 2021A Theoretical Analysis on Independence-driven Importance Weighting for Covariate-shift Generalizationcited by 18
- 2022Distributionally Robust Models with Parametric Likelihood Ratioscited by 17
38 more
- 2022Subgroup Robustness Grows On Trees: An Empirical Baseline Investigationcited by 17
- 2021Preventing dataset shift from breaking machine-learning biomarkerscited by 14
- 2022Distributionally Robust Survival Analysis: A Novel Fairness Loss Without Demographicscited by 14
- 2021Algorithmic Bias and Data Bias: Understanding the Relation between Distributionally Robust Optimization and Data Curationcited by 13
- 2024Fairness in Survival Analysis with Distributionally Robust Optimizationcited by 13
- 2021A comparison of approaches to improve worst-case predictive model performance over patient subpopulationscited by 12
- 2022Learning from a Biased Samplecited by 12
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coveragecited by 12
- 2024Addressing Polarization and Unfairness in Performative Predictioncited by 11
- 2025Task Generalization With AutoRegressive Compositional Structure: Can Learning From D Tasks Generalize to DT Tasks?cited by 11
- 2022Evaluating Robustness to Dataset Shift via Parametric Robustness Setscited by 10
- 2022Understanding the effect of sparsity on neural networks robustnesscited by 8
- 2024ExMap: Leveraging Explainability Heatmaps for Unsupervised Group Robustness to Spurious Correlationscited by 8
- 2021Learning Domain Invariant Representations by Joint Wasserstein Distance Minimizationcited by 6
- 2023Scaff-PD: Communication Efficient Fair and Robust Federated Learningcited by 6
- 2023Is this model reliable for everyone? Testing for strong calibrationcited by 6
- 2023Causality-oriented robustness: exploiting general noise interventionscited by 5
- 2023Geometry-Calibrated DRO: Combating Over-Pessimism with Free Energy Implicationscited by 5
- 2024Evaluating Model Performance Under Worst-case Subpopulationscited by 5
- 2024DRPruning: Efficient Large Language Model Pruning through Distributionally Robust Optimizationcited by 5
- 2025DRO: A Python Library for Distributionally Robust Optimization in Machine Learningcited by 5
- 2022Minimax Optimal Estimation of Stability Under Distribution Shiftcited by 4
- 2023Optimizer's Information Criterion: Dissecting and Correcting Bias in Data-Driven Optimizationcited by 4
- 2025Sample-Efficient Omniprediction for Proper Lossescited by 4
- 2022Long Term Fairness for Minority Groups via Performative Distributionally Robust Optimizationcited by 3
- 2022How does overparametrization affect performance on minority groups?cited by 2
- 2024Federated Fairness without Access to Sensitive Groupscited by 2
- 2024Contextual Optimization under Covariate Shift: A Robust Approach by Intersecting Wasserstein Ballscited by 2
- 2024CherryRec: Enhancing News Recommendation Quality via LLM-driven Frameworkcited by 2
- 2024Minimax rates of convergence for nonparametric regression under adversarial attackscited by 2
- 2023Predictive Heterogeneity: Measures and Applicationscited by 1
- 2024Footprints of Data in a Classifier: Understanding the Privacy Risks and Solution Strategiescited by 1
- 2024On the KL-Divergence-based Robust Satisficing Modelcited by 1
- 2025CTC-DRO: Robust Optimization for Reducing Language Disparities in Speech Recognitioncited by 1
- 2025DivIL: Unveiling and Addressing Over-Invariance for Out-of- Distribution Generalizationcited by 1
- 2025Distributionally Robust Federated Learning with Outlier Resiliencecited by 1
- 2021A Topological-Framework to Improve Analysis of Machine Learning Model Performancecited by 0
- 2021Learning Neural Models for Natural Language Processing in the Face of Distributional Shiftcited by 0
50 most cited of 57 All 57 (JSON)
- 2026The Privacy Price of Tail-Risk Learning: Effective Tail Sample Size in Differentially Private CVaR Optimization15 May 2026
- 2026Deep Learning for Sequential Decision Making under Uncertainty: Foundations, Frameworks, and Frontiers13 Apr 2026
- 2025Sample-Efficient Omniprediction for Proper Losses14 Oct 2025
- 2025Distributionally Robust Federated Learning with Outlier Resilience29 Sep 2025
- 2025ACE and Diverse Generalization via Selective Disagreement9 Sep 2025
- 2025Data Heterogeneity Modeling for Trustworthy Machine Learning1 Jun 2025
- 2025DRO: A Python Library for Distributionally Robust Optimization in Machine Learning29 May 2025
- 2025DivIL: Unveiling and Addressing Over-Invariance for Out-of- Distribution Generalization18 Feb 2025
- 2025Task Generalization With AutoRegressive Compositional Structure: Can Learning From D Tasks Generalize to DT Tasks?13 Feb 2025
- 2025CTC-DRO: Robust Optimization for Reducing Language Disparities in Speech Recognition3 Feb 2025
- 2024DRPruning: Efficient Large Language Model Pruning through Distributionally Robust Optimization21 Nov 2024
- 2024Minimax rates of convergence for nonparametric regression under adversarial attacks12 Oct 2024
38 more
- 2024Fairness in Survival Analysis with Distributionally Robust Optimization31 Aug 2024
- 2024On the KL-Divergence-based Robust Satisficing Model17 Aug 2024
- 2024Footprints of Data in a Classifier: Understanding the Privacy Risks and Solution Strategies2 Jul 2024
- 2024Evaluating Model Performance Under Worst-case Subpopulations1 Jul 2024
- 2024Addressing Polarization and Unfairness in Performative Prediction24 Jun 2024
- 2024CherryRec: Enhancing News Recommendation Quality via LLM-driven Framework18 Jun 2024
- 2024Contextual Optimization under Covariate Shift: A Robust Approach by Intersecting Wasserstein Balls4 Jun 2024
- 2024ExMap: Leveraging Explainability Heatmaps for Unsupervised Group Robustness to Spurious Correlations20 Mar 2024
- 2024Federated Fairness without Access to Sensitive Groups22 Feb 2024
- 2023Geometry-Aware Normalizing Wasserstein Flows for Optimal Causal Inference30 Nov 2023
- 2023Geometry-Calibrated DRO: Combating Over-Pessimism with Free Energy Implications8 Nov 2023
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coverage27 Oct 2023
- 2023Is this model reliable for everyone? Testing for strong calibration28 Jul 2023
- 2023Scaff-PD: Communication Efficient Fair and Robust Federated Learning25 Jul 2023
- 2023Causality-oriented robustness: exploiting general noise interventions18 Jul 2023
- 2023Optimizer's Information Criterion: Dissecting and Correcting Bias in Data-Driven Optimization16 Jun 2023
- 2023Robust Data-driven Prescriptiveness Optimization9 Jun 2023
- 2023Predictive Heterogeneity: Measures and Applications1 Apr 2023
- 2023Diagnosing Model Performance Under Distribution Shift3 Mar 2023
- 2023Meta-Learning Mini-Batch Risk Functionals27 Jan 2023
- 2022Minimax Optimal Estimation of Stability Under Distribution Shift13 Dec 2022
- 2022Subgroup Robustness Grows On Trees: An Empirical Baseline Investigation23 Nov 2022
- 2022Distributionally Robust Survival Analysis: A Novel Fairness Loss Without Demographics18 Nov 2022
- 2022Learning from a Biased Sample5 Sep 2022
- 2022Long Term Fairness for Minority Groups via Performative Distributionally Robust Optimization12 Jul 2022
- 2022Understanding the effect of sparsity on neural networks robustness22 Jun 2022
- 2022Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalization15 Jun 2022
- 2022How does overparametrization affect performance on minority groups?7 Jun 2022
- 2022Evaluating Robustness to Dataset Shift via Parametric Robustness Sets31 May 2022
- 2022NICO++: Towards Better Benchmarking for Domain Generalization17 Apr 2022
- 2022Perfectly Balanced: Improving Transfer and Robustness of Supervised Contrastive Learning15 Apr 2022
- 2022Distributionally Robust Models with Parametric Likelihood Ratios13 Apr 2022
- 2021Simple and near-optimal algorithms for hidden stratification and multi-group learning22 Dec 2021
- 2021A Theoretical Analysis on Independence-driven Importance Weighting for Covariate-shift Generalization3 Nov 2021
- 2021Learning Neural Models for Natural Language Processing in the Face of Distributional Shift3 Sep 2021
- 2021Towards Out-Of-Distribution Generalization: A Survey31 Aug 2021
- 2021A comparison of approaches to improve worst-case predictive model performance over patient subpopulations27 Aug 2021
- 2021Unsupervised Learning of Debiased Representations with Pseudo-Attributes6 Aug 2021
50 newest of 57 All 57 (JSON)
- 2021Towards Out-Of-Distribution Generalization: A Surveycited by 431
- 2022NICO++: Towards Better Benchmarking for Domain Generalizationcited by 86
- 2021The Spotlight: A General Method for Discovering Systematic Errors in Deep Learning Modelscited by 63
- 2021Unsupervised Learning of Debiased Representations with Pseudo-Attributescited by 44
- 2022Perfectly Balanced: Improving Transfer and Robustness of Supervised Contrastive Learningcited by 40
- 2022Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalizationcited by 38
- 2023Diagnosing Model Performance Under Distribution Shiftcited by 37
- 2021An Online Learning Approach to Interpolation and Extrapolation in Domain Generalizationcited by 30
- 2021Simple and near-optimal algorithms for hidden stratification and multi-group learningcited by 26
- 2021Regularizing towards Causal Invariance: Linear Models with Proxiescited by 20
- 2021A Theoretical Analysis on Independence-driven Importance Weighting for Covariate-shift Generalizationcited by 18
- 2022Distributionally Robust Models with Parametric Likelihood Ratioscited by 17
12 most cited of 56 All 56 (JSON)
- 2026The Privacy Price of Tail-Risk Learning: Effective Tail Sample Size in Differentially Private CVaR Optimization15 May 2026
- 2026Deep Learning for Sequential Decision Making under Uncertainty: Foundations, Frameworks, and Frontiers13 Apr 2026
- 2025Sample-Efficient Omniprediction for Proper Losses14 Oct 2025
- 2025Distributionally Robust Federated Learning with Outlier Resilience29 Sep 2025
- 2025ACE and Diverse Generalization via Selective Disagreement9 Sep 2025
- 2025Data Heterogeneity Modeling for Trustworthy Machine Learning1 Jun 2025
- 2025DRO: A Python Library for Distributionally Robust Optimization in Machine Learning29 May 2025
- 2025DivIL: Unveiling and Addressing Over-Invariance for Out-of- Distribution Generalization18 Feb 2025
- 2025Task Generalization With AutoRegressive Compositional Structure: Can Learning From D Tasks Generalize to DT Tasks?13 Feb 2025
- 2025CTC-DRO: Robust Optimization for Reducing Language Disparities in Speech Recognition3 Feb 2025
- 2024DRPruning: Efficient Large Language Model Pruning through Distributionally Robust Optimization21 Nov 2024
- 2024Minimax rates of convergence for nonparametric regression under adversarial attacks12 Oct 2024
12 newest of 56 All 56 (JSON)