Sparse Autoencoders Find Highly Interpretable Features in Language Models

arXiv:2309.08600 · PDF · 20 pp. · its place among the papers

One paper, several tellings. The big picture states the paper’s own thesis and contributions in its own terms; Inside the paper follows the paper’s own arc; Across the corpus traces how it connects to the other papers; The concepts lists everything the paper uses, in reading order. The experiments explains the paper’s experiments and results as one connected account, built from the paper’s concepts, with its figures and tables attached as supporting evidence — each opening into a page of its own. Pick a tab, then use the +/− toggles to open it level by level, or set a granularity to read the whole thing at that zoom.

Strip away the corpus and the citations, and this paper makes one claim in its own words: individual neurons are the wrong place to look for what a language model represents, because a network can pack more concepts than it has neurons by scattering them across shared, overlapping directions instead of clean, separate units.

What follows is the paper's own argument, in its own vocabulary: the thesis that neurons hide more than they show, the autoencoder built to recover the directions underneath them, the evidence that the recovered features score and localize better than every rival decomposition tried, a demonstration on two individual features examined by hand, and the limits the paper reports against its own result.

Read at

The 2023 argument: neurons are the wrong unit, and a learned dictionary is the fix (5 chapters)

Evidence for the thesis runs in three stages: the theoretical case that superposition, not just noisy training, is why individual neurons are polysemantic; the autoencoder built to recover the overcomplete set of directions superposition implies; and two kinds of proof that the recovered features are genuinely better than four rival decompositions, an automated interpretability score run at scale and a causal-patching test on a well-understood behavior, backed up by two individual features examined by hand. The paper reports its own limits alongside the result: reconstruction that never reaches zero, a recipe that partly breaks on MLP activations, and a safety ambition it names as unreached.

Chapter 1The thesis: neurons hide more than they show (5 concepts)

The paper's central claim, stated in its abstract and defended for the rest of its length, is that polysemanticity — individual neurons activating in multiple, semantically distinct contexts — is caused by superposition, models representing more features than they have neurons by assigning them to an overcomplete set of directions rather than to individual neurons. Sparse autoencoders are offered as a scalable, unsupervised method for recovering those directions directly, aimed at providing “a foundation for future mechanistic interpretability work.”

Why a network can represent more than it has room for concept (2 connections)

A hypothesized cause of polysemanticity in which a network represents more features than it has dimensions by assigning them to an overcomplete set of non-orthogonal directions, relying on activation sparsity to limit interference between them. The paper's stated goal is to resolve superposition in language models directly, by recovering these directions via sparse dictionary learning rather than working around the problem.

connections
  • Superpositionmanifests as → Polysemanticity
  • Superpositionmotivates → Sparse autoencoders (SAEs)
The opening problem the whole paper answers to concept (2 connections)

The paper's opening problem: individual neurons activate in multiple, semantically unrelated contexts, preventing concise, human-understandable explanations of what a network is doing internally. Named in the abstract's first sentence as “one of the roadblocks to a better understanding of neural networks' internals,” it is the roadblock the paper's entire method is aimed at removing.

connections
  • Polysemanticityblocks → Mechanistic interpretability
  • Superposition — manifests asPolysemanticity
The property the recovered directions are judged by concept (3 connections)

The property the method is judged by: a feature activating for a single, coherent, human-interpretable concept, the converse of polysemanticity. The paper's case studies — an apostrophe-detecting feature and a closing-parenthesis-predicting feature — are framed as demonstrating that learned dictionary features are highly monosemantic compared to neurons or other baseline directions, even if not every feature covers an entire human-nameable category on its own.

connections
The discipline the method is offered to advance concept (2 connections)

The research agenda the paper names as its home discipline: reverse-engineering a network's internal computations well enough to make targeted changes to them, motivated in part by the worry that AI systems might deceive humans to accomplish undesirable goals. The paper offers its method as “a foundation for future mechanistic interpretability work,” hoping it will “enable greater model transparency and steerability.”

connections
The method: an autoencoder built to unmix superposition concept (14 connections)

The paper's method: a single-hidden-layer, ReLU, tied-weight autoencoder trained with an L1 sparsity penalty to reconstruct a language model's activations as a sparse linear combination of an overcomplete dictionary of learned features. The paper's central claim about this object is that the resulting features are more interpretable, more monosemantic, and more causally precise than neurons or any of four alternative decompositions tested.

connections
Chapter 2The method: sparsity traded against reconstruction (4 concepts)

Building the dictionary means committing to a single training objective with two competing terms: reconstruct the original activation as closely as possible, while keeping the code that reconstructs it as sparse as possible. Every other design choice in the method — how overcomplete the dictionary is, how its weights are structured — is downstream of how that one tradeoff is set.

The central unit: one learned direction concept (5 connections)

A single learned row of the autoencoder's weight matrix, representing one direction in activation space used to reconstruct inputs as part of a sparse linear combination. Individual dictionary features are the paper's central unit of analysis throughout — examined for interpretability, causal importance, and monosemanticity in turn.

connections
The penalty that keeps the code sparse concept (2 connections)

The L1 penalty on the autoencoder's hidden-layer coefficients, encouraging each activation to be reconstructed from as few active features as possible. The sparsity coefficient is the paper's key hyperparameter, swept throughout every experiment to trade sparsity against reconstruction accuracy rather than fixed at one value.

connections
  • Sparsity loss (L1 penalty on feature activations)forces unit norm on → Dictionary feature
  • Sparsity loss (L1 penalty on feature activations)trades off against → Reconstruction loss
The term that admits the dictionary is incomplete concept (2 connections)

The squared-error term measuring how well the autoencoder's output matches the original activation it was given. The paper is explicit that this loss never reaches zero in its experiments — the dictionary does not capture everything a layer's activations carry, a gap the paper treats as an open limitation rather than something the method has yet resolved.

connections
The dial that sets how overcomplete the dictionary is concept (2 connections)

The hyperparameter controlling how overcomplete the learned dictionary is, the ratio between the autoencoder's hidden dimension and the model's own activation dimension. The paper finds interpretability gains hold even for undercomplete dictionaries, with larger dictionaries simply explaining more of a layer's overall activation variance rather than producing individually more interpretable features.

connections
Chapter 3The evidence: features that score and localize better than every alternative (5 concepts)

Two experiments carry the paper's argument that its dictionary features are not just plausible but genuinely superior. An automated scoring pipeline, run across thousands of features, measures interpretability against four baseline decompositions; a causal-patching test on a well-understood model behavior checks whether the winning features do anything at all, not just whether they look interpretable.

The scoring pipeline: explain, then simulate concept (9 connections)

The paper's primary interpretability measure: a language model is shown text where a feature activates and writes a natural-language explanation, then a second model uses that explanation alone to predict the feature's activations on held-out text, with the correlation between predicted and actual activations as the score. Run across thousands of features, dictionary features score higher on average than every one of the paper's four baseline decompositions.

connections
The baseline that loses once causal precision is what's tested concept (3 connections)

One of four baseline decompositions the paper's dictionary features are measured against. On the causal-localization test, PCA needs more patched components and larger edit magnitude than the dictionary to reach a given level of behavioral change — a gap the paper traces specifically to its sparsity penalty, since removing that penalty from the same architecture erases the advantage entirely.

connections
The naive baseline: just read the coordinates directly concept (3 connections)

The naive baseline of treating individual neurons, or individual residual-stream coordinates, as the unit of analysis. The paper's apostrophe case study makes the comparison direct: the best-matching raw coordinate the paper could find is still polysemantic across much of its activation range, where the matched dictionary feature stays clean throughout.

connections
  • Apostrophe dictionary feature (feature 556) — is benchmarked againstDefault (neuron/residual-stream) basis baseline
  • Outlier dimensions — fails to elevateDefault (neuron/residual-stream) basis baseline
  • Default (neuron/residual-stream) basis baselineloses to → Sparse autoencoders (SAEs)
The causal test: patch a feature, and measure what moves concept (3 connections)

The paper's causal test: run the model on a counterfactual sentence, cache the resulting feature activations, then add the encoded differences for a chosen subset of features back into a base sentence's residual stream, measuring how far the output moves toward the counterfactual target. This reaches a given level of change using fewer patched features and a smaller edit magnitude than an equivalent patch built from PCA components.

connections
The behavior used to test causal precision concept (1 connection)

The behavior used to test causal precision: a model correctly naming the recipient in sentences like “Alice and Bob went to the store. Alice gave a snack to ___.” Its circuit was already well-understood externally before this paper touched it, which is what lets the paper confirm that the features its patching procedure identifies as important are actually the ones responsible for the behavior.

connections
  • Dictionary-feature patching procedure — is tested againstIndirect Object Identification (IOI) task
Chapter 4The demonstration: two features, read from both ends (3 concepts)

Aggregate scores establish that dictionary features are interpretable on average; two hand-picked features establish what that interpretability actually looks like up close. One is examined for what activates it and what it causes; the other is examined for what causes it.

A feature that fires on apostrophes, and only apostrophes concept (3 connections)

A feature that activates almost exclusively on apostrophe tokens, the paper's central worked example of monosemanticity. Ablating it specifically suppresses the model's prediction of the following 's' token, giving it an intuitive, predictable effect on the model's output — in clear contrast to the polysemantic raw coordinate that shares its top activations.

connections
A feature whose own weights nearly name its meaning concept (2 connections)

A feature in the model's final layer whose unembedding directly predicts closing-parenthesis tokens — a meaning confirmable almost without any activation data at all, since a final-layer feature's unembedding directly names the tokens it promotes. Its near-unambiguous identity is what makes it a credible test case for the paper's circuit-tracing method.

connections
  • Apostrophe dictionary feature (feature 556) — divides case study labor withClosing-parenthesis dictionary feature
  • Closing-parenthesis dictionary featureseeds → Automatic feature circuit detection
Tracing a feature back to what causes it concept (4 connections)

A method for tracing which upstream features cause a target feature to activate: sample contexts where the target fires, ablate each candidate upstream feature in turn, and rank candidates by how much their removal lowers the target's activation. Applied to the closing-parenthesis feature, it surfaces upstream features detecting dates, acronyms, and other parenthesis-preceding phrases — an interpretable, sensible-looking circuit built from nothing but repeated ablation.

connections
Chapter 5The limits the paper names against its own result (3 concepts)

The paper closes by naming exactly where its own method stops working, rather than letting the interpretability and causal-precision results stand unqualified. Reconstruction never fully succeeds, the recipe validated on the residual stream breaks down when ported to MLP activations, and the safety ambition the method is offered toward is explicitly named as unreached.

A modest-looking loss, made concrete concept (2 connections)

A check on how much information the dictionary actually loses: substitute a layer's activations with the autoencoder's reconstruction and measure the resulting rise in the model's perplexity. For one tested layer, this raises perplexity from 25 to 40, nearly doubling it — concrete evidence that a reconstruction loss which looks small in the abstract still costs the model real predictive accuracy.

connections
The recipe, ported to a harder setting, and half-breaking concept (3 connections)

An attempt to extend the method from the residual stream to MLP sublayer activations, reported as “mixed success”: many resulting features are more interpretable than individual neurons, but the approach suffers from large numbers of features that never activate at all, and does not yet reliably learn an overcomplete dictionary in the model's middle or later MLP layers.

connections
The ambition the paper admits it has not reached concept (3 connections)

The long-term goal the paper names in its conclusion: a complete, human-understandable enumeration of a model's features, sufficient to guarantee it will not perform dangerous behaviors such as deception. The paper is explicit that it offers only “a step towards” this ambition, not an instance of it — reconstruction never reaches zero, MLP dictionaries lose many features, and only a handful of layers on two small models have been tested.

connections