What Do Sparse Autoencoder Feature Discovery Scores Identify? From Activations to Interventions
Organizations: St. Jude Children’s Research Hospital · Columbia University
Abstract
Sparse autoencoder (SAE) features are typically discovered from how their activations co-vary with behaviour, yet they are often used through interventions such as steering or ablation. We take intervention attribution, the first-order effect of a specified intervention on the model output, as our estimand. For a broad class of response functions, we show that observational feature-discovery evidence generally does not identify this attribution, and derive an exact gradient-based decomposition of the gap between linear-probe predictions and intervention attribution. Across four model-SAE pairs, median probe-to-attribution magnitude ratios range from tens to hundreds, while attributions closely track small-step steering. For single-latent ablations with nonzero effects, gradient predictions match the realized sign in 91-97% of cases, compared with 17-35% for probe-implied changes. Attribution-selected sets also produce larger absolute mean ablation effects on held-out prompts than sets selected by correlation, rank tests or probe coefficients. Together, these results establish intervention attribution as a principled, gradient-estimable target for selecting SAE features for intervention.
Figures & tables
| Term | Compares | Zero when | SAE instance |
|---|---|---|---|
| Stein residual | observational regression slope with mean gradient | globally affine, or Gaussian with | a response curved across prompts |
| Omitted variable | probe’s activation pattern with mean edit | , e.g. | an edit at positions the probe does not read |
| Edit-gradient covariance | prompt-varying edit with model sensitivity | fixed edit, as for steering | ablation of a rarely firing latent |
| Size of the gap | Realized interventions | Choosing latents | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Steering | Ablation | Sets, attr. | |||||||||||
| Model | Pairs | (%) [CI] | MAE red. | Sign (%) | Top-1 | FDFI | |||||||
| GPT-2 | 46 | 4.6 [3.0, 5.9] | 0.739 | 0.614 | 0.514 | 0.370 | 1.000 | 7.60 | 0.303 | 92/21 | 1/0/4 (6) | 0.454 | 0.290 |
| Gemma-2-2B | 230 | 3.2 [2.2, 3.6] | 0.770 | 0.922 | 0.593 | 0.296 | 1.000 | 3.68 | 0.119 | 91/22 | 0/0/4 (9) | 0.506 | |
| Gemma-2-9B | 457 | 2.8 [1.9, 3.5] | 0.751 | 0.728 | 0.463 | 0.287 | 1.001 | 2.50 | 0.114 | 93/17 | 0/0/4 (9) | 2.137 | 1.380 |
| Llama-3.1-8B | 23 ∗ | 3.2 [2.2, 5.1] | 0.826 | 0.583 | 0.406 | 0.043 | 1.000 | 3.36 | 0.300 | 97/35 | 2/1/4 (4) | 0.309 | 0.206 |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Law of | Edit | ||||||
|---|---|---|---|---|---|---|---|
| uniform on | |||||||
| , | |||||||
| uniform on |
| Gaussian | JumpReLU code | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| GPT-2 | 2B | 9B | 8B | GPT-2 | 2B | 9B | 8B | GPT-2 | 2B | 9B | 8B | |
| Tasks | 8 | 10 | 10 | 7 | 8 | 10 | 10 | 7 | 8 | 10 | 10 | 7 |
| Stein, mean | 0.961 | 0.971 | 0.970 | 0.967 | 0.918 | 0.925 | 0.927 | 0.925 | 0.835 | 0.821 | 0.824 | 0.820 |
| Stein, test-weighted | 0.955 | 0.971 | 0.970 | 0.964 | 0.915 | 0.929 | 0.925 | 0.921 | 0.863 | 0.814 | 0.829 | 0.830 |
| Stein, | 0.974 | 0.981 | 0.978 | 0.978 | 0.925 | 0.934 | 0.939 | 0.936 | 0.804 | 0.792 | 0.796 | 0.777 |
| Stein, | 0.946 | 0.952 | 0.960 | 0.950 | 0.908 | 0.914 | 0.915 | 0.914 | 0.882 | 0.886 | 0.889 | 0.880 |
| Model | Alternate SAE (primary) | FVU alt. / prim. (thr.) | Pool prim. / alt. | Primary (i) / (ii) / (iii) | Alternate (i) / (ii) / (iii) |
|---|---|---|---|---|---|
| GPT-2 | res_scefr-ajt ( res-jb ) | 0.729 / 0.139 (0.278) | N/A | not eligible | N/A |
| Gemma-2-2B | 32k, 40 (16k, 41) | 0.084 / 0.092 (0.184) | 86 / 62 | / 0.420 / 0.040 | / 0.390 / 0.040 |
| Gemma-2-9B | 16k, 35 (131k, 32) | 0.056 / 0.045 (0.090) | 46 / 71 | / 0.300 / 0.030 | / 0.390 / 0.040 |
| Llama-3.1-8B | l16r_32x ( l16r_8x ) | 0.243 / 0.275 (0.500) | 409 / 278 | / 0.389 / 0.111 | / 0.356 / 0.111 |
| Rejections of | Noise-only null | Both significant, | ||||||
|---|---|---|---|---|---|---|---|---|
| Pairs | Model | (tasks) | Median | [CI] | all | median [q5] | opposite sign | |
| FDFI-selected | GPT-2 | 46 (8) | 0.046 | 0.036 [0.019, 0.064] | 35 | 34/38 | 0.226 [0.156] | 5/20 |
| Gemma-2-2B | 230 (10) | 0.032 | 0.032 [0.022, 0.048] | 65 | 50/92 | 0.080 [0.066] | 11/42 | |
| Gemma-2-9B | 457 (10) | 0.028 | 0.038 [0.021, 0.071] | 110 | 99/225 | 0.067 [0.059] | 9/67 | |
| Llama-3.1-8B | 23 (7) ∗ | 0.032 | 0.033 [0.023, 0.049] | 17 | 13/14 | 0.170 [0.103] | 2/12 | |
| Coefficient-selected | GPT-2 | 90 (9) | [ , 0.013] | 43 | 36/40 | 0.030 [0.023] | 6/16 | |
| Probe-blind share | Opposite sign of and | Median | ||||||
|---|---|---|---|---|---|---|---|---|
| Model | Pairs | final | avg. decoder | avg. code | final | avg. decoder | avg. code | avg. decoder / code |
| GPT-2 | 46 | 0.391 | 0 | 0 | 0.414 | 0.403 | 0.336 | 0.294 / 0.167 |
| Gemma-2-2B | 230 | 0.452 | 0.013 | 0 | 0.268 | 0.487 | 0.456 | 0.100 / 0.119 |
| Gemma-2-9B | 457 | 0.534 | 0.020 | 0 | 0.436 | 0.542 | 0.516 | 0.103 / 0.056 |
| Llama-3.1-8B | 23 | 0.087 | 0 | 0 | 0.141 | 0.165 | 0.160 | 0.298 / 0.127 |
| GPT-2 | Gemma-2-2B | Gemma-2-9B | Llama-3.1-8B | |
| Zero-ablation | ||||
| Estimates (tasks) | 46 (8) | 230 (10) | 457 (10) | 23 (7) ∗ |
| 0.522 | 0.339 | 0.385 | 0.174 | |
| equal task weights | 0.317 | 0.287 | 0.335 | 0.097 |
| strata with | 0.500 | 0.344 | 0.360 | 0.208 |
| 0.261 | 0.174 | 0.182 | 0.043 | |
| Single-latent zero-ablations | GPT-2 | Gemma-2-2B | Gemma-2-9B | Llama-3.1-8B |
|---|---|---|---|---|
| Prompt-latent rows (tasks) | 1,795 (7) | 10,727 (9) | 29,180 (9) | 1,782 (7) |
| MAE of | 0.026 | 0.097 | 0.082 | |
| MAE of | 0.054 | 0.115 | 0.091 | 0.032 |
| MAE of | 0.329 | 0.216 | 0.196 | 0.308 |
| Gain over the probe | 0.303 | 0.119 | 0.114 | 0.300 |
| 95% CI | [0.037, 0.799] | [0.078, 0.167] | [0.064, 0.171] | [0.114, 0.611] |
| Within-stratum standard deviations | Maximum activations | |||||||
| Target | Ranked by | 1 | 2 | 5 | 10 | 1 | 2 | 5 |
| FDFI statistic | 4 | 4 | 3 | 3 | 3 | 4 | 2 | |
| 2 | 2 | 2 | 2 | 1 | 3 | 3 | ||
| 28 | 27 | 19 | 17 | 16 | 10 | 5 | ||
| FDFI statistic | 4 | 4 | 3 | 3 | 5 | 4 | 2 | |
| 7 | 7 | 5 | 4 | 4 | 6 | 8 | ||
| Comparator | Stratum-resampled 95% CI | Two-stage 95% CI | GPT-2 | Gemma-2-2B | Gemma-2-9B | Llama-3.1-8B | Units with | |
|---|---|---|---|---|---|---|---|---|
| Correlation | 0.949 | [0.903, 1.001] | [0.292, 2.147] | 0.396 | 0.370 | 2.033 | 0.185 | 28/28 |
| Rank test | 0.933 | [0.885, 0.984] | [0.277, 2.121] | 0.415 | 0.276 | 2.044 | 0.278 | 28/28 |
| Class-mean difference | 0.724 | [0.640, 0.798] | [0.061, 1.795] | 0.331 | 1.795 | 0.236 | 24/28 | |
| 1.058 | [1.013, 1.112] | [0.371, 2.304] | 0.454 | 0.506 | 2.137 | 0.309 | 28/28 | |
| FDFI | 0.492 | [0.357, 0.602] | [ , 1.441] | 0.290 | 1.380 | 0.206 | 23/28 |
| Prompt | Realized | First-order | Edit | ||
|---|---|---|---|---|---|
| answer (unedited ) | Zero-ablated | change | prediction | norm | |
| In Turkey, people commonly speak | FDFI repr. | 0.353 | 0.015 | 0.012 | 97.006 |
| Turkish (0.349) | attribution repr. | 0.403 | 0.201 | 0.224 | 24.246 |
| FDFI set | 0.334 | 120.406 | |||
| attribution set | 0.457 | 0.381 | 0.479 | 85.688 | |
| Most people in Lebanon speak | FDFI repr. | 0.448 | 18.565 |
Sparse Autoencoders
Sparse autoencoders decompose neural network activations into sparse features to study, interpret, and control learned representations.
See all 218 papers