Understanding Why Generalized Reweighting Does Not Improve Over ERM
Not on the map. Why?
Abstract (arXiv)
Empirical risk minimization (ERM) is known in practice to be non-robust to distributional shift where the training and the test distributions are different. A suite of approaches, such as importance weighting, and variants of distributionally robust optimization (DRO), have been proposed to solve this problem. But a line of recent work has empirically shown that these approaches do not significantly improve over ERM in real applications with distribution shift. The goal of this work is to obtain a comprehensive theoretical understanding of this intriguing phenomenon. We first posit the class of Generalized Reweighting (GRW) algorithms, as a broad category of approaches that iteratively update model parameters based on iterative reweighting of the training samples. We show that when overparameterized models are trained under GRW, the resulting models are close to that obtained by ERM. We also show that adding small regularization which does not greatly affect the empirical training accuracy does not help. Together, our results show that a broad category of what we term GRW approaches are not able to achieve distributionally robust generalization. Our work thus has the following sobering takeaway: to make progress towards distributionally robust generalization, we either have to develop non-GRW approaches, or perhaps devise novel classification/regression loss functions that are adapted to the class of GRW approaches.
Key facts. Cited by 29 papers in the map (data through 7 Oct 2026). Builds on 44 of its 65 references that are in the map. First posted on arXiv 28 Jan 2022.
Builds on
44 of 65 references.
References · 44
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2016Equality of Opportunity in Supervised Learningcited by 2,000
- 2019Invariant Risk Minimizationcited by 1,832
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,366
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,194
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 987
- 2017Counterfactual Fairnesscited by 802
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
- 2018On Lazy Training in Differentiable Programmingcited by 671
32 more
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
- 2020Long-tail learning via logit adjustmentcited by 605
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
- 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 522
- 2016Fairness Beyond Disparate Treatment & Disparate Impact: Learning Classification without Disparate Mistreatmentcited by 368
- 2018Fairness Without Demographics in Repeated Loss Minimizationcited by 363
- 2016Unsupervised Domain Adaptation with Residual Transfer Networkscited by 340
- 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 334
- 2017Unified Deep Supervised Domain Adaptation and Generalizationcited by 322
- 2020An Investigation of Why Overparameterization Exacerbates Spurious Correlationscited by 310
- 2020The Risks of Invariant Risk Minimizationcited by 262
- 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 259
- 2018What is the Effect of Importance Weighting in Deep Learning?cited by 250
- 2015Simultaneous Deep Transfer Across Domains and Taskscited by 205
- 2016Demographic Dialectal Variation in Social Media: A Case Study of African-American Englishcited by 194
- 2021A Fine-Grained Analysis on Distribution Shiftcited by 171
- 2019Distributionally Robust Language Modelingcited by 142
- 2021Contrastive Learning based Hybrid Networks for Long-Tailed Image Classificationcited by 130
- 2021Self-supervised Learning is More Robust to Dataset Imbalancecited by 122
- 2020Rethinking Importance Weighting for Deep Learning under Distribution Shiftcited by 92
- 2021Extending the WILDS Benchmark for Unsupervised Adaptationcited by 85
- 2020Identifying and Compensating for Feature Deviation in Imbalanced Deep Learningcited by 76
- 2021Label-Imbalanced and Group-Sensitive Classification under Overparameterizationcited by 74
- 2021Self Supervision to Distillation for Long-Tailed Visual Recognitioncited by 73
- 2020Gradient descent follows the regularization path for general lossescited by 57
- 2021DORO: Distributional and Outlier Robust Optimizationcited by 54
- 2019Adjusting Decision Boundary for Class Imbalanced Learningcited by 52
- 2022AutoBalance: Optimized Loss Functions for Imbalanced Datacited by 49
- 2021Understanding the role of importance weighting for deep learningcited by 38
- 2020Class-Weighted Classification: Trade-offs and Robust Approachescited by 30
- 2021Is Importance Weighting Incompatible with Interpolating Classifiers?cited by 19
- 2021Boosted CVaR Classificationcited by 12
Built on by
Citing papers per year: 2022: 5; 2023: 5; 2024: 10; 2025: 7; 2026: 2.
29 papers.
Sort:
By area:
- 2022Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalizationcited by 38
- 2022UMIX: Improving Importance Weighting for Subpopulation Shift via Uncertainty-Aware Mixupcited by 34
- 2023When is Importance Weighting Correction Needed for Covariate Shift Adaptation?cited by 11
- 2024Controllable Prompt Tuning For Balancing Group Distributional Robustnesscited by 10
- 2022On the Alignment of Group Fairness with Attribute Privacycited by 9
- 2022Importance Tempering: Group Robustness for Overparameterized Modelscited by 6
- 2022Malign Overfitting: Interpolation Can Provably Preclude Invariancecited by 6
- 2023SoK: Unintended Interactions among Machine Learning Defenses and Riskscited by 6
- 2025Group-robust Sample Reweighting for Subpopulation Shifts via Influence Functionscited by 6
- 2024High-Dimensional Kernel Methods under Covariate Shift: Data-Dependent Implicit Regularizationcited by 4
- 2024Edge Classification on Graphs: New Directions in Topological Imbalancecited by 4
- 2024Upsample or Upweight? Balanced Training on Heavily Imbalanced Datasetscited by 4
17 more
- 2023Stubborn Lexical Bias in Data and Modelscited by 3
- 2024Changing the Training Data Distribution to Reduce Simplicity Bias Improves In-distribution Generalizationcited by 3
- 2023Reweighted Mixup for Subpopulation Shiftcited by 2
- 2025Implicit Bias of Per-sample Adam on Separable Data: Departure from the Full-batch Regimecited by 2
- 2025Bilevel Data Curation for LLM Fine-tuning: Offline Selection and Online Self-Refining Generationcited by 2
- 2025What Is Preference Optimization Doing, and Why?cited by 2
- 2023Multitask Learning Can Improve Worst-Group Outcomescited by 1
- 2024Cross-modality debiasing: using language to mitigate sub-population shifts in imagingcited by 1
- 2024Multiplicative Logit Adjustment Approximates Neural-Collapse-Aware Decision Boundary Adjustmentcited by 1
- 2025Reweighting Improves Conditional Risk Boundscited by 1
- 2025LT-Soups: Bridging Head and Tail Classes via Subsampled Model Soupscited by 1
- 2026Why Loss Re-weighting Works If You Stop Early: Training Dynamics of Unconstrained Featurescited by 1
- 2024Towards a Better Evaluation of Out-of-Domain Generalizationcited by 0
- 2024Ultra-imbalanced classification guided by statistical informationcited by 0
- 2024Unsupervised Domain Adaptation Via Data Pruningcited by 0
- 2025Revisiting Meta-Learning with Noisy Labels: Reweighting Dynamics and Theoretical Guaranteescited by 0
- 2026Entropic Projection Alignment: Estimating, Explaining, and Improving Model Performance Under Distribution Shiftcited by 0
- 2026Entropic Projection Alignment: Estimating, Explaining, and Improving Model Performance Under Distribution Shift29 May 2026
- 2026Why Loss Re-weighting Works If You Stop Early: Training Dynamics of Unconstrained Features17 Jan 2026
- 2025What Is Preference Optimization Doing, and Why?30 Nov 2025
- 2025Bilevel Data Curation for LLM Fine-tuning: Offline Selection and Online Self-Refining Generation26 Nov 2025
- 2025LT-Soups: Bridging Head and Tail Classes via Subsampled Model Soups11 Nov 2025
- 2025Implicit Bias of Per-sample Adam on Separable Data: Departure from the Full-batch Regime30 Oct 2025
- 2025Revisiting Meta-Learning with Noisy Labels: Reweighting Dynamics and Theoretical Guarantees14 Oct 2025
- 2025Group-robust Sample Reweighting for Subpopulation Shifts via Influence Functions10 Mar 2025
- 2025Reweighting Improves Conditional Risk Bounds4 Jan 2025
- 2024Upsample or Upweight? Balanced Training on Heavily Imbalanced Datasets6 Oct 2024
- 2024Multiplicative Logit Adjustment Approximates Neural-Collapse-Aware Decision Boundary Adjustment26 Sep 2024
- 2024Unsupervised Domain Adaptation Via Data Pruning18 Sep 2024
17 more
- 2024Ultra-imbalanced classification guided by statistical information6 Sep 2024
- 2024Edge Classification on Graphs: New Directions in Topological Imbalance17 Jun 2024
- 2024High-Dimensional Kernel Methods under Covariate Shift: Data-Dependent Implicit Regularization5 Jun 2024
- 2024Towards a Better Evaluation of Out-of-Domain Generalization30 May 2024
- 2024Changing the Training Data Distribution to Reduce Simplicity Bias Improves In-distribution Generalization27 Apr 2024
- 2024Controllable Prompt Tuning For Balancing Group Distributional Robustness5 Mar 2024
- 2024Cross-modality debiasing: using language to mitigate sub-population shifts in imaging2 Feb 2024
- 2023SoK: Unintended Interactions among Machine Learning Defenses and Risks7 Dec 2023
- 2023Multitask Learning Can Improve Worst-Group Outcomes5 Dec 2023
- 2023Stubborn Lexical Bias in Data and Models3 Jun 2023
- 2023Reweighted Mixup for Subpopulation Shift9 Apr 2023
- 2023When is Importance Weighting Correction Needed for Covariate Shift Adaptation?7 Mar 2023
- 2022Malign Overfitting: Interpolation Can Provably Preclude Invariance28 Nov 2022
- 2022On the Alignment of Group Fairness with Attribute Privacy18 Nov 2022
- 2022UMIX: Improving Importance Weighting for Subpopulation Shift via Uncertainty-Aware Mixup19 Sep 2022
- 2022Importance Tempering: Group Robustness for Overparameterized Models19 Sep 2022
- 2022Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalization15 Jun 2022
- 2025What Is Preference Optimization Doing, and Why?cited by 2
- 2025What Is Preference Optimization Doing, and Why?30 Nov 2025
- 2022Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalizationcited by 38
- 2022UMIX: Improving Importance Weighting for Subpopulation Shift via Uncertainty-Aware Mixupcited by 34
- 2023When is Importance Weighting Correction Needed for Covariate Shift Adaptation?cited by 11
- 2024Controllable Prompt Tuning For Balancing Group Distributional Robustnesscited by 10
- 2022On the Alignment of Group Fairness with Attribute Privacycited by 9
- 2022Importance Tempering: Group Robustness for Overparameterized Modelscited by 6
- 2022Malign Overfitting: Interpolation Can Provably Preclude Invariancecited by 6
- 2023SoK: Unintended Interactions among Machine Learning Defenses and Riskscited by 6
- 2025Group-robust Sample Reweighting for Subpopulation Shifts via Influence Functionscited by 6
- 2024High-Dimensional Kernel Methods under Covariate Shift: Data-Dependent Implicit Regularizationcited by 4
- 2024Edge Classification on Graphs: New Directions in Topological Imbalancecited by 4
- 2024Upsample or Upweight? Balanced Training on Heavily Imbalanced Datasetscited by 4
12 most cited of 28 All 28 (JSON)
- 2026Entropic Projection Alignment: Estimating, Explaining, and Improving Model Performance Under Distribution Shift29 May 2026
- 2026Why Loss Re-weighting Works If You Stop Early: Training Dynamics of Unconstrained Features17 Jan 2026
- 2025Bilevel Data Curation for LLM Fine-tuning: Offline Selection and Online Self-Refining Generation26 Nov 2025
- 2025LT-Soups: Bridging Head and Tail Classes via Subsampled Model Soups11 Nov 2025
- 2025Implicit Bias of Per-sample Adam on Separable Data: Departure from the Full-batch Regime30 Oct 2025
- 2025Revisiting Meta-Learning with Noisy Labels: Reweighting Dynamics and Theoretical Guarantees14 Oct 2025
- 2025Group-robust Sample Reweighting for Subpopulation Shifts via Influence Functions10 Mar 2025
- 2025Reweighting Improves Conditional Risk Bounds4 Jan 2025
- 2024Upsample or Upweight? Balanced Training on Heavily Imbalanced Datasets6 Oct 2024
- 2024Multiplicative Logit Adjustment Approximates Neural-Collapse-Aware Decision Boundary Adjustment26 Sep 2024
- 2024Unsupervised Domain Adaptation Via Data Pruning18 Sep 2024
- 2024Ultra-imbalanced classification guided by statistical information6 Sep 2024
12 newest of 28 All 28 (JSON)