mlsanityData to 14 Oct 2026

On the Limitations of Dataset Balancing: The Lost Battle Against Spurious Correlations

arXiv 2204.12708 ↗First posted 27 Apr 2022cs.CLcited by 19

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

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  4. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
  5. 2017On Calibration of Modern Neural Networkscited by 4,960
  6. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
  7. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  8. 2017Axiomatic Attribution for Deep Networkscited by 3,384
  9. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
  10. 2016Making the V in VQA Matter: Elevating the Role of Image Understanding in Visual Question Answeringcited by 2,516
  11. 2015VQA: Visual Question Answeringcited by 2,506
  12. 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,371
38 more
  1. 2018Deep contextualized word representationscited by 2,290
  2. 2021Pre-train, Prompt, and Predict: A Systematic Survey of Prompting Methods in Natural Language Processingcited by 2,234
  3. 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,181
  4. 2015A large annotated corpus for learning natural language inferencecited by 1,847
  5. 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
  6. 2020RealToxicityPrompts: Evaluating Neural Toxic Degeneration in Language Modelscited by 1,255
  7. 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,254
  8. 2021DeBERTaV3: Improving DeBERTa using ELECTRA-Style Pre-Training with Gradient-Disentangled Embedding Sharingcited by 1,254
  9. 2017SmoothGrad: removing noise by adding noisecited by 1,109
  10. 2019DROP: A Reading Comprehension Benchmark Requiring Discrete Reasoning Over Paragraphscited by 903
  11. 2020How Much Knowledge Can You Pack Into the Parameters of a Language Model?cited by 719
  12. 2021Deduplicating Training Data Makes Language Models Bettercited by 714
  13. 2018Annotation Artifacts in Natural Language Inference Datacited by 671
  14. 2020It's Not Just Size That Matters: Small Language Models Are Also Few-Shot Learnerscited by 669
  15. 2019Green AIcited by 627
  16. 2019ERNIE: Enhanced Language Representation with Informative Entitiescited by 528
  17. 2021Documenting Large Webtext Corpora: A Case Study on the Colossal Clean Crawled Corpuscited by 492
  18. 2020Dataset Cartography: Mapping and Diagnosing Datasets with Training Dynamicscited by 425
  19. 2017Men Also Like Shopping: Reducing Gender Bias Amplification using Corpus-level Constraintscited by 420
  20. 2021AI and the Everything in the Whole Wide World Benchmarkcited by 349
  21. 2018SWAG: A Large-Scale Adversarial Dataset for Grounded Commonsense Inferencecited by 333
  22. 2021Multimodal datasets: misogyny, pornography, and malignant stereotypescited by 327
  23. 2019Abductive Commonsense Reasoningcited by 301
  24. 2019Knowledge Enhanced Contextual Word Representationscited by 296
  25. 2018Hypothesis Only Baselines in Natural Language Inferencecited by 295
  26. 2019Evaluating Gender Bias in Machine Translationcited by 249
  27. 2020Selective Question Answering under Domain Shiftcited by 238
  28. 2021PPT: Pre-trained Prompt Tuning for Few-shot Learningcited by 232
  29. 2018ReCoRD: Bridging the Gap between Human and Machine Commonsense Reading Comprehensioncited by 200
  30. 2020Adversarial Filters of Dataset Biasescited by 160
  31. 2020With Little Power Comes Great Responsibilitycited by 142
  32. 2018Breaking NLI Systems with Sentences that Require Simple Lexical Inferencescited by 134
  33. 2021Competency Problems: On Finding and Removing Artifacts in Language Datacited by 91
  34. 2020Back to the Future: Unsupervised Backprop-based Decoding for Counterfactual and Abductive Commonsense Reasoningcited by 65
  35. 2020Identifying Spurious Correlations for Robust Text Classificationcited by 59
  36. 2021Provable Limitations of Acquiring Meaning from Ungrounded Form: What Will Future Language Models Understand?cited by 55
  37. 2021Changing the World by Changing the Datacited by 51
  38. 2021Knowing More About Questions Can Help: Improving Calibration in Question Answeringcited by 47

50 most cited of 63 All 63 (JSON)

Built on by

Cited by year

Citing papers per year: 2022: 2; 2023: 6; 2024: 6; 2025: 5.

19 papers.

Sort:

  1. 2022Are All Spurious Features in Natural Language Alike? An Analysis through a Causal Lenscited by 28
  2. 2023TTIDA: Controllable Generative Data Augmentation via Text-to-Text and Text-to-Image Modelscited by 14
  3. 2023Fighting Bias with Bias: Promoting Model Robustness by Amplifying Dataset Biasescited by 5
  4. 2024Data Debiasing with Datamodels (D3M): Improving Subgroup Robustness via Data Selectioncited by 5
  5. 2023Principles from Clinical Research for NLP Model Generalizationcited by 4
  6. 2024Compare without Despair: Reliable Preference Evaluation with Generation Separabilitycited by 4
  7. 2025Beyond Benchmarks: On The False Promise of AI Regulationcited by 4
  8. 2025The Impact of Coreset Selection on Spurious Correlations and Group Robustnesscited by 4
  9. 2022Looking at the Overlooked: An Analysis on the Word-Overlap Bias in Natural Language Inferencecited by 3
  10. 2023Comparing Humans and Models on a Similar Scale: Towards Cognitive Gender Bias Evaluation in Coreference Resolutioncited by 3
  11. 2023Stubborn Lexical Bias in Data and Modelscited by 3
  12. 2024Optimizing importance weighting in the presence of sub-population shiftscited by 3
7 more
  1. 2025HBO: Hierarchical Balancing Optimization for Fine-Tuning Large Language Modelscited by 2
  2. 2023On Degrees of Freedom in Defining and Testing Natural Language Understandingcited by 1
  3. 2025Cross-modal Associations in Vision and Language Models: Revisiting the Bouba-Kiki Effectcited by 1
  4. 2024Characterizing Stereotypical Bias from Privacy-preserving Pre-Trainingcited by 0
  5. 2024Effective Guidance for Model Attention with Simple Yes-no Annotationscited by 0
  6. 2024On the Shortcut Learning in Multilingual Neural Machine Translationcited by 0
  7. 2025Representation Invariance and Allocation: When Subgroup Balance Matterscited by 0