mlsanityData to 14 Oct 2026

Bias as a Virtue: Rethinking Generalization under Distribution Shifts

arXiv 2506.00407 ↗First posted 31 May 2025cs.LG cs.AI stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Machine learning models often degrade when deployed on data distributions different from their training data. Challenging conventional validation paradigms, we demonstrate that higher in-distribution (ID) bias can lead to better out-of-distribution (OOD) generalization. Our Adaptive Distribution Bridge (ADB) framework implements this insight by introducing controlled statistical diversity during training, enabling models to develop bias profiles that effectively generalize across distributions. Empirically, we observe a robust negative correlation where higher ID bias corresponds to lower OOD error--a finding that contradicts standard practices focused on minimizing validation error. Evaluation on multiple datasets shows our approach significantly improves OOD generalization. ADB achieves robust mean error reductions of up to 26.8% compared to traditional cross-validation, and consistently identifies high-performing training strategies, evidenced by percentile ranks often exceeding 74.4%. Our work provides both a practical method for improving generalization and a theoretical framework for reconsidering the role of bias in robust machine learning.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 22 of its 39 references that are in the map. First posted on arXiv 31 May 2025.

Builds on

22 of 39 references.

References · 22

  1. 2017Neural Message Passing for Quantum Chemistrycited by 3,487
  2. 2015Domain-Adversarial Training of Neural Networkscited by 3,044
  3. 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
  4. 2019Invariant Risk Minimizationcited by 1,854
  5. 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,375
  6. 2017Adversarial Discriminative Domain Adaptationcited by 1,311
  7. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
  8. 2020In Search of Lost Domain Generalizationcited by 1,054
  9. 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 739
  10. 2015Causal inference using invariant prediction: identification and confidence intervalscited by 536
  11. 2020Underspecification Presents Challenges for Credibility in Modern Machine Learningcited by 496
  12. 2020Measuring Robustness to Natural Distribution Shifts in Image Classificationcited by 454
10 more
  1. 2021Towards Out-Of-Distribution Generalization: A Surveycited by 436
  2. 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 335
  3. 2020An Investigation of Why Overparameterization Exacerbates Spurious Correlationscited by 310
  4. 2019On The Power of Curriculum Learning in Training Deep Networkscited by 286
  5. 2021GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Trainingcited by 274
  6. 2017Automated Curriculum Learning for Neural Networkscited by 257
  7. 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 236
  8. 2023Data-centric Artificial Intelligence: A Surveycited by 203
  9. 2021A Fine-Grained Analysis on Distribution Shiftcited by 171
  10. 2023Improved Complexity Analysis of the Sinkhorn and Greenkhorn Algorithms for Optimal Transportcited by 8

Built on by

Not cited yet.