mlsanityData to 14 Oct 2026

Causal Dimensionality of Transformer Representations: Measurement, Scaling, and Layer Structure

arXiv 2605.08740 ↗First posted 9 May 2026cs.LG cs.AIcited by 0

On the map: Mechanistic interpretability → Sparse autoencoder training and evaluation

Abstract (arXiv)

Sparse autoencoders (SAEs) decompose transformer residual streams into interpretable feature dictionaries, yet the relationship between SAE width and causal influence on model output has not been systematically characterised. We introduce causal dimensionality kappa(L, M, T), defined as the effective rank of the expected Jacobian outer product at layer L, and show it can be estimated via the SAE width sweep paired with attribution patching. Across seven SAE widths from 16,384 to 1,048,576 features on Gemma-2-2B layer 12, representational capacity grows 15.6x while causal capacity grows only 4.35x: a robust separation we term the representational-causal wedge. A saturating fit yields kappa-hat approximately 1,990 with kappa-hat / d_model = 0.86 and participation-ratio lower bound kappa_PR approximately 280. Crucially, kappa is invariant to model scaling: Gemma-2-9B and Gemma-2-2B yield identical N_causal = 328 at the same SAE width despite a 3.46x parameter increase (the count is forced to 2% of SAE width by calibration; the substantive empirical claim is shape invariance of the AtP score distribution under matched seq=512 conditions). Across eight network depths kappa is constant while the absolute attribution threshold drops 20x from layer 1 to layer 23. Five controls (architecture invariance, threshold robustness, geometric privilege, synthetic ground-truth recovery, and a four-cell encoder/decoder ablation) pin down what kappa measures and what it does not. Our findings establish kappa as a measurable, model-intrinsic property of transformer layers: sub-linearly recoverable by SAE width, invariant to model scaling, and structured across network depth.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Sparse autoencoder training and evaluation, one of 13 directions in Mechanistic interpretability. Builds on 15 of its 22 references that are in the map. First posted on arXiv 9 May 2026.

Builds on

15 of 22 references.

Closest ancestors · 1

Same direction, newest first

  1. 2022Toy Models of Superposition21 Sep

Other references · 14

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  3. 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 724
  4. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 716
  5. 2024Scaling and evaluating sparse autoencoderscited by 553
  6. 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 435
  7. 2019Intrinsic dimension of data representations in deep neural networkscited by 282
  8. 2013k-Sparse Autoencoderscited by 215
  9. 2024A is for Absorption: Studying Feature Splitting and Absorption in Sparse Autoencoderscited by 160
  10. 2023The Quantization Model of Neural Scalingcited by 138
  11. 2025SAEBench: A Comprehensive Benchmark for Sparse Autoencoders in Language Model Interpretabilitycited by 124
  12. 2024AtP*: An efficient and scalable method for localizing LLM behaviour to componentscited by 111
2 more
  1. 2024Emergence of a High-Dimensional Abstraction Phase in Language Transformerscited by 56
  2. 2022Rank Diminishing in Deep Neural Networkscited by 47

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024Interpreting Attention Layer Outputs with Sparse Autoencoders
  2. 2026From Geometric Recovery to Causal Validation: A Reproducible Audit of Sparse Autoencoder Features, from Superposition Geometry to Causal Inertness
  3. 2026From Correlation to Cause: A Five-Stage Methodology for Feature Analysis in Transformer Language Models
  4. 2024Residual Stream Analysis with Multi-Layer SAEs
  5. 2026Comparing Latent Concept Formation in State Space Models and Transformers via Sparse Autoencoders
  6. 2025Sparsification and Reconstruction from the Perspective of Representation Geometry
  7. 2024Analyzing (In)Abilities of SAEs via Formal Languages
  8. 2025Automated Interpretability Metrics Do Not Distinguish Trained and Random Transformers