On the Limitations of Dataset Balancing: The Lost Battle Against Spurious Correlations
Not on the map. Why?
Abstract (arXiv)
Recent work has shown that deep learning models in NLP are highly sensitive to low-level correlations between simple features and specific output labels, leading to overfitting and lack of generalization. To mitigate this problem, a common practice is to balance datasets by adding new instances or by filtering out "easy" instances (Sakaguchi et al., 2020), culminating in a recent proposal to eliminate single-word correlations altogether (Gardner et al., 2021). In this opinion paper, we identify that despite these efforts, increasingly-powerful models keep exploiting ever-smaller spurious correlations, and as a result even balancing all single-word features is insufficient for mitigating all of these correlations. In parallel, a truly balanced dataset may be bound to "throw the baby out with the bathwater" and miss important signal encoding common sense and world knowledge. We highlight several alternatives to dataset balancing, focusing on enhancing datasets with richer contexts, allowing models to abstain and interact with users, and turning from large-scale fine-tuning to zero- or few-shot setups.
Key facts. Cited by 19 papers in the map (data through 14 Oct 2026). Builds on 63 of its 107 references that are in the map. First posted on arXiv 27 Apr 2022.
Builds on
63 of 107 references.
References · 63
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2017Axiomatic Attribution for Deep Networkscited by 3,384
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2016Making the V in VQA Matter: Elevating the Role of Image Understanding in Visual Question Answeringcited by 2,516
- 2015VQA: Visual Question Answeringcited by 2,506
- 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,371
38 more
- 2018Deep contextualized word representationscited by 2,290
- 2021Pre-train, Prompt, and Predict: A Systematic Survey of Prompting Methods in Natural Language Processingcited by 2,234
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,181
- 2015A large annotated corpus for learning natural language inferencecited by 1,847
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
- 2020RealToxicityPrompts: Evaluating Neural Toxic Degeneration in Language Modelscited by 1,255
- 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,254
- 2021DeBERTaV3: Improving DeBERTa using ELECTRA-Style Pre-Training with Gradient-Disentangled Embedding Sharingcited by 1,254
- 2017SmoothGrad: removing noise by adding noisecited by 1,109
- 2019DROP: A Reading Comprehension Benchmark Requiring Discrete Reasoning Over Paragraphscited by 903
- 2020How Much Knowledge Can You Pack Into the Parameters of a Language Model?cited by 719
- 2021Deduplicating Training Data Makes Language Models Bettercited by 714
- 2018Annotation Artifacts in Natural Language Inference Datacited by 671
- 2020It's Not Just Size That Matters: Small Language Models Are Also Few-Shot Learnerscited by 669
- 2019Green AIcited by 627
- 2019ERNIE: Enhanced Language Representation with Informative Entitiescited by 528
- 2021Documenting Large Webtext Corpora: A Case Study on the Colossal Clean Crawled Corpuscited by 492
- 2020Dataset Cartography: Mapping and Diagnosing Datasets with Training Dynamicscited by 425
- 2017Men Also Like Shopping: Reducing Gender Bias Amplification using Corpus-level Constraintscited by 420
- 2021AI and the Everything in the Whole Wide World Benchmarkcited by 349
- 2018SWAG: A Large-Scale Adversarial Dataset for Grounded Commonsense Inferencecited by 333
- 2021Multimodal datasets: misogyny, pornography, and malignant stereotypescited by 327
- 2019Abductive Commonsense Reasoningcited by 301
- 2019Knowledge Enhanced Contextual Word Representationscited by 296
- 2018Hypothesis Only Baselines in Natural Language Inferencecited by 295
- 2019Evaluating Gender Bias in Machine Translationcited by 249
- 2020Selective Question Answering under Domain Shiftcited by 238
- 2021PPT: Pre-trained Prompt Tuning for Few-shot Learningcited by 232
- 2018ReCoRD: Bridging the Gap between Human and Machine Commonsense Reading Comprehensioncited by 200
- 2020Adversarial Filters of Dataset Biasescited by 160
- 2020With Little Power Comes Great Responsibilitycited by 142
- 2018Breaking NLI Systems with Sentences that Require Simple Lexical Inferencescited by 134
- 2021Competency Problems: On Finding and Removing Artifacts in Language Datacited by 91
- 2020Back to the Future: Unsupervised Backprop-based Decoding for Counterfactual and Abductive Commonsense Reasoningcited by 65
- 2020Identifying Spurious Correlations for Robust Text Classificationcited by 59
- 2021Provable Limitations of Acquiring Meaning from Ungrounded Form: What Will Future Language Models Understand?cited by 55
- 2021Changing the World by Changing the Datacited by 51
- 2021Knowing More About Questions Can Help: Improving Calibration in Question Answeringcited by 47
50 most cited of 63 All 63 (JSON)
Built on by
Citing papers per year: 2022: 2; 2023: 6; 2024: 6; 2025: 5.
19 papers.
Sort:
- 2022Are All Spurious Features in Natural Language Alike? An Analysis through a Causal Lenscited by 28
- 2023TTIDA: Controllable Generative Data Augmentation via Text-to-Text and Text-to-Image Modelscited by 14
- 2023Fighting Bias with Bias: Promoting Model Robustness by Amplifying Dataset Biasescited by 5
- 2024Data Debiasing with Datamodels (D3M): Improving Subgroup Robustness via Data Selectioncited by 5
- 2023Principles from Clinical Research for NLP Model Generalizationcited by 4
- 2024Compare without Despair: Reliable Preference Evaluation with Generation Separabilitycited by 4
- 2025Beyond Benchmarks: On The False Promise of AI Regulationcited by 4
- 2025The Impact of Coreset Selection on Spurious Correlations and Group Robustnesscited by 4
- 2022Looking at the Overlooked: An Analysis on the Word-Overlap Bias in Natural Language Inferencecited by 3
- 2023Comparing Humans and Models on a Similar Scale: Towards Cognitive Gender Bias Evaluation in Coreference Resolutioncited by 3
- 2023Stubborn Lexical Bias in Data and Modelscited by 3
- 2024Optimizing importance weighting in the presence of sub-population shiftscited by 3
7 more
- 2025HBO: Hierarchical Balancing Optimization for Fine-Tuning Large Language Modelscited by 2
- 2023On Degrees of Freedom in Defining and Testing Natural Language Understandingcited by 1
- 2025Cross-modal Associations in Vision and Language Models: Revisiting the Bouba-Kiki Effectcited by 1
- 2024Characterizing Stereotypical Bias from Privacy-preserving Pre-Trainingcited by 0
- 2024Effective Guidance for Model Attention with Simple Yes-no Annotationscited by 0
- 2024On the Shortcut Learning in Multilingual Neural Machine Translationcited by 0
- 2025Representation Invariance and Allocation: When Subgroup Balance Matterscited by 0
- 2025Representation Invariance and Allocation: When Subgroup Balance Matters10 Dec 2025
- 2025The Impact of Coreset Selection on Spurious Correlations and Group Robustness15 Jul 2025
- 2025Cross-modal Associations in Vision and Language Models: Revisiting the Bouba-Kiki Effect14 Jul 2025
- 2025HBO: Hierarchical Balancing Optimization for Fine-Tuning Large Language Models18 May 2025
- 2025Beyond Benchmarks: On The False Promise of AI Regulation26 Jan 2025
- 2024On the Shortcut Learning in Multilingual Neural Machine Translation15 Nov 2024
- 2024Effective Guidance for Model Attention with Simple Yes-no Annotations29 Oct 2024
- 2024Optimizing importance weighting in the presence of sub-population shifts18 Oct 2024
- 2024Compare without Despair: Reliable Preference Evaluation with Generation Separability2 Jul 2024
- 2024Characterizing Stereotypical Bias from Privacy-preserving Pre-Training30 Jun 2024
- 2024Data Debiasing with Datamodels (D3M): Improving Subgroup Robustness via Data Selection24 Jun 2024
- 2023Principles from Clinical Research for NLP Model Generalization7 Nov 2023
7 more
- 2023Stubborn Lexical Bias in Data and Models3 Jun 2023
- 2023Fighting Bias with Bias: Promoting Model Robustness by Amplifying Dataset Biases30 May 2023
- 2023Comparing Humans and Models on a Similar Scale: Towards Cognitive Gender Bias Evaluation in Coreference Resolution24 May 2023
- 2023On Degrees of Freedom in Defining and Testing Natural Language Understanding24 May 2023
- 2023TTIDA: Controllable Generative Data Augmentation via Text-to-Text and Text-to-Image Models18 Apr 2023
- 2022Looking at the Overlooked: An Analysis on the Word-Overlap Bias in Natural Language Inference7 Nov 2022
- 2022Are All Spurious Features in Natural Language Alike? An Analysis through a Causal Lens25 Oct 2022
- 2022Are All Spurious Features in Natural Language Alike? An Analysis through a Causal Lenscited by 28
- 2023TTIDA: Controllable Generative Data Augmentation via Text-to-Text and Text-to-Image Modelscited by 14
- 2023Fighting Bias with Bias: Promoting Model Robustness by Amplifying Dataset Biasescited by 5
- 2024Data Debiasing with Datamodels (D3M): Improving Subgroup Robustness via Data Selectioncited by 5
- 2023Principles from Clinical Research for NLP Model Generalizationcited by 4
- 2024Compare without Despair: Reliable Preference Evaluation with Generation Separabilitycited by 4
- 2025Beyond Benchmarks: On The False Promise of AI Regulationcited by 4
- 2025The Impact of Coreset Selection on Spurious Correlations and Group Robustnesscited by 4
- 2022Looking at the Overlooked: An Analysis on the Word-Overlap Bias in Natural Language Inferencecited by 3
- 2023Comparing Humans and Models on a Similar Scale: Towards Cognitive Gender Bias Evaluation in Coreference Resolutioncited by 3
- 2023Stubborn Lexical Bias in Data and Modelscited by 3
- 2024Optimizing importance weighting in the presence of sub-population shiftscited by 3
12 most cited of 19 All 19 (JSON)
- 2025Representation Invariance and Allocation: When Subgroup Balance Matters10 Dec 2025
- 2025The Impact of Coreset Selection on Spurious Correlations and Group Robustness15 Jul 2025
- 2025Cross-modal Associations in Vision and Language Models: Revisiting the Bouba-Kiki Effect14 Jul 2025
- 2025HBO: Hierarchical Balancing Optimization for Fine-Tuning Large Language Models18 May 2025
- 2025Beyond Benchmarks: On The False Promise of AI Regulation26 Jan 2025
- 2024On the Shortcut Learning in Multilingual Neural Machine Translation15 Nov 2024
- 2024Effective Guidance for Model Attention with Simple Yes-no Annotations29 Oct 2024
- 2024Optimizing importance weighting in the presence of sub-population shifts18 Oct 2024
- 2024Compare without Despair: Reliable Preference Evaluation with Generation Separability2 Jul 2024
- 2024Characterizing Stereotypical Bias from Privacy-preserving Pre-Training30 Jun 2024
- 2024Data Debiasing with Datamodels (D3M): Improving Subgroup Robustness via Data Selection24 Jun 2024
- 2023Principles from Clinical Research for NLP Model Generalization7 Nov 2023
12 newest of 19 All 19 (JSON)