mlsanityData to 14 Oct 2026

Geometry of Semantics in Next-Token Prediction: How Optimization Implicitly Organizes Linguistic Representations

arXiv 2505.08348 ↗First posted 13 May 2025cs.CLcited by 2

Not on the map. Why?

Abstract (arXiv)

We investigate how next-token prediction (NTP) optimization leads language models to extract and organize semantic structure from text. Our analysis, based on a tractable mathematical model and controlled synthetic data, reveals that NTP implicitly guides models to factor a centered support matrix encoding context-to-next-token co-occurrence patterns via singular value decomposition (SVD). While models never explicitly construct this matrix, learned word and context embeddings converge to its SVD factors, with singular vectors encoding latent semantic concepts through their sign patterns. We demonstrate that concepts corresponding to larger singular values are learned earlier during training, yielding a natural semantic hierarchy where broad categories emerge before fine-grained ones. This insight motivates orthant-based clustering, a method that combines concept signs to identify interpretable semantic categories. We validate our findings on synthetic datasets and pretrained language models, recovering diverse semantic structures such as grammatical categories, named entity types, and topical distinctions (medical, entertainment). Our work bridges classical distributional semantics and neural collapse geometry, characterizing how gradient-based optimization implicitly determines both the matrix representation and factorization method that encode semantic structure.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 42 of its 62 references that are in the map. First posted on arXiv 13 May 2025.

Builds on

42 of 62 references.

References · 42

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2013Efficient Estimation of Word Representations in Vector Spacecited by 4,375
  3. 2013Distributed Representations of Words and Phrases and their Compositionalitycited by 3,474
  4. 2023Qwen Technical Reportcited by 3,344
  5. 2016Pointer Sentinel Mixture Modelscited by 3,096
  6. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
  7. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  8. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
  9. 2022Toy Models of Superpositioncited by 971
  10. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
  11. 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
  12. 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 716
30 more
  1. 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 712
  2. 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
  3. 2019Designing and Interpreting Probes with Control Taskscited by 640
  4. 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 449
  5. 2019Visualizing and Measuring the Geometry of BERTcited by 216
  6. 2018A mathematical theory of semantic development in deep neural networkscited by 186
  7. 2021Exploring Deep Neural Networks via Layer-Peeled Model: Minority Collapse in Imbalanced Trainingcited by 166
  8. 2016Linear Algebraic Structure of Word Senses, with Applications to Polysemycited by 165
  9. 2024The Geometry of Categorical and Hierarchical Concepts in Large Language Modelscited by 159
  10. 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 158
  11. 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 152
  12. 2020Neural collapse with unconstrained featurescited by 142
  13. 2020Evaluation of Neural Architectures Trained with Square Loss vs Cross-Entropy in Classification Taskscited by 138
  14. 2018Regularization Matters: Generalization and Optimization of Neural Nets v.s. their Induced Kernelcited by 131
  15. 2019Implicit Regularization of Discrete Gradient Dynamics in Linear Neural Networkscited by 124
  16. 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 102
  17. 2022Imbalance Trouble: Revisiting Neural-Collapse Geometrycited by 80
  18. 2019Analogies Explained: Towards Understanding Word Embeddingscited by 64
  19. 2020Gradient descent follows the regularization path for general lossescited by 59
  20. 2018Towards Understanding Linear Word Analogiescited by 57
  21. 2020On the emergence of simplex symmetry in the final and penultimate layers of neural network classifierscited by 52
  22. 2016Empath: Understanding Topic Signals in Large-Scale Textcited by 51
  23. 2015Sparse Overcomplete Word Vector Representationscited by 44
  24. 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
  25. 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Datacited by 38
  26. 2018SemAxis: A Lightweight Framework to Characterize Domain-Specific Word Semantics Beyond Sentimentcited by 26
  27. 2024Implicit Geometry of Next-token Prediction: From Language Sparsity Patterns to Model Representationscited by 24
  28. 2019Word Embedding Visualization Via Dictionary Learningcited by 12
  29. 2024The Exploration of Neural Collapse under Imbalanced Datacited by 4
  30. 2023On the Relationship between Sentence Analogy Identification and Sentence Structure Encoding in Large Language Modelscited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.