REVIEW 5 minor 25 references
Equal reconstruction error and sparsity can leave which linear readouts survive entirely undetermined; a decoder-preserving objective recovers up to 11.4% more held-out readout fidelity on GPT-2 without worsening reconstruction.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · deepseek-v4-flash
2026-08-01 17:58 UTC pith:YNKHA3QS
load-bearing objection A clean, honest paper: it proves MSE and L0 don't determine linear-readout fidelity, trains an SAE objective that directly minimizes a ridge-prediction distortion, and transparently shows the gain does not transfer to IOI or concept-probe families.
Decoder-Preserving Sparse Autoencoders: Which Readouts Survive Sparse Compression?
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The paper's central claim is formalized as Proposition 1: for every ridge penalty there are two equally sparse reconstructions with identical reconstruction error and identical isotropic decoder distance whose taskwise distortion operators are incomparable in the Loewner order — each is better than the other on some target. The companion rank relaxation (Theorem 3) shows that an isotropic decoder objective cannot reorder PCA's singular directions under a pure rank bottleneck, only replace variance-proportional omission costs with ridge-saturated ones; any reordering must come from structure like sparse support allocation. A structured task prior, however, can change which modes are retained.
What carries the argument
The central object is the ridge prediction operator K_X = X(X^T X + nλI)^{-1} X^T, which maps any target vector to the predictions of the optimal ridge readout fitted on representation X. Comparing K_X with K_hatX yields the signed operator error and the matrix-valued distortion operator A_X = (K_X - K_hatX)^T(K_X - K_hatX); a task prior with second moment Σ_y scalarizes it as tr(Σ_y A_X). DPSAE adds a normalized finite-group estimate of relative decoder distortion ||(K_X - K_hatX)Y||_F^2 / ||K_X Y||_F^2 to MSE and sparsity terms. Proposition 1 constructs the two-coordinate example showing MSE and L0 leave taskwise distortion underdetermined; Proposition 5 shows zero decoder distance exactly
Load-bearing premise
The empirical claim rests on the assumption that ridge-prediction distortion computed on one 16,384-token held-out group, with the ridge penalty fixed by tr(K_X)/n = 0.25, represents the activation distribution where downstream readouts actually matter; the paper itself notes that its task-ellipsoid bound is not a generalization guarantee for new activation samples.
What would settle it
Train paired MSE and DPSAE SAEs at matched NMSE and L0, then fit a fixed set of independently chosen linear probes (for instance, held-out real semantic labels) on a fresh activation range with ridge penalty selected by cross-validation rather than by tr(K_X)/n. If the DPSAE advantage on those probes is no better than MSE, or if a replication across layers and seeds fails to reproduce the 10.6–11.4% reduction, the central claim of decoder preservation would be falsified. Directly, one could construct the Proposition 1 example at scale and find a downstream task on which the reconstruction with
If this is right
- If reconstruction and sparsity do not determine readout fidelity, SAE benchmarks that report only NMSE and L0 are incomplete; decoder distortion should be reported as a paired metric.
- A task prior can direct sparse capacity toward a declared family of readouts; in the controlled generator, protected-task distortion drops 25.3% with no observed NMSE penalty, so task-specific SAEs are feasible.
- Under an isotropic prior and a pure rank bottleneck, the decoder objective cannot change which modes are kept; observed reordering must come from sparse support and overcompleteness, which guides where to look for differences.
- Readout preservation is a distinct property: it does not imply frozen-network compatibility or concentration of benchmark concepts into a few features, so each downstream use needs its own evaluation.
- The DPSAE objective is cheap enough to be practical: it adds about 21.3% to the isolated optimization step on cached activations.
Where Pith is reading between the lines
- Because the headline gain is measured on random Gaussian target directions and a fixed ridge calibration, a natural extension is to evaluate DPSAE against task priors built from real, independently assembled prediction families; the protection may or may not transfer to those actual readouts.
- The result implies a practical diagnostic for any new sparse autoencoder: plot paired decoder distortion against NMSE across decoder-weight settings; a method that lowers NMSE without lowering readout distortion is not necessarily better for interpretability.
- One could test the mechanism directly by measuring whether the DPSAE gain grows with the proportion of sparse support shared with the protected subspace, or shrinks when the task prior is anticorrelated with the signal directions.
- If readout fidelity is separable from frozen-model behavior, future SAE design may need multiple objectives rather than one: a decoder term for refittable probes, a KL term for frozen behavior, and sparsity for interpretability — and the paper's evidence suggests these cannot be collapsed into a single metric.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. This paper studies the ambiguity left by reconstruction error and sparsity in sparse autoencoders: two SAEs can have equal MSE and L0 but preserve different linear readouts. The authors define a matrix-valued distortion between ridge-prediction operators induced by original and reconstructed activations, prove that MSE and L0 do not determine taskwise fidelity (Prop. 1), analyze a rank relaxation in which an isotropic prior preserves PCA's mode order while a structured prior can reorder modes (Thm. 3, Cor. 4), and introduce DPSAE, which adds an estimate of this distortion to the training loss. Controlled sparse experiments show a structured prior protects its declared task family. On GPT-2 small block 8, DPSAE reduces the exact finite-group decoder distortion by 10.61–11.38% at matched NMSE across three paired runs, while natural-text output-KL noninferiority passes; IOI accuracy is worse and a matched Pythia pair shows no improvement in sparse concept probes. The authors explicitly scope their claims to refitted regularized linear readouts.
Significance. The formal part is valuable: Proposition 1 gives a crisp constructive demonstration that taskwise readout distortion is underdetermined by MSE and L0; Theorem 3 and Corollary 4 are clean applications of Eckart–Young. The empirical protocol is exemplary: predeclared selection rules, paired runs, bootstrap intervals, exact finite-group evaluation, and a clear separation between primary and boundary results. The paper honestly reports the limits of the main result: the LM improvement is measured on the same isotropic ridge-probe family that DPSAE is trained to minimize, and the independent IOI and Pythia concept probes show no transfer. This does not undermine the internal claims, but it means the practical significance is for the declared readout family rather than a general statement about semantically meaningful downstream readouts. Reproducibility artifacts and run manifests strengthen the paper.
minor comments (5)
- [Title and Abstract] The title 'Which Readouts Survive Sparse Compression?' and parts of the abstract could be read as a general answer; the body and Section 3.7 carefully restrict the claims to refitted regularized linear readouts under a declared task prior. A qualifier in the title/abstract would align the packaging with the evidence.
- [§3.6 / Eq. (14)] The symbol \hat R_m in Eq. (13) becomes \hat R_{G,m} in Eq. (14) without definition. Clarify that the latter is the corpus aggregation defined in Appendix A.1.
- [Figure 1 caption] The caption 'with W=I_2 and Z_j = bX_j' is confusing because bX_j is already the reconstruction. Write Z_j = \hat X_j or state that the decoder is the identity.
- [§3.3] The phrase 'PCA's ordering' is informal; specify 'the ordering by singular values σ_i'.
- [§5.3 / Figure 5] The 'declared 5% scale' for materiality is only defined in Appendix A.4; a parenthetical definition in the main text would improve readability.
Circularity Check
No significant circularity: the central underdetermination result is an explicit construction, and the empirical gain is a held-out validation of the disclosed objective, with independent boundary tests that fail to transfer.
full rationale
The paper's load-bearing formal claim (Proposition 1) is proved by an explicit two-dimensional construction (Appendix B: X = aI2, Z1 and Z2 differing in one diagonal coordinate), not by assuming the conclusion. Theorem 3 and Corollary 4 follow from Eckart-Young applied to K_X and K_X Σ_y^{1/2}; no self-citation is load-bearing. The main empirical result reports the same finite-group ridge-prediction distortion that appears in the DPSAE objective (Eq. 14), but this is transparent: the evaluation is performed on a disjoint activation range with a predeclared γ chosen on a separate 25M-token sweep, and the paper explicitly says in Section 3.5 that Eq. 12 is not a generalization guarantee for new activation samples. Optimizing a loss and then measuring that loss on held-out data is a standard validation, not a forced prediction. The paper's boundary tests (IOI accuracy, Pythia sparse-concept probes) use independently specified readout families and show no improvement, which is evidence against a definitional circle: the authors do not redefine 'readout preservation' to match the training objective. The only mild concern is that the isotropic readout family is both the training prior and the success metric, but the paper discloses and limits this scope. No steps meet the standard for circularity.
Axiom & Free-Parameter Ledger
free parameters (3)
- decoder loss weight gamma =
0.03125
- ridge penalty lambda =
1.6049035191535947
- structured prior weight beta =
swept 0.25–4 relative to 2D crossover scale
axioms (4)
- domain assumption Ridge prediction operator K_X = X(X^T X + nλI)^{-1} X^T is the correct model of a refittable linear readout.
- domain assumption Activations and reconstructions share the same feature dimension, and separate ridge refits on X and X_hat are well posed.
- standard math Eckart–Young theorem for rank-r approximation of K_X in the Frobenius norm.
- ad hoc to paper Fixed-radius Gaussian target directions (uniform on sphere) are a reasonable isotropic task prior for language-model readouts.
read the original abstract
Sparse autoencoders (SAEs) compress model activations into sparse codes, but equal reconstruction error and sparsity can preserve different linearly decodable signals. We formalize this ambiguity as a matrix-valued distortion between optimal ridge-prediction operators and train decoder-preserving SAEs by combining this distortion with reconstruction loss. In a rank relaxation, an isotropic task prior saturates per-mode omission costs without changing PCA's ordering, whereas a structured prior can change which modes are retained. A controlled sparse experiment shows that a declared prior protects held-out combinations from its task subspace. On GPT-2 small block 8, DPSAE reduces held-out decoder distortion by 10.6--11.4% across three paired runs while matching reconstruction NMSE. The same checkpoints pass an average natural-text output-KL noninferiority test, but one matched Pythia pair shows no improvement in probes restricted to a few sparse features. These results show that reconstruction quality does not determine which refitted linear readouts survive sparse compression, and that readout preservation is distinct from learning cleaner benchmark concepts or preserving every frozen-model behavior.
Figures
Reference graph
Works this paper leans on
-
[1]
The Twelfth International Conference on Learning Representations , year =
Sparse Autoencoders Find Highly Interpretable Features in Language Models , author =. The Twelfth International Conference on Learning Representations , year =
-
[2]
Transformer Circuits Thread , year =
Towards Monosemanticity: Decomposing Language Models With Dictionary Learning , author =. Transformer Circuits Thread , year =
-
[3]
The Thirteenth International Conference on Learning Representations , year =
Scaling and Evaluating Sparse Autoencoders , author =. The Thirteenth International Conference on Learning Representations , year =
-
[4]
Advances in Neural Information Processing Systems , volume =
Boix-Adser. Advances in Neural Information Processing Systems , volume =. 2022 , url =
2022
-
[5]
Proceedings of UniReps: the Second Edition of the Workshop on Unifying Representations in Neural Models , series =
What Representational Similarity Measures Imply about Decodable Information , author =. Proceedings of UniReps: the Second Edition of the Workshop on Unifying Representations in Neural Models , series =. 2024 , url =
2024
-
[6]
Bussmann, Bart and Leask, Patrick and Nanda, Neel , year =. 2412.06410 , archiveprefix =
-
[7]
Ayonrinde, Kola and Pearce, Michael T. and Sharkey, Lee , year =. Interpretability as Compression: Reconsidering. doi:10.48550/arXiv.2410.11179 , url =. 2410.11179 , archiveprefix =
-
[8]
2019 , url =
Language Models Are Unsupervised Multitask Learners , author =. 2019 , url =
2019
-
[9]
Proceedings of the 40th International Conference on Machine Learning , series =
Pythia: A Suite for Analyzing Large Language Models Across Training and Scaling , author =. Proceedings of the 40th International Conference on Machine Learning , series =. 2023 , url =
2023
-
[10]
Penedo, Guilherme and Kydl. The. Advances in Neural Information Processing Systems , volume =. 2024 , doi =
2024
-
[11]
2026 , eprint =
Manifold Steering Reveals the Shared Geometry of Neural Network Representation and Behavior , author =. 2026 , eprint =
2026
-
[12]
Jumping Ahead: Improving Reconstruction Fidelity with
Rajamanoharan, Senthooran and Lieberum, Tom and Sonnerat, Nicolas and Conmy, Arthur and Varma, Vikrant and Kramar, Janos and Nanda, Neel , year =. Jumping Ahead: Improving Reconstruction Fidelity with. 2407.14435 , archiveprefix =
-
[13]
Karvonen, Adam and Rager, Can and Lin, Johnny and Tigges, Curt and Bloom, Joseph and Chanin, David and Lau, Yeu-Tong and Farrell, Eoin and McDougall, Callum and Ayonrinde, Kola and Wearden, Matthew and Conmy, Arthur and Marks, Samuel and Nanda, Neel , year =. 2503.09532 , archiveprefix =
-
[14]
2026 , eprint =
Are Sparse Autoencoder Benchmarks Reliable? , author =. 2026 , eprint =
2026
-
[15]
Proceedings of the IEEE/CVF International Conference on Computer Vision , pages =
Similarity-Preserving Knowledge Distillation , author =. Proceedings of the IEEE/CVF International Conference on Computer Vision , pages =. 2019 , url =
2019
-
[16]
Advances in Neural Information Processing Systems , volume =
From Flat to Hierarchical: Extracting Sparse Representations with Matching Pursuit , author =. Advances in Neural Information Processing Systems , volume =. 2025 , url =
2025
-
[17]
Advances in Neural Information Processing Systems , volume =
Projecting Assumptions: The Duality Between Sparse Autoencoders and Concept Geometry , author =. Advances in Neural Information Processing Systems , volume =. 2025 , url =
2025
-
[18]
The Fourteenth International Conference on Learning Representations , year =
Temporal Sparse Autoencoders: Leveraging the Sequential Nature of Language for Interpretability , author =. The Fourteenth International Conference on Learning Representations , year =
-
[19]
The Fourteenth International Conference on Learning Representations , year =
Cross-Modal Redundancy and the Geometry of Vision--Language Embeddings , author =. The Fourteenth International Conference on Learning Representations , year =
-
[20]
IEEE Transactions on Pattern Analysis and Machine Intelligence , volume =
Task-Driven Dictionary Learning , author =. IEEE Transactions on Pattern Analysis and Machine Intelligence , volume =. 2012 , doi =
2012
-
[21]
Advances in Neural Information Processing Systems , volume =
Measuring and Guiding Monosemanticity , author =. Advances in Neural Information Processing Systems , volume =. 2025 , url =
2025
-
[22]
Unveiling Decision-Making in
Le Bail, Mathis and Dentan, J. Unveiling Decision-Making in. Findings of the Association for Computational Linguistics: EACL 2026 , pages =. 2026 , doi =
2026
-
[23]
2026 , url =
Yao, Yifei and Zhang, Hanrong and Du, Mengnan , booktitle =. 2026 , url =
2026
-
[24]
Advances in Neural Information Processing Systems , volume =
Identifying Functionally Important Features with End-to-End Sparse Dictionary Learning , author =. Advances in Neural Information Processing Systems , volume =. 2024 , doi =
2024
-
[25]
2025 , eprint =
Revisiting End-To-End Sparse Autoencoder Training: A Short Finetune Is All You Need , author =. 2025 , eprint =
2025
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.