mlsanityData to 14 Oct 2026

Beyond Interpretability: The Gains of Feature Monosemanticity on Model Robustness

arXiv 2410.21331 ↗First posted 27 Oct 2024cs.LG cs.AIcited by 13

Not on the map. Why?

Abstract (arXiv)

Deep learning models often suffer from a lack of interpretability due to polysemanticity, where individual neurons are activated by multiple unrelated semantics, resulting in unclear attributions of model behavior. Recent advances in monosemanticity, where neurons correspond to consistent and distinct semantics, have significantly improved interpretability but are commonly believed to compromise accuracy. In this work, we challenge the prevailing belief of the accuracy-interpretability tradeoff, showing that monosemantic features not only enhance interpretability but also bring concrete gains in model performance. Across multiple robust learning scenarios-including input and label noise, few-shot learning, and out-of-domain generalization-our results show that models leveraging monosemantic features significantly outperform those relying on polysemantic features. Furthermore, we provide empirical and theoretical understandings on the robustness gains of feature monosemanticity. Our preliminary analysis suggests that monosemanticity, by promoting better separation of feature representations, leads to more robust decision boundaries. This diverse evidence highlights the generality of monosemanticity in improving model robustness. As a first step in this new direction, we embark on exploring the learning benefits of monosemanticity beyond interpretability, supporting the long-standing hypothesis of linking interpretability and robustness. Code is available at urlhttps://github.com/PKU-ML/BeyondInterpretability.

Key facts. Cited by 13 papers in the map (data through 14 Oct 2026). Builds on 43 of its 61 references that are in the map. First posted on arXiv 27 Oct 2024.

Builds on

43 of 61 references.

References · 43

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  3. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  4. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  5. 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
  6. 2018ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustnesscited by 1,432
  7. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
  8. 2023Fine-tuning Aligned Language Models Compromises Safety, Even When Users Do Not Intend To!cited by 1,154
  9. 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,059
  10. 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,013
  11. 2022Toy Models of Superpositioncited by 971
  12. 2023BeaverTails: Towards Improved Safety Alignment of LLM via a Human-Preference Datasetcited by 860
31 more
  1. 2021Generalized Out-of-Distribution Detection: A Surveycited by 802
  2. 2020Learning from Noisy Labels with Deep Neural Networks: A Surveycited by 582
  3. 2024Scaling and evaluating sparse autoencoderscited by 553
  4. 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
  5. 2024Large Language Models: A Surveycited by 525
  6. 2019Symmetric Cross Entropy for Robust Learning with Noisy Labelscited by 500
  7. 2017Robust Loss Functions under Label Noise for Deep Neural Networkscited by 443
  8. 2024Mechanistic Interpretability for AI Safety -- A Reviewcited by 422
  9. 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 394
  10. 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 334
  11. 2014Training Deep Neural Networks on Noisy Labels with Bootstrappingcited by 322
  12. 2020Normalized Loss Functions for Deep Learning with Noisy Labelscited by 257
  13. 2024ShieldGemma: Generative AI Content Moderation Based on Gemmacited by 243
  14. 2019Understanding and Utilizing Deep Neural Networks Trained with Noisy Labelscited by 209
  15. 2024When Scaling Meets LLM Finetuning: The Effect of Data, Model and Finetuning Methodcited by 204
  16. 2018Dimensionality-Driven Learning with Noisy Labelscited by 173
  17. 2020Compositional Explanations of Neuronscited by 172
  18. 2016Linear Algebraic Structure of Word Senses, with Applications to Polysemycited by 165
  19. 2019A Meta-Transfer Objective for Learning to Disentangle Causal Mechanismscited by 158
  20. 2015Learning with Symmetric Label Noise: The Importance of Being Unhingedcited by 120
  21. 2018Iterative Learning with Open-set Noisy Labelscited by 108
  22. 2021Measure and Improve Robustness in NLP Models: A Surveycited by 107
  23. 2016Multifaceted Feature Visualization: Uncovering the Different Types of Features Learned By Each Neuron in Deep Neural Networkscited by 106
  24. 2023Dynamical versus Bayesian Phase Transitions in a Toy Model of Superpositioncited by 29
  25. 2024Fine Tuning LLM for Enterprise: Practical Guidelines and Recommendationscited by 28
  26. 2023What Causes Polysemanticity? An Alternative Origin Story of Mixed Selectivity from Incidental Causescited by 19
  27. 2024Non-negative Contrastive Learningcited by 19
  28. 2022Improving Out-of-Distribution Generalization by Adversarial Training with Structured Priorscited by 17
  29. 2024Understanding polysemanticity in neural networks through coding theorycited by 17
  30. 2023Identifiable Contrastive Learning with Automatic Feature Importance Discoverycited by 14
  31. 2022On the Connection between Invariant Learning and Adversarial Training for Out-of-Distribution Generalizationcited by 5

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 3; 2026: 10.

13 papers.

Sort:

By area: