Representational Capacity: Geometric Limits on Feature Representation in Transformer Language Models
Not on the map. Why?
Abstract (arXiv)
Model dimension (dmodel) is a fundamental hyperparameter in transformer language models, yet its role in setting the geometric limits of feature representation remains under-explored. Grounded in the Linear Representation and Superposition Hypotheses - which propose that models encode features as near-orthogonal directions in latent space - we develop a framework for estimating how many such directions a model can support. We first establish the embedding matrix as a measurable proxy for near-orthogonality constraints across the latent space: the boundary between meaningful token relationships and incidental similarity in the pairwise cosine similarity distribution gives a concrete estimate of the model's accepted deviation varepsilon from perfect orthogonality. Applying this metric across dozens of open-source models reveals two classes: models with high varepsilon whose embeddings lack near-orthogonal structure, and models with low varepsilon that maintain it. We then show that the standard Johnson-Lindenstrauss lemma greatly underestimates the packing efficiency of trained representations, and derive an adjusted capacity formula in which the number of near-orthogonal directions depends on the ratio of vectors to dimensions (k/d) rather than the raw count - a single modification that cuts prediction error by two orders of magnitude with no extra parameters. Combining these results, we define representational capacity as an upper bound on the number of distinguishable directions available for features and embeddings in a model's latent space. Capacity is exponentially sensitive to varepsilon, and larger models favor tighter orthogonality constraints over maximizing raw capacity - a pattern compatible with several explanations (a stability-capacity trade-off, a ceiling on usable concepts, or confounds with model scale) that we leave to future work.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 10 of its 25 references that are in the map. First posted on arXiv 1 Jun 2026.
Builds on
10 of 25 references.
References · 10
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2023Mistral 7Bcited by 3,164
- 2024Qwen2 Technical Reportcited by 2,120
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
- 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,917
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
- 2022Toy Models of Superpositioncited by 971
- 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 716
- 2024TinyLlama: An Open-Source Small Language Modelcited by 618
- 2026Scaling Monosemanticity: Extracting Interpretable Features from Claude 3 Sonnetcited by 191
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Global Divergence, Local Convergence: Representation Geometry in SSMs and Transformers
- 2025The Origins of Representation Manifolds in Large Language Models
- 2026Causal Dimensionality of Transformer Representations: Measurement, Scaling, and Layer Structure
- 2024Exploring the Impact of a Transformer's Latent Space Geometry on Downstream Task Performance
- 2026Dispersion Loss Counteracts Embedding Condensation and Improves Generalization in Small Language Models
- 2026Geometric Factual Recall in Transformers
- 2025Bridging the Dimensional Chasm: Uncover Layer-wise Dimensional Reduction in Transformers through Token Correlation
- 2025Beyond Components: Singular Vector-Based Interpretability of Transformer Circuits