mlsanityData to 14 Oct 2026

In Defense of Information Leakage in Concept-based Models

arXiv 2606.10669 ↗First posted 9 Jun 2026cs.LG cs.AI cs.CRcited by 0

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

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  3. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
  4. 2020Shortcut Learning in Deep Neural Networkscited by 1,969
  5. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
  6. 2017Towards A Rigorous Science of Interpretable Machine Learningcited by 1,416
  7. 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
  8. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
  9. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
  10. 2020Concept Bottleneck Modelscited by 1,032
  11. 2019Attention is not Explanationcited by 929
  12. 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
38 more
  1. 2019Attention is not not Explanationcited by 636
  2. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
  3. 2019Benign Overfitting in Linear Regressioncited by 549
  4. 2017Zero-Shot Learning -- A Comprehensive Evaluation of the Good, the Bad and the Uglycited by 538
  5. 2018Towards Robust Interpretability with Self-Explaining Neural Networkscited by 505
  6. 2019Is Attention Interpretable?cited by 389
  7. 2018How Does Batch Normalization Help Optimization?cited by 357
  8. 2020The Pitfalls of Simplicity Bias in Neural Networkscited by 351
  9. 2022Language in a Bottle: Language Model Guided Concept Bottlenecks for Interpretable Image Classificationcited by 312
  10. 2022Post-hoc Concept Bottleneck Modelscited by 293
  11. 2023Label-Free Concept Bottleneck Modelscited by 285
  12. 2019On Completeness-aware Concept-Based Explanations in Deep Neural Networkscited by 239
  13. 2018Multiaccuracy: Black-Box Post-Processing for Fairness in Classificationcited by 222
  14. 2020No Subclass Left Behind: Fine-Grained Robustness in Coarse-Grained Classification Problemscited by 209
  15. 2026Scaling Monosemanticity: Extracting Interpretable Features from Claude 3 Sonnetcited by 191
  16. 2018Deep learning generalizes because the parameter-function map is biased towards simple functionscited by 183
  17. 2018Understanding Batch Normalizationcited by 159
  18. 2021Promises and Pitfalls of Black-Box Concept Learning Modelscited by 131
  19. 2021Do Concept Bottleneck Models Learn as Intended?cited by 116
  20. 2018Troubling Trends in Machine Learning Scholarshipcited by 108
  21. 2022Linear Adversarial Concept Erasurecited by 101
  22. 2023Probabilistic Concept Bottleneck Modelscited by 101
  23. 2023Concept-based Explainable Artificial Intelligence: A Surveycited by 98
  24. 2024Discover-then-Name: Task-Agnostic Concept Bottlenecks via Automated Concept Discoverycited by 91
  25. 2022Concept Bottleneck Model with Additional Unsupervised Conceptscited by 79
  26. 2022Interactive Concept Bottleneck Modelscited by 79
  27. 2025SAEs Are Good for Steering -- If You Select the Right Featurescited by 78
  28. 2022GlanceNets: Interpretabile, Leak-proof Concept-based Modelscited by 73
  29. 2023A Closer Look at the Intervention Procedure of Concept Bottleneck Modelscited by 64
  30. 2023Not All Neuro-Symbolic Concepts Are Created Equal: Analysis and Mitigation of Reasoning Shortcutscited by 63
  31. 2022Concept Embedding Models: Beyond the Accuracy-Explainability Trade-Offcited by 61
  32. 2025Deep Learning is Not So Mysterious or Differentcited by 56
  33. 2024Stochastic Concept Bottleneck Modelscited by 51
  34. 2019How model accuracy and explanation fidelity influence user trustcited by 45
  35. 2023Learning to Receive Help: Intervention-Aware Concept Embedding Modelscited by 41
  36. 2023Human Uncertainty in Concept-Based AI Systemscited by 39
  37. 2023Auxiliary Losses for Learning Generalizable Concept-based Modelscited by 35
  38. 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.