Towards Interpreting and Mitigating Shortcut Learning Behavior of NLU Models
Not on the map. Why?
Abstract (arXiv)
Recent studies indicate that NLU models are prone to rely on shortcut features for prediction, without achieving true language understanding. As a result, these models fail to generalize to real-world out-of-distribution data. In this work, we show that the words in the NLU training set can be modeled as a long-tailed distribution. There are two findings: 1) NLU models have strong preference for features located at the head of the long-tailed distribution, and 2) Shortcut features are picked up during very early few iterations of the model training. These two observations are further employed to formulate a measurement which can quantify the shortcut degree of each training sample. Based on this shortcut measurement, we propose a shortcut mitigation framework LTGR, to suppress the model from making overconfident predictions for samples with large shortcut degree. Experimental results on three NLU benchmarks demonstrate that our long-tailed distribution explanation accurately reflects the shortcut learning behavior of NLU models. Experimental analysis further indicates that LTGR can improve the generalization accuracy on OOD data, while preserving the accuracy on in-distribution data.
Key facts. Cited by 78 papers in the map (data through 7 Oct 2026). Builds on 48 of its 52 references that are in the map. First posted on arXiv 11 Mar 2021.
Builds on
48 of 52 references.
References · 48
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2017Axiomatic Attribution for Deep Networkscited by 3,350
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,665
- 2020Shortcut Learning in Deep Neural Networkscited by 1,928
- 2018ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustnesscited by 1,430
- 2018FEVER: a large-scale dataset for Fact Extraction and VERificationcited by 1,362
- 2019When Does Label Smoothing Help?cited by 985
- 2019Adversarial Examples Are Not Bugs, They Are Featurescited by 950
- 2019Right for the Wrong Reasons: Diagnosing Syntactic Heuristics in Natural Language Inferencecited by 864
36 more
- 2018Annotation Artifacts in Natural Language Inference Datacited by 656
- 2019Score-CAM: Score-Weighted Visual Explanations for Convolutional Neural Networkscited by 575
- 2017Adversarial Examples for Evaluating Reading Comprehension Systemscited by 568
- 2017Supervised Learning of Universal Sentence Representations from Natural Language Inference Datacited by 428
- 2020The Pitfalls of Simplicity Bias in Neural Networkscited by 349
- 2017Methods for Interpreting and Understanding Deep Neural Networkscited by 342
- 2017Don't Just Assume; Look and Answer: Overcoming Priors for Visual Question Answeringcited by 337
- 2018SWAG: A Large-Scale Adversarial Dataset for Grounded Commonsense Inferencecited by 333
- 2020Weight Poisoning Attacks on Pre-trained Modelscited by 318
- 2019High Frequency Component Helps Explain the Generalization of Convolutional Neural Networkscited by 310
- 2019Don't Take the Easy Way Out: Ensemble Based Methods for Avoiding Known Dataset Biasescited by 279
- 2018Learning Not to Learn: Training Deep Neural Networks with Biased Datacited by 255
- 2018Techniques for Interpretable Machine Learningcited by 237
- 2017Embedding Watermarks into Deep Neural Networkscited by 227
- 2019Probing Neural Network Comprehension of Natural Language Argumentscited by 212
- 2019Learning De-biased Representations with Biased Representationscited by 199
- 2019RUBi: Reducing Unimodal Biases in Visual Question Answeringcited by 183
- 2019Towards Debiasing Fact Verification Modelscited by 143
- 2020An Empirical Study on Robustness to Spurious Correlations using Pre-trained Language Modelscited by 133
- 2019Taking a HINT: Leveraging Explanations to Make Vision and Language Models More Groundedcited by 127
- 2019End-to-End Bias Mitigation by Modelling Biases in Corporacited by 127
- 2019Learning Robust Representations by Projecting Superficial Statistics Outcited by 123
- 2019Unlearn Dataset Bias in Natural Language Inference by Fitting the Residualcited by 119
- 2020Towards Debiasing NLU Models from Unknown Biasescited by 116
- 2020An Embarrassingly Simple Approach for Trojan Attack in Deep Neural Networkscited by 92
- 2017Measuring the tendency of CNNs to Learn Surface Statistical Regularitiescited by 87
- 2020Mind the Trade-off: Debiasing NLU Models without Degrading the In-distribution Performancecited by 85
- 2019[Extended version] Rethinking Deep Neural Network Ownership Verification: Embedding Passports to Defeat Ambiguity Attackscited by 82
- 2018Did the Model Understand the Question?cited by 60
- 2018Knowledgeable Reader: Enhancing Cloze-Style Reading Comprehension with External Commonsense Knowledgecited by 48
- 2018Adversarially Regularising Neural NLI Models to Integrate Logical Background Knowledgecited by 45
- 2018Explicit Bias Discovery in Visual Question Answering Modelscited by 42
- 2017Neural Natural Language Inference Models Enhanced with External Knowledgecited by 40
- 2020Towards Robustifying NLI Models Against Lexical Dataset Biasescited by 34
- 2019What does BERT Learn from Multiple-Choice Reading Comprehension Datasets?cited by 30
- 2020A Unified Taylor Framework for Revisiting Attribution Methodscited by 13
Built on by
Citing papers per year: 2021: 8; 2022: 17; 2023: 22; 2024: 15; 2025: 13; 2026: 3.
78 papers.
Sort:
By area:
- 2023Explainability for Large Language Models: A Surveycited by 331
- 2022State-of-the-art generalisation research in NLP: A taxonomy and reviewcited by 97
- 2023Revisiting Out-of-distribution Robustness in NLP: Benchmark, Analysis, and LLMs Evaluationscited by 96
- 2022Shortcut Learning of Large Language Models in Natural Language Understandingcited by 83
- 2021Identifying and Mitigating Spurious Correlations for Improving Robustness in NLP Modelscited by 76
- 2021"Will You Find These Shortcuts?" A Protocol for Evaluating the Faithfulness of Input Salience Methods for Text Classificationcited by 66
- 2022Detecting Shortcut Learning for Fair Medical AI using Shortcut Testingcited by 52
- 2023Won't Get Fooled Again: Answering Questions with False Premisescited by 47
- 2024Usable XAI: 10 Strategies Towards Exploiting Explainability in the LLM Eracited by 45
- 2025Large Language Models Hallucination: A Comprehensive Surveycited by 45
- 2021The Dangers of Underclaiming: Reasons for Caution When Reporting How NLP Systems Failcited by 38
- 2024InjecGuard: Benchmarking and Mitigating Over-defense in Prompt Injection Guardrail Modelscited by 36
38 more
- 2022PCL: Peer-Contrastive Learning with Diverse Augmentations for Unsupervised Sentence Embeddingscited by 29
- 2023What do neural networks learn in image classification? A frequency shortcut perspectivecited by 28
- 2025Benchmark Designers Should "Train on the Test Set" to Expose Exploitable Non-Visual Shortcutscited by 26
- 2021Introspective Distillation for Robust Question Answeringcited by 25
- 2022ClidSum: A Benchmark Dataset for Cross-Lingual Dialogue Summarizationcited by 25
- 2021Avoiding Inference Heuristics in Few-shot Prompt-based Finetuningcited by 23
- 2022Eye-gaze-guided Vision Transformer for Rectifying Shortcut Learningcited by 20
- 2023Distilling Cognitive Backdoor Patterns within an Imagecited by 20
- 2023A Survey on the Robustness of Computer Vision Models against Common Corruptionscited by 20
- 2024SLANG: New Concept Comprehension of Large Language Modelscited by 20
- 2021Measuring and Improving Model-Moderator Collaboration using Uncertainty Estimationcited by 18
- 2024Disperse-Then-Merge: Pushing the Limits of Instruction Tuning via Alignment Tax Reductioncited by 18
- 2022Does Your Model Classify Entities Reasonably? Diagnosing and Mitigating Spurious Correlations in Entity Typingcited by 16
- 2022Rectify ViT Shortcut Learning by Visual Saliencycited by 14
- 2024UFID: A Unified Framework for Input-level Backdoor Detection on Diffusion Modelscited by 14
- 2025Why and How LLMs Hallucinate: Connecting the Dots with Subsequence Associationscited by 14
- 2022Less Learn Shortcut: Analyzing and Mitigating Learning of Spurious Feature-Label Correlationcited by 13
- 2021Robustness Challenges in Model Distillation and Pruning for Natural Language Understandingcited by 12
- 2023A Comprehensive Survey of Sentence Representations: From the BERT Epoch to the ChatGPT Era and Beyondcited by 11
- 2024Co-occurrence is not Factual Association in Language Modelscited by 11
- 2022Feature-Level Debiased Natural Language Understandingcited by 10
- 2023Robust Natural Language Understanding with Residual Attention Debiasingcited by 10
- 2024Navigating the Shortcut Maze: A Comprehensive Analysis of Shortcut Learning in Text Classification by Language Modelscited by 10
- 2024Evaluating the Performance and Robustness of LLMs in Materials Science Q&A and Property Predictionscited by 9
- 2022XMD: An End-to-End Framework for Interactive Explanation-Based Debugging of NLP Modelscited by 8
- 2023Towards Robust Pruning: An Adaptive Knowledge-Retention Pruning Strategy for Language Modelscited by 7
- 2021Regularization for Long Named Entity Recognitioncited by 5
- 2022Debiasing Stance Detection Models with Counterfactual Reasoning and Adversarial Bias Learningcited by 5
- 2025Short-circuiting Shortcuts: Mechanistic Investigation of Shortcuts in Text Classificationcited by 5
- 2025Physics-informed Temporal Alignment for Auto-regressive PDE Foundation Modelscited by 5
- 2022AutoCAD: Automatically Generating Counterfactuals for Mitigating Shortcut Learningcited by 4
- 2023The Impacts of Unanswerable Questions on the Robustness of Machine Reading Comprehension Modelscited by 4
- 2023A Simple yet Effective Self-Debiasing Framework for Transformer Modelscited by 4
- 2023DFM-X: Augmentation by Leveraging Prior Knowledge of Shortcut Learningcited by 4
- 2024Learning Shortcuts: On the Misleading Promise of NLU in Language Modelscited by 4
- 2025Improving the OOD Performance of Closed-Source LLMs on NLI Through Strategic Data Selectioncited by 4
- 2022Pathologies of Pre-trained Language Models in Few-shot Fine-tuningcited by 3
- 2022Explaining Predictive Uncertainty by Looking Back at Model Explanationscited by 2
50 most cited of 78 All 78 (JSON)
- 2026Calibration vs Decision Making: Revisiting the Reliability Paradox in Unlearned Language Models20 May 2026
- 2026Models Know Their Shortcuts: Deployment-Time Shortcut Mitigation14 Apr 2026
- 2026AI Evaluation Should Require Standardized Item-Level Data Releases27 Feb 2026
- 2025Benchmark Designers Should "Train on the Test Set" to Expose Exploitable Non-Visual Shortcuts6 Nov 2025
- 2025Semantic-Preserving Cross-Style Visual Reasoning for Robust Multi-Modal Understanding in Large Vision-Language Models26 Oct 2025
- 2025Large Language Models Hallucination: A Comprehensive Survey5 Oct 2025
- 2025Can Out-of-Distribution Evaluations Uncover Reliance on Shortcuts? A Case Study in Question Answering25 Aug 2025
- 2025Detecting and Pruning Prominent but Detrimental Neurons in Large Language Models12 Jul 2025
- 2025Mitigating Shortcut Learning with InterpoLated Learning7 Jul 2025
- 2025Not quite Sherlock Holmes: Language model predictions do not reliably differentiate impossible from improbable events7 Jun 2025
- 2025Improving the OOD Performance of Closed-Source LLMs on NLI Through Strategic Data Selection26 May 2025
- 2025Physics-informed Temporal Alignment for Auto-regressive PDE Foundation Models16 May 2025
38 more
- 2025Short-circuiting Shortcuts: Mechanistic Investigation of Shortcuts in Text Classification9 May 2025
- 2025Why and How LLMs Hallucinate: Connecting the Dots with Subsequence Associations17 Apr 2025
- 2025MiMu: Mitigating Multiple Shortcut Learning Behavior of Transformers14 Apr 2025
- 2025DBR: Divergence-Based Regularization for Debiasing Natural Language Understanding Models25 Feb 2025
- 2024InjecGuard: Benchmarking and Mitigating Over-defense in Prompt Injection Guardrail Models30 Oct 2024
- 2024Automated Trustworthiness Oracle Generation for Machine Learning Text Classifiers30 Oct 2024
- 2024Generating Synthetic Datasets for Few-shot Prompt Tuning8 Oct 2024
- 2024Navigating the Shortcut Maze: A Comprehensive Analysis of Shortcut Learning in Text Classification by Language Models26 Sep 2024
- 2024Evaluating the Performance and Robustness of LLMs in Materials Science Q&A and Property Predictions22 Sep 2024
- 2024Co-occurrence is not Factual Association in Language Models21 Sep 2024
- 2024Not Eliminate but Aggregate: Post-Hoc Control over Mixture-of-Experts to Address Shortcut Shifts in Natural Language Understanding17 Jun 2024
- 2024A Nurse is Blue and Elephant is Rugby: Cross Domain Alignment in Large Language Models Reveal Human-like Patterns23 May 2024
- 2024Disperse-Then-Merge: Pushing the Limits of Instruction Tuning via Alignment Tax Reduction22 May 2024
- 2024UFID: A Unified Framework for Input-level Backdoor Detection on Diffusion Models1 Apr 2024
- 2024Usable XAI: 10 Strategies Towards Exploiting Explainability in the LLM Era13 Mar 2024
- 2024Pointing out the Shortcomings of Relation Extraction Models with Semantically Motivated Adversarials29 Feb 2024
- 2024SLANG: New Concept Comprehension of Large Language Models23 Jan 2024
- 2024Learning Shortcuts: On the Misleading Promise of NLU in Language Models17 Jan 2024
- 2024Self-Supervised Position Debiasing for Large Language Models2 Jan 2024
- 2023Modality-Collaborative Transformer with Hybrid Feature Reconstruction for Robust Emotion Recognition26 Dec 2023
- 2023TACIT: A Target-Agnostic Feature Disentanglement Framework for Cross-Domain Text Classification25 Dec 2023
- 2023Adaptive Shortcut Debiasing for Online Continual Learning14 Dec 2023
- 2023IBADR: an Iterative Bias-Aware Dataset Refinement Framework for Debiasing NLU models1 Nov 2023
- 2023Towards Robust Pruning: An Adaptive Knowledge-Retention Pruning Strategy for Language Models19 Oct 2023
- 2023Accurate Use of Label Dependency in Multi-Label Text Classification Through the Lens of Causality11 Oct 2023
- 2023Mitigating Shortcuts in Language Models with Soft Label Encoding17 Sep 2023
- 2023Explainability for Large Language Models: A Survey2 Sep 2023
- 2023Benign Shortcut for Debiasing: Fair Visual Recognition via Intervention with Shortcut Features13 Aug 2023
- 2023DFM-X: Augmentation by Leveraging Prior Knowledge of Shortcut Learning12 Aug 2023
- 2023Toward Zero-Shot Instruction Following4 Aug 2023
- 2023What do neural networks learn in image classification? A frequency shortcut perspective19 Jul 2023
- 2023Won't Get Fooled Again: Answering Questions with False Premises5 Jul 2023
- 2023Revisiting Out-of-distribution Robustness in NLP: Benchmark, Analysis, and LLMs Evaluations7 Jun 2023
- 2023A Simple yet Effective Self-Debiasing Framework for Transformer Models2 Jun 2023
- 2023Robust Natural Language Understanding with Residual Attention Debiasing28 May 2023
- 2023Out-of-Distribution Generalization in Text Classification: Past, Present, and Future23 May 2023
- 2023A Comprehensive Survey of Sentence Representations: From the BERT Epoch to the ChatGPT Era and Beyond22 May 2023
- 2023A Survey on the Robustness of Computer Vision Models against Common Corruptions10 May 2023
50 newest of 78 All 78 (JSON)
- 2024Generating Synthetic Datasets for Few-shot Prompt Tuningcited by 1
- 2024Generating Synthetic Datasets for Few-shot Prompt Tuning8 Oct 2024
- 2023Explainability for Large Language Models: A Surveycited by 331
- 2022State-of-the-art generalisation research in NLP: A taxonomy and reviewcited by 97
- 2023Revisiting Out-of-distribution Robustness in NLP: Benchmark, Analysis, and LLMs Evaluationscited by 96
- 2022Shortcut Learning of Large Language Models in Natural Language Understandingcited by 83
- 2021Identifying and Mitigating Spurious Correlations for Improving Robustness in NLP Modelscited by 76
- 2021"Will You Find These Shortcuts?" A Protocol for Evaluating the Faithfulness of Input Salience Methods for Text Classificationcited by 66
- 2022Detecting Shortcut Learning for Fair Medical AI using Shortcut Testingcited by 52
- 2023Won't Get Fooled Again: Answering Questions with False Premisescited by 47
- 2024Usable XAI: 10 Strategies Towards Exploiting Explainability in the LLM Eracited by 45
- 2025Large Language Models Hallucination: A Comprehensive Surveycited by 45
- 2021The Dangers of Underclaiming: Reasons for Caution When Reporting How NLP Systems Failcited by 38
- 2022PCL: Peer-Contrastive Learning with Diverse Augmentations for Unsupervised Sentence Embeddingscited by 29
12 most cited of 70 All 70 (JSON)
- 2026Models Know Their Shortcuts: Deployment-Time Shortcut Mitigation14 Apr 2026
- 2026AI Evaluation Should Require Standardized Item-Level Data Releases27 Feb 2026
- 2025Large Language Models Hallucination: A Comprehensive Survey5 Oct 2025
- 2025Can Out-of-Distribution Evaluations Uncover Reliance on Shortcuts? A Case Study in Question Answering25 Aug 2025
- 2025Detecting and Pruning Prominent but Detrimental Neurons in Large Language Models12 Jul 2025
- 2025Mitigating Shortcut Learning with InterpoLated Learning7 Jul 2025
- 2025Not quite Sherlock Holmes: Language model predictions do not reliably differentiate impossible from improbable events7 Jun 2025
- 2025Improving the OOD Performance of Closed-Source LLMs on NLI Through Strategic Data Selection26 May 2025
- 2025Physics-informed Temporal Alignment for Auto-regressive PDE Foundation Models16 May 2025
- 2025Short-circuiting Shortcuts: Mechanistic Investigation of Shortcuts in Text Classification9 May 2025
- 2025Why and How LLMs Hallucinate: Connecting the Dots with Subsequence Associations17 Apr 2025
- 2025MiMu: Mitigating Multiple Shortcut Learning Behavior of Transformers14 Apr 2025
12 newest of 70 All 70 (JSON)