Removing Spurious Concepts from Neural Network Representations via Joint Subspace Estimation
Not on the map. Why?
Abstract (arXiv)
Out-of-distribution generalization in neural networks is often hampered by spurious correlations. A common strategy is to mitigate this by removing spurious concepts from the neural network representation of the data. Existing concept-removal methods tend to be overzealous by inadvertently eliminating features associated with the main task of the model, thereby harming model performance. We propose an iterative algorithm that separates spurious from main-task concepts by jointly identifying two low-dimensional orthogonal subspaces in the neural network representation. We evaluate the algorithm on benchmark datasets for computer vision (Waterbirds, CelebA) and natural language processing (MultiNLI), and show that it outperforms existing concept removal methods
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 56 of its 71 references that are in the map. First posted on arXiv 18 Oct 2023.
Builds on
56 of 71 references.
References · 56
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,733
- 2014Deep Learning Face Attributes in the Wildcited by 3,693
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,682
- 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2014Unsupervised Domain Adaptation by Backpropagationcited by 1,917
- 2019Invariant Risk Minimizationcited by 1,854
- 2018ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustnesscited by 1,432
- 2016Understanding intermediate layers using linear classifier probescited by 1,353
- 2018On the Spectral Bias of Neural Networkscited by 1,341
38 more
- 2017Interpretability Beyond Feature Attribution: Quantitative Testing with Concept Activation Vectors (TCAV)cited by 1,319
- 2016Man is to Computer Programmer as Woman is to Homemaker? Debiasing Word Embeddingscited by 1,312
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
- 2019Right for the Wrong Reasons: Diagnosing Syntactic Heuristics in Natural Language Inferencecited by 876
- 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 876
- 2017Network Dissection: Quantifying Interpretability of Deep Visual Representationscited by 708
- 2018Annotation Artifacts in Natural Language Inference Datacited by 671
- 2018Mitigating Unwanted Biases with Adversarial Learningcited by 663
- 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 523
- 2020Underspecification Presents Challenges for Credibility in Modern Machine Learningcited by 496
- 2020Null It Out: Guarding Protected Attributes by Iterative Nullspace Projectioncited by 455
- 2022Last Layer Re-Training is Sufficient for Robustness to Spurious Correlationscited by 381
- 2021Interpretable Machine Learning: Fundamental Principles and 10 Grand Challengescited by 367
- 2020Noise or Signal: The Role of Image Backgrounds in Object Recognitioncited by 343
- 2020An Investigation of Why Overparameterization Exacerbates Spurious Correlationscited by 310
- 2020Gradient Starvation: A Learning Proclivity in Neural Networkscited by 267
- 2023LEACE: Perfect linear concept erasure in closed formcited by 267
- 2020Concept Whitening for Interpretable Image Recognitioncited by 225
- 2021Invariance Principle Meets Information Bottleneck for Out-of-Distribution Generalizationcited by 220
- 2019Probing Neural Network Comprehension of Natural Language Argumentscited by 214
- 2015Censoring Representations with an Adversarycited by 192
- 2021Simple data balancing achieves competitive worst-group-accuracycited by 178
- 2018Adversarial Removal of Demographic Attributes from Text Datacited by 170
- 2020Probing the Probing Paradigm: Does Probing Accuracy Entail Task Relevance?cited by 151
- 2022On Feature Learning in the Presence of Spurious Correlationscited by 145
- 2021Salient ImageNet: How to discover spurious features in Deep Learning?cited by 114
- 2022Linear Adversarial Concept Erasurecited by 101
- 2018How Much Reading Does Reading Comprehension Require? A Critical Investigation of Popular Benchmarkscited by 88
- 2016Places: An Image Database for Deep Scene Understandingcited by 76
- 2022Domain-Adjusted Regression or: ERM May Already Learn Features Sufficient for Out-of-Distribution Generalizationcited by 69
- 2021Examining and Combating Spurious Features under Distribution Shiftcited by 65
- 2019Attenuating Bias in Word Vectorscited by 64
- 2020Robustness to Spurious Correlations via Human Annotationscited by 62
- 2019Understanding Undesirable Word Embedding Associationscited by 59
- 2020Identifying Spurious Correlations for Robust Text Classificationcited by 59
- 2021Counterfactual Interventions Reveal the Causal Effect of Relative Clause Representations on Agreement Predictioncited by 59
- 2022Discovering Latent Concepts Learned in BERTcited by 56
- 2022Kernelized Concept Erasurecited by 48
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2024: 3; 2025: 1; 2026: 2.
6 papers.
- 2024The Clever Hans Mirage: A Comprehensive Survey on Spurious Correlations in Machine Learningcited by 41
- 2024Navigating Shortcuts, Spurious Correlations, and Confounders: From Origins via Detection to Mitigationcited by 25
- 2025Preserving Task-Relevant Information Under Linear Concept Removalcited by 4
- 2024Optimizing importance weighting in the presence of sub-population shiftscited by 3
- 2026Bias Leaves a Gradient Trail: Label-Free Bias Identification via Gradient Probes on Concept Decompositionscited by 0
- 2026Operational Abstractions of Neural Network Concepts via Topological Representationscited by 0
- 2026Operational Abstractions of Neural Network Concepts via Topological Representations4 Oct 2026
- 2026Bias Leaves a Gradient Trail: Label-Free Bias Identification via Gradient Probes on Concept Decompositions27 May 2026
- 2025Preserving Task-Relevant Information Under Linear Concept Removal12 Jun 2025
- 2024Navigating Shortcuts, Spurious Correlations, and Confounders: From Origins via Detection to Mitigation6 Dec 2024
- 2024Optimizing importance weighting in the presence of sub-population shifts18 Oct 2024
- 2024The Clever Hans Mirage: A Comprehensive Survey on Spurious Correlations in Machine Learning20 Feb 2024