Beyond Interpretability: The Gains of Feature Monosemanticity on Model Robustness
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
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
- 2018ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustnesscited by 1,432
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
- 2023Fine-tuning Aligned Language Models Compromises Safety, Even When Users Do Not Intend To!cited by 1,154
- 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,059
- 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,013
- 2022Toy Models of Superpositioncited by 971
- 2023BeaverTails: Towards Improved Safety Alignment of LLM via a Human-Preference Datasetcited by 860
31 more
- 2021Generalized Out-of-Distribution Detection: A Surveycited by 802
- 2020Learning from Noisy Labels with Deep Neural Networks: A Surveycited by 582
- 2024Scaling and evaluating sparse autoencoderscited by 553
- 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
- 2024Large Language Models: A Surveycited by 525
- 2019Symmetric Cross Entropy for Robust Learning with Noisy Labelscited by 500
- 2017Robust Loss Functions under Label Noise for Deep Neural Networkscited by 443
- 2024Mechanistic Interpretability for AI Safety -- A Reviewcited by 422
- 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 394
- 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 334
- 2014Training Deep Neural Networks on Noisy Labels with Bootstrappingcited by 322
- 2020Normalized Loss Functions for Deep Learning with Noisy Labelscited by 257
- 2024ShieldGemma: Generative AI Content Moderation Based on Gemmacited by 243
- 2019Understanding and Utilizing Deep Neural Networks Trained with Noisy Labelscited by 209
- 2024When Scaling Meets LLM Finetuning: The Effect of Data, Model and Finetuning Methodcited by 204
- 2018Dimensionality-Driven Learning with Noisy Labelscited by 173
- 2020Compositional Explanations of Neuronscited by 172
- 2016Linear Algebraic Structure of Word Senses, with Applications to Polysemycited by 165
- 2019A Meta-Transfer Objective for Learning to Disentangle Causal Mechanismscited by 158
- 2015Learning with Symmetric Label Noise: The Importance of Being Unhingedcited by 120
- 2018Iterative Learning with Open-set Noisy Labelscited by 108
- 2021Measure and Improve Robustness in NLP Models: A Surveycited by 107
- 2016Multifaceted Feature Visualization: Uncovering the Different Types of Features Learned By Each Neuron in Deep Neural Networkscited by 106
- 2023Dynamical versus Bayesian Phase Transitions in a Toy Model of Superpositioncited by 29
- 2024Fine Tuning LLM for Enterprise: Practical Guidelines and Recommendationscited by 28
- 2023What Causes Polysemanticity? An Alternative Origin Story of Mixed Selectivity from Incidental Causescited by 19
- 2024Non-negative Contrastive Learningcited by 19
- 2022Improving Out-of-Distribution Generalization by Adversarial Training with Structured Priorscited by 17
- 2024Understanding polysemanticity in neural networks through coding theorycited by 17
- 2023Identifiable Contrastive Learning with Automatic Feature Importance Discoverycited by 14
- 2022On the Connection between Invariant Learning and Adversarial Training for Out-of-Distribution Generalizationcited by 5
Built on by
Citing papers per year: 2025: 3; 2026: 10.
13 papers.
Sort:
By area:
- 2025On the Limits of Sparse Autoencoders: A Theoretical Framework and Reweighted Remedycited by 14
- 2025Using Mechanistic Interpretability to Craft Adversarial Attacks against Large Language Modelscited by 10
- 2026Visual Self-Fulfilling Alignment: Shaping Safety-Oriented Personas via Threat-Related Imagescited by 3
- 2026SAE as a Crystal Ball: Interpretable Features Predict Cross-domain Transferability of LLMs without Trainingcited by 2
- 2026Finding and Reactivating Post-Trained LLMs' Hidden Safety Mechanismscited by 2
- 2025Extracting Interaction-Aware Monosemantic Concepts in Recommender Systemscited by 1
- 2026SoftSAE: Dynamic Top-K Selection for Adaptive Sparse Autoencoderscited by 1
- 2026Beyond Accuracy: Interpreting Topic Representation in Suicide Ideation Detection Modelscited by 1
- 2026ToxScreen: Detecting Whether an LLM Has Been Poisonedcited by 1
- 2026Hiding in Plain Text: Detecting Concealed Jailbreaks via Activation Disentanglementcited by 0
- 2026EXPOSE: Explainable and Domain-Robust Embeddings from Pathology Vision Foundation Models using Sparse Autoencoderscited by 0
- 2026A Theoretical Framework for Masked Pretraining (MPT)cited by 0
- 2026Representational Simplicity and Circuit Size Dissociate in a Threshold-Dependent Way: A Controlled Test via Adversarial Training27 Sep 2026
- 2026A Theoretical Framework for Masked Pretraining (MPT)6 Sep 2026
- 2026EXPOSE: Explainable and Domain-Robust Embeddings from Pathology Vision Foundation Models using Sparse Autoencoders28 Aug 2026
- 2026ToxScreen: Detecting Whether an LLM Has Been Poisoned29 Jul 2026
- 2026Beyond Accuracy: Interpreting Topic Representation in Suicide Ideation Detection Models5 Jun 2026
- 2026SoftSAE: Dynamic Top-K Selection for Adaptive Sparse Autoencoders7 May 2026
- 2026Finding and Reactivating Post-Trained LLMs' Hidden Safety Mechanisms10 Mar 2026
- 2026Visual Self-Fulfilling Alignment: Shaping Safety-Oriented Personas via Threat-Related Images9 Mar 2026
- 2026SAE as a Crystal Ball: Interpretable Features Predict Cross-domain Transferability of LLMs without Training3 Mar 2026
- 2026Hiding in Plain Text: Detecting Concealed Jailbreaks via Activation Disentanglement23 Feb 2026
- 2025Extracting Interaction-Aware Monosemantic Concepts in Recommender Systems22 Nov 2025
- 2025On the Limits of Sparse Autoencoders: A Theoretical Framework and Reweighted Remedy19 Jun 2025
- 2025On the Limits of Sparse Autoencoders: A Theoretical Framework and Reweighted Remedycited by 14
- 2025Using Mechanistic Interpretability to Craft Adversarial Attacks against Large Language Modelscited by 10
- 2026SAE as a Crystal Ball: Interpretable Features Predict Cross-domain Transferability of LLMs without Trainingcited by 2
- 2026SoftSAE: Dynamic Top-K Selection for Adaptive Sparse Autoencoderscited by 1
- 2026EXPOSE: Explainable and Domain-Robust Embeddings from Pathology Vision Foundation Models using Sparse Autoencoderscited by 0
- 2026Representational Simplicity and Circuit Size Dissociate in a Threshold-Dependent Way: A Controlled Test via Adversarial Trainingcited by 0
- 2026Representational Simplicity and Circuit Size Dissociate in a Threshold-Dependent Way: A Controlled Test via Adversarial Training27 Sep 2026
- 2026EXPOSE: Explainable and Domain-Robust Embeddings from Pathology Vision Foundation Models using Sparse Autoencoders28 Aug 2026
- 2026SoftSAE: Dynamic Top-K Selection for Adaptive Sparse Autoencoders7 May 2026
- 2026SAE as a Crystal Ball: Interpretable Features Predict Cross-domain Transferability of LLMs without Training3 Mar 2026
- 2025On the Limits of Sparse Autoencoders: A Theoretical Framework and Reweighted Remedy19 Jun 2025
- 2025Using Mechanistic Interpretability to Craft Adversarial Attacks against Large Language Models8 Mar 2025