Token embeddings violate the manifold hypothesis
Not on the map. Why?
Abstract (arXiv)
A full understanding of the behavior of a large language model (LLM) requires our grasp of its input token space. If this space differs from our assumptions, our comprehension of and conclusions about the LLM will likely be flawed. We elucidate the structure of the token embeddings both empirically and theoretically. We present a novel statistical test assuming that the neighborhood around each token has a relatively flat and smooth structure as the null hypothesis. Failing to reject the null is uninformative, but rejecting it at a specific token psi implies an irregularity in the token subspace in a psi-neighborhood, B(psi). The structure assumed in the null is a generalization of a manifold with boundary called a smooth fiber bundle (which can be split into two spatial regimes -- small and large radius), so we denote our new hypothesis test as the ``fiber bundle hypothesis.'' By running our test over several open-source LLMs, each with unique token embeddings, we find that the null is frequently rejected, and so the evidence suggests that the token subspace is not a fiber bundle and hence also not a manifold. As a consequence of our findings, when an LLM is presented with two semantically equivalent prompts, if one prompt contains a token implicated by our test, the response to that prompt will likely exhibit less stability than the other.
Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Builds on 34 of its 70 references that are in the map. First posted on arXiv 1 Apr 2025.
Builds on
34 of 70 references.
References · 34
- 2018UMAP: Uniform Manifold Approximation and Projection for Dimension Reductioncited by 2,841
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2023Jailbreaking Black Box Large Language Models in Twenty Queriescited by 1,326
- 2023Quantifying Language Models' Sensitivity to Spurious Features in Prompt Design or: How I learned to start worrying about prompt formattingcited by 778
- 2023On the Planning Abilities of Large Language Models : A Critical Investigationcited by 308
- 2018Estimating the intrinsic dimension of datasets by a minimal neighborhood informationcited by 246
- 2024Scaling Large Language Model-based Multi-Agent Collaborationcited by 235
- 2023A mathematical perspective on Transformerscited by 160
- 2024Does Prompt Formatting Have Any Impact on LLM Performance?cited by 132
- 2023Intrinsic Dimension Estimation for Robust Detection of AI-Generated Textscited by 111
- 2019Understanding and Improving Transformer From a Multi-Particle Dynamic System Point of Viewcited by 108
- 2024Ensemble Learning for Heterogeneous Large Language Models with Deep Parallel Collaborationcited by 66
22 more
- 2023Hidden symmetries of ReLU networkscited by 38
- 2025Shared Global and Local Geometry of Language Model Embeddingscited by 34
- 2024LLMs learn governing principles of dynamical systems, revealing an in-context neural scaling lawcited by 32
- 2024Glitch Tokens in Large Language Models: Categorization Taxonomy and Effective Detectioncited by 27
- 2024Implicit Geometry of Next-token Prediction: From Language Sparsity Patterns to Model Representationscited by 24
- 2021An enriched category theory of language: from syntax to semanticscited by 21
- 2023Limits for Learning with Language Modelscited by 17
- 2024On the Brittle Foundations of ReAct Prompting for Agentic Large Language Modelscited by 17
- 2020Topology of Word Embeddings: Singularities Reflect Polysemycited by 15
- 2024The Unreasonable Effectiveness of Eccentric Automatic Promptscited by 12
- 2022Topological Singularity Detection at Multiple Scalescited by 11
- 2024The structure of the token space for large language modelscited by 10
- 2022Transparency Helps Reveal When Language Models Learn Meaningcited by 8
- 2020Estimating the reach of a manifold via its convexity defect functioncited by 5
- 2024The Information of Large Language Model Geometrycited by 5
- 2025The magnitude of categories of texts enriched by language modelscited by 5
- 2025Measuring Intrinsic Dimension of Token Embeddingscited by 5
- 2023HADES: Fast Singularity Detection with Local Measure Comparisoncited by 4
- 2023A Language and Its Dimensions: Intrinsic Dimensions of Language Fractal Structurescited by 4
- 2024Building Intelligence Identification System via Large Language Model Watermarking: A Survey and Beyondcited by 4
- 2024Understanding Generative AI Content with Embedding Modelscited by 4
- 2025Probing the topology of the space of tokens with structured promptscited by 3
Built on by
Citing papers per year: 2025: 4; 2026: 10.
14 papers.
Sort:
By area:
- 2025Beyond Exponential Decay: Rethinking Error Accumulation in Large Language Modelscited by 11
- 2026The Information Geometry of Softmax: Probing and Steeringcited by 9
- 2025Routing Manifold Alignment Improves Generalization of Mixture-of-Experts LLMscited by 6
- 2026Semantic Tube Prediction: Beating LLM Data Efficiency with JEPAcited by 6
- 2026Latent Semantic Manifolds in Large Language Modelscited by 5
- 2026Revisiting Anisotropy in Language Transformers: The Geometry of Learning Dynamicscited by 3
- 2025Exploring the Stratified Space Structure of an RL Game with the Volume Growth Transformcited by 1
- 2025Advancing mathematics research with generative AIcited by 1
- 2026Decoding in Geometry: Alleviating Embedding-Space Crowding for Complex Reasoningcited by 1
- 2026Stratifying Reinforcement Learning with Signal Temporal Logiccited by 0
- 2026Rethinking Intrinsic Dimension Estimation in Neural Representationscited by 0
- 2026Riemannian Geometry for Pre-trained Language Model Embeddingscited by 0
- 2026A Hub of Short Rows Inflates Intrinsic Dimension Estimation of Token Embeddings30 Aug 2026
- 2026Geometric Filtering of LLM-Generated Samples for Few-Shot Text Classification14 Aug 2026
- 2026Riemannian Geometry for Pre-trained Language Model Embeddings8 Jul 2026
- 2026Rethinking Intrinsic Dimension Estimation in Neural Representations22 Apr 2026
- 2026Revisiting Anisotropy in Language Transformers: The Geometry of Learning Dynamics9 Apr 2026
- 2026Stratifying Reinforcement Learning with Signal Temporal Logic6 Apr 2026
- 2026Latent Semantic Manifolds in Large Language Models17 Mar 2026
- 2026Semantic Tube Prediction: Beating LLM Data Efficiency with JEPA26 Feb 2026
- 2026The Information Geometry of Softmax: Probing and Steering17 Feb 2026
- 2026Decoding in Geometry: Alleviating Embedding-Space Crowding for Complex Reasoning30 Jan 2026
- 2025Routing Manifold Alignment Improves Generalization of Mixture-of-Experts LLMs10 Nov 2025
- 2025Advancing mathematics research with generative AI29 Sep 2025
- 2026The Information Geometry of Softmax: Probing and Steeringcited by 9
- 2026Semantic Tube Prediction: Beating LLM Data Efficiency with JEPAcited by 6
- 2026Latent Semantic Manifolds in Large Language Modelscited by 5
- 2026Revisiting Anisotropy in Language Transformers: The Geometry of Learning Dynamicscited by 3
- 2025Exploring the Stratified Space Structure of an RL Game with the Volume Growth Transformcited by 1
- 2026Stratifying Reinforcement Learning with Signal Temporal Logiccited by 0
- 2026Rethinking Intrinsic Dimension Estimation in Neural Representationscited by 0
- 2026Riemannian Geometry for Pre-trained Language Model Embeddingscited by 0
- 2026Geometric Filtering of LLM-Generated Samples for Few-Shot Text Classificationcited by 0
- 2026A Hub of Short Rows Inflates Intrinsic Dimension Estimation of Token Embeddingscited by 0
- 2026A Hub of Short Rows Inflates Intrinsic Dimension Estimation of Token Embeddings30 Aug 2026
- 2026Geometric Filtering of LLM-Generated Samples for Few-Shot Text Classification14 Aug 2026
- 2026Riemannian Geometry for Pre-trained Language Model Embeddings8 Jul 2026
- 2026Rethinking Intrinsic Dimension Estimation in Neural Representations22 Apr 2026
- 2026Revisiting Anisotropy in Language Transformers: The Geometry of Learning Dynamics9 Apr 2026
- 2026Stratifying Reinforcement Learning with Signal Temporal Logic6 Apr 2026
- 2026Latent Semantic Manifolds in Large Language Models17 Mar 2026
- 2026Semantic Tube Prediction: Beating LLM Data Efficiency with JEPA26 Feb 2026
- 2026The Information Geometry of Softmax: Probing and Steering17 Feb 2026
- 2025Exploring the Stratified Space Structure of an RL Game with the Volume Growth Transform29 Jul 2025