A Unified Theory of Sparse Dictionary Learning in Mechanistic Interpretability: Piecewise Biconvexity and Spurious Minima
On the map: Mechanistic interpretability → Sparse autoencoder training and evaluation
Abstract (arXiv)
As AI models achieve remarkable capabilities across diverse domains, understanding what representations they learn and how they encode concepts has become increasingly important for both scientific progress and trustworthy deployment. Recent works in mechanistic interpretability have widely reported that neural networks represent meaningful concepts as linear directions in their representation spaces and often encode diverse concepts in superposition. Various sparse dictionary learning (SDL) methods, including sparse autoencoders, transcoders, and crosscoders, are utilized to address this by training auxiliary models with sparsity constraints to disentangle these superposed concepts into monosemantic features. These methods are the backbone of modern mechanistic interpretability, yet in practice they consistently produce polysemantic features, feature absorption, and dead neurons, with very limited theoretical understanding of why these phenomena occur. Existing theoretical work is limited to tied-weight sparse autoencoders, leaving the broader family of SDL methods without formal grounding. We develop the first unified theoretical framework that casts all major SDL variants as a single piecewise biconvex optimization problem, and characterize its global solution set, non-identifiability, and spurious optima. This analysis yields principled explanations for feature absorption and dead neurons. To expose these pathologies under full ground-truth access, we introduce the Linear Representation Bench. Guided by our theory, we propose feature anchoring, a novel technique that restores SDL identifiability, substantially improving feature recovery across synthetic benchmarks and real neural representations.
Key facts. Cited by 2 papers in the map (data through 13 Oct 2026). Placed in Sparse autoencoder training and evaluation, one of 13 directions in Mechanistic interpretability. Builds on 39 of its 63 references that are in the map. First posted on arXiv 5 Dec 2025.
Builds on
39 of 63 references.
Closest ancestors · 3
Same direction, newest first
- 2025How does My Model Fail? Automatic Identification and Interpretation of Physical Plausibility Failure Modes with Matryoshka Transcoders13 Nov
- 2025Learning Multi-Level Features with Matryoshka Sparse Autoencoders21 Mar
- 2025Transcoders Beat Sparse Autoencoders for Interpretability31 Jan
Other references · 36
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,588
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,476
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,713
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,515
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,391
- 2017Interpretability Beyond Feature Attribution: Quantitative Testing with Concept Activation Vectors (TCAV)cited by 1,316
- 2020Concept Bottleneck Modelscited by 1,027
- 2016The Mythos of Model Interpretabilitycited by 718
- 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 706
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 693
- 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 423
- 2024Mechanistic Interpretability for AI Safety -- A Reviewcited by 420
24 more
- 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Modelscited by 316
- 2024Jumping Ahead: Improving Reconstruction Fidelity with JumpReLU Sparse Autoencoderscited by 292
- 2013k-Sparse Autoencoderscited by 214
- 2024Not All Language Model Features Are One-Dimensionally Linearcited by 205
- 2024Transcoders Find Interpretable LLM Feature Circuitscited by 193
- 2024Improving Dictionary Learning with Gated Sparse Autoencoderscited by 177
- 2024The Geometry of Categorical and Hierarchical Concepts in Large Language Modelscited by 158
- 2024A is for Absorption: Studying Feature Splitting and Absorption in Sparse Autoencoderscited by 156
- 2017Non-convex Optimization for Machine Learningcited by 132
- 2016Matrix Completion has No Spurious Local Minimumcited by 124
- 2017Spurious Local Minima are Common in Two-Layer ReLU Neural Networkscited by 104
- 2014Guaranteed Matrix Completion via Non-convex Factorizationcited by 82
- 2024On the Origins of Linear Representations in Large Language Modelscited by 78
- 2025A Survey on Sparse Autoencoders: Interpreting the Internal Mechanisms of Large Language Modelscited by 75
- 2024InterPLM: Discovering Interpretable Features in Protein Language Models via Sparse Autoencoderscited by 66
- 2024Measuring Progress in Dictionary Learning for Language Model Interpretability with Board Game Modelscited by 51
- 2024LLM as Dataset Analyst: Subpopulation Structure Discovery with Large Language Modelcited by 43
- 2016A note on the sample complexity of the Er-SpUD algorithm by Spielman, Wang and Wright for exact recovery of sparsely used dictionariescited by 26
- 2024An X-Ray Is Worth 15 Features: Sparse Autoencoders for Interpretable Radiology Report Generationcited by 25
- 2023Prompt Engineering Through the Lens of Optimal Controlcited by 16
- 2025Rep2Text: Decoding Full Text from a Single LLM Token Representationcited by 7
- 2025Human-like Content Analysis for Generative AI with Language-Grounded Sparse Encoderscited by 6
- 2024Demonstration Notebook: Finding the Most Suited In-Context Learning Example from Interactionscited by 4
- 2025Sparse Autoencoders Bridge The Deep Learning Model and The Braincited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Mechanistic Interpretability for Neural Networks: Circuits, Sparse Features and Symbolic Reasoning
- 2025Equivariant Sparse Autoencoders: Mechanistic Interpretability of Neural Networks on Symmetric Data
- 2025From superposition to sparse codes: interpretable representations in neural networks
- 2025Position: Mechanistic Interpretability Should Prioritize Feature Consistency in SAEs
- 2026Inference and learning in sparse autoencoders as natural gradient flow
- 2026Subspace-Aware Sparse Autoencoders for Effective Mechanistic Interpretability
- 2025Temporal Sparse Autoencoders: Leveraging the Sequential Nature of Language for Interpretability
- 2025Archetypal SAE: Adaptive and Stable Dictionary Learning for Concept Extraction in Large Vision Models