Why does Throwing Away Data Improve Worst-Group Error?
Kamalika Chaudhuri, Kartik Ahuja, Martín Arjovsky, David Lopez-Paz
Abstract
When facing data with imbalanced classes or groups, practitioners follow an intriguing strategy to achieve best results. They throw away examples until the classes or groups are balanced in size, and then perform empirical risk minimization on the reduced training set. This opposes common wisdom in learning theory, where the expected error is supposed to decrease as the dataset grows in size. In this work, we leverage extreme value theory to address this apparent contradiction. Our results show that the tails of the data distribution play an important role in determining the worst-group-accuracy of linear classifiers. When learning on data with heavy tails, throwing away data restores the geometric symmetry of the resulting classifier, and therefore improves its worst-group generalization.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers9
- Restoring balance: principled under/oversampling of data for optimal classificationEmanuele Loffredo, Mauro Pastore, Simona Cocco, Rémi MonassonICML 2024 · 13 citations
- The Group Robustness is in the Details: Revisiting Finetuning under Spurious CorrelationsTyler LaBonte, John C. Hill, Xinchen Zhang, Vidya Muthukumar et al.NeurIPS 2024 · 8 citations
- Understanding Domain Generalization: A Noise Robustness PerspectiveRui Qiao, Bryan Kian Hsiang LowICLR 2024 · 8 citations
- Balanced Data, Imbalanced Spectra: Unveiling Class Disparities with Spectral ImbalanceChiraag Kaushik, Ran Liu, Chi-Heng Lin, Amrit Khera et al.ICML 2024 · 4 citations
- Enhancing Robustness of Last Layer Two-Stage Fair Model CorrectionsNathan Stromberg, Rohan Ayyagari, Sanmi Koyejo, Richard Nock et al.NeurIPS 2024 · 3 citations
Builds on2
Related papers
- Learning to Reject Meets Long-tail LearningHarikrishna Narasimhan, Aditya Krishna Menon, Wittawat Jitkrittum, Neha Gupta et al.ICLR 2024 · 7 citations
- Cross-Domain Empirical Risk Minimization for Unbiased Long-Tailed ClassificationBeier Zhu, Yulei Niu, Xian-Sheng Hua, Hanwang ZhangAAAI 2022 · 50 citations
- Long-Tailed Learning as Multi-Objective OptimizationWeiqi Li, Fan Lyu, Fanhua Shang, Liang Wan et al.AAAI 2024 · 10 citations
- SSE-SAM: Balancing Head and Tail Classes Gradually Through Stage-Wise SAMXingyu Lyu, Qianqian Xu, Zhiyong Yang, Shaojie Lyu et al.AAAI 2025 · 2 citations
- When do Minimax-fair Learning and Empirical Risk Minimization Coincide?Harvineet Singh, Matthäus Kleindessner, Volkan Cevher, Rumi Chunara et al.ICML 2023 · 6 citations
