In Defense of Information Leakage in Concept-based Models
Not on the map. Why?
Abstract (arXiv)
Concept-based models (CMs), deep neural networks that ground their predictions on representations aligned with human-understandable concepts (e.g., "round", "stripes", etc.), have been shown to learn representations that leak concept-irrelevant information. As the traditional narrative goes, this leakage is undesirable and should be eradicated as it leads to uninterpretable models. In this paper, we posit that this conventional view of leakage in CMs is not only ill-posed, as the evidence of how leakage makes a model less interpretable is often inconclusive, but also bound to lead to impractical CMs under common real-world constraints. Specifically, we argue that in real-world settings where concept incompleteness is the norm, some leakage is often necessary for constructing accurate and intervenable CMs. To this end, we propose that there is such a thing as benign leakage and show that, by optimizing a reframing of the typical CM training objective, CMs can encourage and exploit this form of leakage without sacrificing accuracy or intervenability.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 67 of its 76 references that are in the map. First posted on arXiv 9 Jun 2026.
Builds on
67 of 76 references.
References · 67
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
- 2017Towards A Rigorous Science of Interpretable Machine Learningcited by 1,416
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
- 2020Concept Bottleneck Modelscited by 1,032
- 2019Attention is not Explanationcited by 929
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
38 more
- 2019Attention is not not Explanationcited by 636
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2019Benign Overfitting in Linear Regressioncited by 549
- 2017Zero-Shot Learning -- A Comprehensive Evaluation of the Good, the Bad and the Uglycited by 538
- 2018Towards Robust Interpretability with Self-Explaining Neural Networkscited by 505
- 2019Is Attention Interpretable?cited by 389
- 2018How Does Batch Normalization Help Optimization?cited by 357
- 2020The Pitfalls of Simplicity Bias in Neural Networkscited by 351
- 2022Language in a Bottle: Language Model Guided Concept Bottlenecks for Interpretable Image Classificationcited by 312
- 2022Post-hoc Concept Bottleneck Modelscited by 293
- 2023Label-Free Concept Bottleneck Modelscited by 285
- 2019On Completeness-aware Concept-Based Explanations in Deep Neural Networkscited by 239
- 2018Multiaccuracy: Black-Box Post-Processing for Fairness in Classificationcited by 222
- 2020No Subclass Left Behind: Fine-Grained Robustness in Coarse-Grained Classification Problemscited by 209
- 2026Scaling Monosemanticity: Extracting Interpretable Features from Claude 3 Sonnetcited by 191
- 2018Deep learning generalizes because the parameter-function map is biased towards simple functionscited by 183
- 2018Understanding Batch Normalizationcited by 159
- 2021Promises and Pitfalls of Black-Box Concept Learning Modelscited by 131
- 2021Do Concept Bottleneck Models Learn as Intended?cited by 116
- 2018Troubling Trends in Machine Learning Scholarshipcited by 108
- 2022Linear Adversarial Concept Erasurecited by 101
- 2023Probabilistic Concept Bottleneck Modelscited by 101
- 2023Concept-based Explainable Artificial Intelligence: A Surveycited by 98
- 2024Discover-then-Name: Task-Agnostic Concept Bottlenecks via Automated Concept Discoverycited by 91
- 2022Concept Bottleneck Model with Additional Unsupervised Conceptscited by 79
- 2022Interactive Concept Bottleneck Modelscited by 79
- 2025SAEs Are Good for Steering -- If You Select the Right Featurescited by 78
- 2022GlanceNets: Interpretabile, Leak-proof Concept-based Modelscited by 73
- 2023A Closer Look at the Intervention Procedure of Concept Bottleneck Modelscited by 64
- 2023Not All Neuro-Symbolic Concepts Are Created Equal: Analysis and Mitigation of Reasoning Shortcutscited by 63
- 2022Concept Embedding Models: Beyond the Accuracy-Explainability Trade-Offcited by 61
- 2025Deep Learning is Not So Mysterious or Differentcited by 56
- 2024Stochastic Concept Bottleneck Modelscited by 51
- 2019How model accuracy and explanation fidelity influence user trustcited by 45
- 2023Learning to Receive Help: Intervention-Aware Concept Embedding Modelscited by 41
- 2023Human Uncertainty in Concept-Based AI Systemscited by 39
- 2023Auxiliary Losses for Learning Generalizable Concept-based Modelscited by 35
- 2024Energy-Based Concept Bottleneck Models: Unifying Prediction, Concept Intervention, and Probabilistic Interpretationscited by 33
50 most cited of 67 All 67 (JSON)
Built on by
Not cited yet.