REVIEW 5 major objections 5 minor 25 references
Empirical Evaluation of Progressive Coding for Sparse Autoencoders
T0 review · 5 major / 5 minor · reviewed 2026-08-16 · deepseek-v4-flash
Pith's one-line read Nested sparse autoencoders beat pruned ones as progressive coders
desk verdict Useful comparison of progressive SAE methods, but the claimed superiority over pruning leans on a single unoptimized baseline. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the progressive coder: an SAE in which decoding only the first $G$ latent dimensions, for any $G \leq N$, yields a usable reconstruction, so computation scales with $G$ rather than $N$. The paper builds this in two ways. The pruning route exploits permutation invariance: because each feature activation depends only on its own row of the encoder, reordering latent dimensions does not change the reconstruction, so sorting features by mean squared activation, $E[\text{activation}^2]$, and keeping the first $G$ converts any pretrained SAE into a progressive coder. The Matryoshka route jointly trains all granularities at once with a shared encoder and decoder, adding a weighted reconstruction loss for each nested head $m \in M$; because the smallest code is a prefix of the largest, one encoding pass serves all granularities, so training $|M|$ nested models costs roughly what the largest one costs. The dictionary power law is what makes the hierarchy learnable: the decoder's covariance eigenvalues and activation importance decay as power laws, so the first columns can, in principle, capture most of the variance.
What would settle it
Compute the optimal subset of $G$ features for each granularity on a pretrained TopK SAE, using exhaustive search for small dictionaries or greedy/orthogonal matching pursuit for large ones, and compare its FVU and recaptured language-modeling loss against the Matryoshka SAE at the same granularity. If the optimized pruned SAE matches or beats the Matryoshka SAE on the frontier, the paper's conclusion that Matryoshka SAEs are more efficient progressive coders would be overturned.
Extended reading notes
Core claim
The paper's central claim is that jointly training nested SAEs makes them strictly better progressive coders than pruning an already-trained vanilla SAE by activation importance. For all granularities in $\{2^{14}, 2^{15}, 2^{16}\}$ and at fixed relative sparsity, the Matryoshka SAE outperforms both the baseline TopK SAE and the column-permuted baseline on fraction-of-variance-unexplained, recaptured language-modeling loss, and representational similarity analysis (RSA), the correlation between distance matrices of original and reconstructed activations. A second, separate finding is that dictionary importance in vanilla SAEs follows a power law: a small number of decoder directions account for most of the variance, with eigenvalues of the decoder covariance decaying with exponents around $-0.54$ to $-0.72$ and mean squared activations decaying with exponents around $-1.03$ to $-1.27$. This supports the pruning baseline, but also explains why pruning degrades at small granularities through feature splitting, where one semantic feature in a small SAE is fragmented across many latents in a large one. The paper further shows that applying the permutation ordering to a Matryoshka SAE improves it slightly, and that a Matryoshka SAE truncated to 16K or 32K latents matches an SAE trained from scratch at that size, suggesting one large model can be distilled into smaller ones efficiently.
Load-bearing premise
The pruning baseline is assumed to be strong: sorting features by how strongly they fire on average is the best tested ordering, but no optimal subset selection or alternative ranking heuristics are compared, so a better pruning method could close the gap with Matryoshka SAEs.
Editorial extensions
If this is right
- A single Matryoshka SAE can replace several separately trained SAEs: at a fixed computational budget, train one large nested model and truncate it to any granularity, with training overhead roughly 1.25 times the largest model.
- At 16K and 32K granularities, both a 64K Matryoshka SAE and a permuted 64K TopK SAE match a same-size-from-scratch SAE, so distillation from one large SAE is a viable cheaper alternative.
- The gap between Matryoshka and pruned baselines shrinks as the ratio of granularity to model size decreases, due to feature splitting; methods that recombine split features could recover the lost performance.
- Sampling granularities uniformly during training concentrates activation mass in the first features and improves progressive-coding metrics further.
- Pruned vanilla features are more interpretable than Matryoshka features, but within Matryoshka SAEs the innermost granularities are more interpretable than the outermost, suggesting the loss already imposes a meaningful ordering.
Reading between the lines
- Extension: The paper's pruning baseline is heuristic; an optimal-subset baseline, such as greedy selection on the decoder or solving the subset-selection problem per granularity, could establish whether the Matryoshka advantage is real or just reflects a weak comparator.
- Extension: The power-law finding suggests that feature ordering by $E[\text{activation}^2]$ could be combined with feature merging to reverse feature splitting; this is a natural next experiment not run in the paper.
- Extension: If the frontier result transfers to larger dictionaries with millions of features and to other activation functions such as JumpReLU, then progressive coding could become the default way to serve SAEs at multiple resolutions, since only one model needs to be trained and stored.
- Extension: The interpretability gap might be reducible by initializing the Matryoshka decoder with a permuted pretrained dictionary or by adding a feature-identity regularization; the paper does not test either, so the trade-off may not be intrinsic.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper addresses the practical problem of obtaining sparse autoencoders (SAEs) of many sizes efficiently, by inducing progressive coders whose first G latents give a useful reconstruction. It proposes and evaluates two approaches: pruning a pretrained vanilla TopK SAE after permuting its dictionary columns by a per-feature importance statistic, and jointly training nested Matryoshka SAEs with shared encoder/decoder weights. Experiments on Gemma-2-2b residual-stream activations compare these methods on reconstruction FVU, recaptured language-modeling loss, and representational similarity, and report that Matryoshka SAEs dominate the granularity-versus-fidelity frontier, while pruned vanilla features remain more interpretable. The paper also reports a power-law structure in SAE dictionaries and fits a scaling law for progressive reconstruction loss.
Significance. If the central comparison is robust, the paper gives useful practical guidance: jointly trained nested SAEs can serve as better progressive coders than a simple pruning baseline at roughly the training cost of the largest model, at the price of some interpretability. The evaluation is not circular, because the compared methods are scored on held-out metrics that are not used to fit them. The paper also credits concurrent Matryoshka SAE work and is candid about limitations such as model scale, kernel inefficiency, and feature splitting. The main weakness is that the pruning comparator is a single heuristic ordering, so the central frontier claim is not yet established against stronger baselines; variance estimates are also missing.
major comments (5)
- [§4, §6] The central comparison is against a single pruning heuristic: Section 4 selects E[activation^2] over activation-frequency ranking, and Section 6 then concludes that Matryoshka SAEs outperform "the baseline column permuted SAE" on the granularity-versus-reconstruction-fidelity frontier. The paper does not compare against other natural orderings (decoder-column norms, leave-one-out FVU increase, or an oracle subset selected by reconstruction loss), and it reports in Section 6 that applying column permutation to the Matryoshka SAE itself further improves performance, which shows the learned Matryoshka ordering is not optimal. If a stronger vanilla-dictionary ordering closes the gap, the claim that joint nested training is a more efficient progressive coding strategy is weakened. Please add at least one stronger ordering and an oracle comparison, and report whether the frontier claim survives.
- [§4 (Eq. 2)] The text states that "Each feature activation z_j depends solely on the dot product between the j-th row of W_enc and the centered input..., plus its bias term" and uses this to justify per-feature importance sorting. For TopK SAEs this is incorrect: the TopK activation selects the K largest pre-activations globally, so z_j depends on all rows of W_enc through the selection mechanism. The permutation-invariance construction remains valid, but the stated conditional independence does not hold, and the per-feature statistic E[activation^2] ignores the coupling introduced by global top-K selection. Please correct the derivation and discuss how this affects the interpretation of the ranking.
- [§6 (Results)] All headline comparisons in Section 6 are based on single training runs. The claim that "For all granularities, the Matryoshka SAE outperforms..." is reported without error bars, confidence intervals, or multiple seeds; given the stochasticity of SAE training, the observed differences on the frontier may be within run-to-run noise. Please report results over at least three independent seeds or initializations (or provide a variance estimate) for FVU, recaptured cross-entropy loss, and RSA.
- [§6 (Interpretability)] The interpretability evaluation uses 10 test sequences and 20 training sequences for explanation generation, and Figure 14 is based on 400 randomly selected features. The abstract's claim that pruned vanilla SAEs are more interpretable rests on this small sample and on a single automated-interpretability pipeline. Please provide confidence intervals or increase the evaluation sample, and temper the interpretability conclusion if the estimates remain this noisy.
- [§7 (Eq. 9)] The scaling law in Eq. (9) is fit to the 16k, 32k, and 65k TopK SAEs and displayed against the same data in Figure 15; no held-out evaluation or cross-validation is reported, and parameter estimates are given without uncertainty. Since the section frames the law as predictive ("how well can the performance of the model be predicted"), please validate on held-out configurations (e.g., unseen granularity/sparsity combinations) or report cross-validated prediction error.
minor comments (5)
- [§6] The notation for granularities is inconsistent: Section 6 defines M={214, 215, 216} and later G∈M={5000,10000,...}; please unify the notation (likely 2^14, 2^15, 2^16) and define all symbols once.
- [§6, Appendix B] Several cross-references are broken: the text refers to "Appendix 21" and "Section ??", and the correlation of about 0.8 between FVU and recaptured LM loss is attributed to a footnote that does not exist; please repair these references.
- [Figure captions 7 and 8] The captions say "Fvu per token for gemma-2-2b divided by the cross-entropy loss using the SAE reconstruction," which does not describe the plotted quantity; the captions should state the metric actually shown.
- [§3] Section 3 reports power-law exponents and R2 values for eigenvalue and activation statistics but gives no confidence intervals or alternative distribution fits; because the tail deviates visibly from a power law, the phrase "dictionary importance follows a power law" is stronger than the evidence shown.
- [Table 1] The hyperparameter table omits batch size, number of training steps, and the exact TopK/k schedule; please add these details to make the training setup reproducible.
Circularity Check
No significant circularity: the central Matryoshka-versus-pruning comparison is evaluated on held-out metrics, and the only fitted scaling law is peripheral.
full rationale
This is an empirical comparison paper, and its central claims are not circular. The main result, that Matryoshka SAEs outperform pruned and column-permuted vanilla SAEs on the granularity-versus-reconstruction-fidelity frontier, is supported by held-out FVU, recaptured LM cross-entropy, and RSA computed on 10^5 held-out tokens; these metrics are not the quantities used to rank features for the pruning baseline (E[activation^2]), nor are they identical to the Matryoshka training loss on the training set. The choice of the pruning ordering is justified by an empirical comparison of two heuristics, which is a legitimate though limited baseline choice rather than a definitional identity; the concern that a stronger ordering might close the gap is a correctness or baseline-strength concern, not circularity. The Section 7 scaling law is explicitly fitted to validation data from the 16k, 32k, and 65k TopK SAEs, so the 'predicted' losses in Figure 15 are in-sample evaluations of a fitted curve; this is a peripheral modeling exercise and is not load-bearing for the abstract's comparative claim. The only co-authored self-citation, Li et al. (2024a), is used to motivate RSA as a representational-similarity metric and does not supply any assumption needed for the comparative result. Appendix C explicitly lists limitations such as modest SAE scale, kernel overhead, and feature-splitting, which shows the authors distinguish limitations from claimed results. Under the rules that self-citation matters only when load-bearing and that a fitted empirical law is not a derivation, I find no circular step and assign score 0.
Assumptions & free parameters
free parameters (2)
- Scaling law parameters =
alpha=-3.60, beta_k=0.69, beta_n=0.19, beta_g=0.08, gamma_n=0.02, gamma_g=-0.10, zeta=-2.13, eta=-0.13
- Matryoshka granularity weights c_m =
unreported
assumptions (4)
- ad hoc to paper E[activation^2] ranking is a good proxy for feature importance in pruning
- domain assumption FVU, recaptured CE loss, and RSA are jointly adequate metrics for progressive coder quality
- domain assumption Relative sparsity (K/G constant) is the right comparison regime for progressive coders
- domain assumption Results on 65k-wide SAEs trained on 50M tokens generalize to larger SAEs
Cite this review
Pith. "Pith review of Empirical Evaluation of Progressive Coding for Sparse Autoencoders." pith.science (2026). https://pith.science/paper/IRXZ4NHC
@misc{pith2026250500190,
author = {Pith},
title = {Pith review of: Empirical Evaluation of Progressive Coding for Sparse Autoencoders},
year = {2026},
howpublished = {\url{https://pith.science/paper/IRXZ4NHC}},
note = {Machine review of arXiv:2505.00190}
}
read the original abstract
Sparse autoencoders (SAEs) \citep{bricken2023monosemanticity,gao2024scalingevaluatingsparseautoencoders} rely on dictionary learning to extract interpretable features from neural networks at scale in an unsupervised manner, with applications to representation engineering and information retrieval. SAEs are, however, computationally expensive \citep{lieberum2024gemmascopeopensparse}, especially when multiple SAEs of different sizes are needed. We show that dictionary importance in vanilla SAEs follows a power law. We compare progressive coding based on subset pruning of SAEs -- to jointly training nested SAEs, or so-called {\em Matryoshka} SAEs \citep{bussmann2024learning,nabeshima2024Matryoshka} -- on a language modeling task. We show Matryoshka SAEs exhibit lower reconstruction loss and recaptured language modeling loss, as well as higher representational similarity. Pruned vanilla SAEs are more interpretable, however. We discuss the origins and implications of this trade-off.
Figures
Figures from the paper (14 more)
Reference graph
Works this paper leans on
-
[4]
Chowdhery, A., Narang, S., Devlin, J., Bosma, M., Mishra, G., Roberts, A., Barham, P., Chung, H
URL https://arxiv.org/ abs/2405.17430. Chowdhery, A., Narang, S., Devlin, J., Bosma, M., Mishra, G., Roberts, A., Barham, P., Chung, H. W., Sutton, C., Gehrmann, S., Schuh, P., Shi, K., Tsvyashchenko, S., Maynez, J., Rao, A., Barnes, P., Tay, Y ., Shazeer, N., Prabhakaran, V ., Reif, E., Du, N., Hutchinson, B., Pope, R., Bradbury, J., Austin, J., Isard, M...
-
[6]
URL https:// arxiv.org/abs/2309.08600. Devvrit, Kudugunta, S., Kusupati, A., Dettmers, T., Chen, K., Dhillon, I., Tsvetkov, Y ., Hajishirzi, H., Kakade, S., Farhadi, A., and Jain, P. Matformer: Nested transformer for elastic inference,
-
[7]
URL https://arxiv. org/abs/2310.07707. EleutherAI. sae-auto-interp. https://github.com/ EleutherAI/sae-auto-interp,
-
[8]
URL https://transformer-circuits. pub/2022/toy_model/index.html. Gao, L., la Tour, T. D., Tillman, H., Goh, G., Troll, R., Radford, A., Sutskever, I., Leike, J., and Wu, J. Scaling and evaluating sparse autoencoders,
work page 2022
-
[9]
URL https: //arxiv.org/abs/2406.04093. Grattafiori, A., Dubey, A., Jauhri, A., Pandey, A., Kadian, A., Al-Dahle, A., Letman, A., Mathur, A., Schelten, A., Vaughan, A., Yang, A., Fan, A., Goyal, A., Hartshorn, A., Yang, A., Mitra, A., Sravankumar, A., Korenev, A., Hinsvark, A., Rao, A., Zhang, A., Rodriguez, A., Gregerson, A., Spataru, A., Roziere, B., Bir...
-
[10]
Gu, J., Zhai, S., Zhang, Y ., Susskind, J., and Jaitly, N
URL https://arxiv.org/abs/2407.21783. Gu, J., Zhai, S., Zhang, Y ., Susskind, J., and Jaitly, N. Matryoshka diffusion models,
-
[11]
URL https: //arxiv.org/abs/2310.15111. Hoffmann, J., Borgeaud, S., Mensch, A., Buchatskaya, E., Cai, T., Rutherford, E., de Las Casas, D., Hendricks, L. A., Welbl, J., Clark, A., Hennigan, T., Noland, E., Millican, K., van den Driessche, G., Damoc, B., Guy, A., Osindero, S., Simonyan, K., Elsen, E., Rae, J. W., 10 Empirical Evaluation of Progressive SAEs ...
- [12]
Show all 25 references
-
[13]
URL https://arxiv.org/abs/2001. 08361. Klabunde, M., Schumacher, T., Strohmaier, M., and Lem- merich, F. Similarity of neural network models: A survey of functional and representational measures,
2001
-
[14]
Kusupati, A., Bhatt, G., Rege, A., Wallingford, M., Sinha, A., Ramanujan, V ., Howard-Snyder, W., Chen, K., Kakade, S., Jain, P., and Farhadi, A
URL https://arxiv.org/abs/2305.06329. Kusupati, A., Bhatt, G., Rege, A., Wallingford, M., Sinha, A., Ramanujan, V ., Howard-Snyder, W., Chen, K., Kakade, S., Jain, P., and Farhadi, A. Matryoshka represen- tation learning,
-
[15]
Li, J., Kementchedjhieva, Y ., Fierro, C., and Søgaard, A
URL https://arxiv.org/ abs/2205.13147. Li, J., Kementchedjhieva, Y ., Fierro, C., and Søgaard, A. Do vision and language models share concepts? a vec- tor space alignment study. Transactions of the Asso- ciation for Computational Linguistics , 12:1232–1249, 2024a. doi: 10.1162...
-
[16]
Makelov, A., Lange, G., and Nanda, N
URL https: //arxiv.org/abs/2408.05147. Makelov, A., Lange, G., and Nanda, N. Towards princi- pled evaluations of sparse autoencoders for interpretabil- ity and control,
-
[17]
Mudide, A., Engels, J., Michaud, E
URL https://arxiv.org/ abs/2405.08366. Mudide, A., Engels, J., Michaud, E. J., Tegmark, M., and de Witt, C. S. Efficient dictionary learning with switch sparse autoencoders,
-
[18]
org/abs/2410.08201
URL https://arxiv. org/abs/2410.08201. Nabeshima, N. Matryoshka sparse autoencoders. AI Align- ment Forum, 12
-
[19]
Rajamanoharan, S., Lieberum, T., Sonnerat, N., Conmy, A., Varma, V ., Kram´ar, J., and Nanda, N
URL https://arxiv.org/ abs/2410.13928. Rajamanoharan, S., Lieberum, T., Sonnerat, N., Conmy, A., Varma, V ., Kram´ar, J., and Nanda, N. Jumping ahead: Improving reconstruction fidelity with jumprelu sparse au- toencoders,
-
[20]
Rippel, O., Gelbart, M
URL https://arxiv.org/abs/ 2407.14435. Rippel, O., Gelbart, M. A., and Adams, R. P. Learning ordered representations with nested dropout,
-
[21]
The scatter plots show pairwise relationships with linear regression fits, displaying both Pearson correlation coefficients (r) and coefficients of determination (R²)
Correlation analysis between different evaluation metrics (FVU, CE Loss, and RSA). The scatter plots show pairwise relationships with linear regression fits, displaying both Pearson correlation coefficients (r) and coefficients of determination (R²). C. Limitations and Future ...
2024
-
[22]
Templeton, A., Conerly, T., Marcus, J., Lindsey, J., Bricken, T., Chen, B., Pearce, A., Citro, C., Ameisen, E., Jones, A., Cunningham, H., Turner, N
URL https://arxiv.org/abs/2408.00118. Templeton, A., Conerly, T., Marcus, J., Lindsey, J., Bricken, T., Chen, B., Pearce, A., Citro, C., Ameisen, E., Jones, A., Cunningham, H., Turner, N. L., McDougall, C., MacDiarmid, M., Freeman, C. D., Sumers, T. R., Rees, E., Batson, J., J...
-
[23]
Yun, Z., Chen, Y ., Olshausen, B
URL https: //transformer-circuits.pub/2024/ scaling-monosemanticity/index.html. Yun, Z., Chen, Y ., Olshausen, B. A., and LeCun, Y . Trans- former visualization via dictionary learning: contextual- ized embedding as a linear superposition of transformer factors,
2024
-
[24]
12 Empirical Evaluation of Progressive SAEs A
URL https://arxiv.org/abs/ 2103.15949. 12 Empirical Evaluation of Progressive SAEs A. Feature Splitting Feature splitting is the phenomenon where as the dictionary grows in size, one basis vector gets decomposed into multiple separate basis vectors. In contrast, the standard T...
-
[2014]
Skodras, A., Christopoulos, C., and Ebrahimi, T
URL https://arxiv.org/abs/1402.0915. Skodras, A., Christopoulos, C., and Ebrahimi, T. The jpeg 2000 still image compression standard. IEEE Signal Processing Magazine, 18(5):36–58,
2000 arXiv
-
[2020]
Bussmann, B., Leask, P., and Nanda, N
URL https:// arxiv.org/abs/2005.14165. Bussmann, B., Leask, P., and Nanda, N. Learn- ing multi-level features with matryoshka saes. AI Alignment Forum, 12
2005 arXiv
-
[2022]
Cunningham, H., Ewart, A., Riggs, L., Huben, R., and Sharkey, L
URL https://arxiv.org/abs/2204.02311. Cunningham, H., Ewart, A., Riggs, L., Huben, R., and Sharkey, L. Sparse autoencoders find highly interpretable features in language models,
-
[2023]
Brown, T
https://transformer- circuits.pub/2023/monosemantic-features/index.html. Brown, T. B., Mann, B., Ryder, N., Subbiah, M., Kaplan, J., Dhariwal, P., Neelakantan, A., Shyam, P., Sastry, G., Askell, A., Agarwal, S., Herbert-V oss, A., Krueger, G., Henighan, T., Child, R., Ramesh, ...
2023
-
[2024]
URL https: //arxiv.org/abs/2405.12241. Bricken, T., Templeton, A., Batson, J., Chen, B., Jermyn, A., Conerly, T., Turner, N., Anil, C., Denison, C., Askell, A., Lasenby, R., Wu, Y ., Kravec, S., Schiefer, N., Maxwell, T., Joseph, N., Hatfield-Dodds, Z., Tamkin, A., Nguyen, K.,...
Reviewed August 16, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.