atlas

Research map / Architectures and efficiency / RNNs and normalization

Is This the Subspace You Are Looking for? An Interpretability Illusion for Subspace Activation Patching

ICLR 2024 · Architectures and efficiency › RNNs and normalization · approach: Mechanistic circuit interpretability · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that 1-dimensional subspace activation patching can yield an illusory interpretability result by activating a dormant pathway through a causally disconnected MLP subspace, demonstrated on IOI in GPT-2 Small and factual recall in GPT2-XL, and linked to rank-1 (ROME) fact editing.

Problem

Subspace activation patching (e.g., via DAS) can make a model's output behave as if a feature changed, but the found subspace may not faithfully represent that feature; the paper asks whether such patches localize real features in the computation.

Approach

A patched subspace can combine a causally disconnected direction (in the kernel of the MLP down-projection) that varies with the feature and a dormant direction that is causal only when set out of distribution; patching then activates the dormant pathway. The paper links this to ROME rank-1 edits, which are approximately equivalent to 1-D subspace interventions.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Patch the 1-dimensional DAS direction in MLP8 rather than the full MLP8 activationGPT2-Small · IOIFLDD: -8\% -> 46.7\%Helped
Remove the nullspace component from the MLP8 DAS direction (ablation)GPT2-Small MLP8 DAS patch · IOIFLDD: 46.7\% -> 13.5\%Hurt
Patch a 100-dimensional DAS subspace instead of the 1-dimensional directionGPT2-Small residual stream layer 8 · IOIFLDD: 190 \% (100-dimensional); 140.7\% (1-dimensional)Helped
Patch a 100-dimensional DAS subspace instead of the 1-dimensional directionGPT2-Small MLP8 · IOIFLDD: 62 \% (100-dimensional); 46.7\% (1-dimensional)Helped
Decompose the MLP8 100D patch and patch the resulting 200D space (ablation)GPT2-Small MLP8 DAS patch · IOIFLDD: 62 \% -> 17 \%Hurt
Decompose the residual stream 100D patch and patch the resulting 200D space (ablation)GPT2-Small residual stream layer 8 DAS patch · IOIFLDD: 190 \% -> 200 \%Mixed
Replace the MLP weights with randomly sampled weights and train DAS on the resulting activations (ablation)GPT2-Small MLP8 · IOIlogit difference decreaseHelped
Remove the nullspace component from the approximate ROME subspace intervention (ablation)GPT2-XL ROME-matching intervention · CounterFactrewrite score: all rewrite scores are less than $10^{-3}$Hurt
Patch the MLP8 DAS direction rather than its rowspace component or the full MLP activationGPT2-XL · Factual recallfractional logit differenceHelped
Train the DAS subspace on only two names rather than the full IOI dataset (ablation)GPT2-Small · IOI testFLDDHurt

Surprising findings

Limitations

Experimental setup

Datasets: IOI synthetic templates CounterFact OpenWebText

Benchmarks: IOI task CounterFact

Metrics: fractional logit difference decrease interchange accuracy rewrite score cosine similarity total variance ratio

Baselines: full MLP patching full residual stream patching ROME

Builds on: distributed alignment search ROME activation patching indirect object identification circuit

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy