mlsanityData to 7 Oct 2026

Dynamics of Spontaneous Topic Changes in Next Token Prediction with Self-Attention

arXiv 2501.06382 ↗First posted 10 Jan 2025cs.CL cs.AI stat.MLcited by 2

Not on the map. Why?

Abstract (arXiv)

Human cognition is punctuated by abrupt, spontaneous shifts between topics-driven by emotional, contextual, or associative cues-a phenomenon known as spontaneous thought in neuroscience. In contrast, self-attention based models depend on structured patterns over their inputs to predict each next token, lacking spontaneity. Motivated by this distinction, we characterize spontaneous topic changes in self-attention architectures, revealing both their similarities and their divergences from spontaneous human thought. First, we establish theoretical results under a simplified, single-layer self-attention model with suitable conditions by defining the topic as a set of Token Priority Graphs (TPGs). Specifically, we demonstrate that (1) the model maintains the priority order of tokens related to the input topic, (2) a spontaneous topic change can occur only if lower-priority tokens outnumber all higher-priority tokens of the input topic, and (3) unlike human cognition, the longer context length or the more ambiguous input topic reduces the likelihood of spontaneous change. Second, we empirically validate that these dynamics persist in modern, state-of-the-art LLMs, underscoring a fundamental disparity between human cognition and AI behaviour in the context of spontaneous topic changes. To the best of our knowledge, no prior work has explored these questions with a focus as closely aligned to human thought.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 40 of its 69 references that are in the map. First posted on arXiv 10 Jan 2025.

Builds on

40 of 69 references.

References · 40

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020Language Models are Few-Shot Learnerscited by 32,199
  4. 2024The Llama 3 Herd of Modelscited by 14,192
  5. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  6. 2024GPT-4o System Cardcited by 4,467
  7. 2022Survey of Hallucination in Natural Language Generationcited by 2,056
  8. 2020Reformer: The Efficient Transformercited by 1,658
  9. 2020Rethinking Attention with Performerscited by 1,584
  10. 2020Linformer: Self-Attention with Linear Complexitycited by 1,462
  11. 2022Least-to-Most Prompting Enables Complex Reasoning in Large Language Modelscited by 1,397
  12. 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 994
28 more
  1. 2022In-context Learning and Induction Headscited by 930
  2. 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
  3. 2020On Faithfulness and Factuality in Abstractive Summarizationcited by 895
  4. 2022Transformers learn in-context by gradient descentcited by 714
  5. 2024Better & Faster Large Language Models via Multi-token Predictioncited by 344
  6. 2016Using the Output Embedding to Improve Language Modelscited by 266
  7. 2023Think before you speak: Training Language Models With Pause Tokenscited by 254
  8. 2020Experience Grounds Languagecited by 241
  9. 2024The pitfalls of next-token predictioncited by 156
  10. 2019Adaptive Attention Span in Transformerscited by 148
  11. 2021Recent Advances in Deep Learning Based Dialogue Systems: A Systematic Surveycited by 130
  12. 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden Statecited by 116
  13. 2023How Do Transformers Learn Topic Structure: Towards a Mechanistic Understandingcited by 85
  14. 2023Synthetic Prompting: Generating Chain-of-Thought Demonstrations for Large Language Modelscited by 61
  15. 2023Max-Margin Token Selection in Attention Mechanismcited by 57
  16. 2024Mechanics of Next Token Prediction with Self-Attentioncited by 34
  17. 2021TIAGE: A Benchmark for Topic-Shift Aware Dialog Modelingcited by 22
  18. 2023Why Softmax Attention Outperforms Linear Attentioncited by 15
  19. 2024Exploring the Frontiers of Softmax: Provable Optimization, Applications in Diffusion Model, and Beyondcited by 15
  20. 2024A Law of Next-Token Prediction in Large Language Modelscited by 15
  21. 2022Language models are better than humans at next-token predictioncited by 14
  22. 2024Training Nonlinear Transformers for Chain-of-Thought Inference: A Theoretical Generalization Analysiscited by 14
  23. 2020TAN-NTM: Topic Attention Networks for Neural Topic Modelingcited by 10
  24. 2024Optimizing Attention with Mirror Descent: Generalized Max-Margin Token Selectioncited by 9
  25. 2023Multi-Granularity Prompts for Topic Shift Detection in Dialoguecited by 6
  26. 2024Towards Understanding the Word Sensitivity of Attention Layers: A Study via Random Featurescited by 5
  27. 2021An Empirical Study of Topic Transition in Dialoguecited by 4
  28. 2024MP2D: An Automated Topic Shift Dialogue Generation Framework Leveraging Knowledge Graphscited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 1; 2026: 1.

2 papers.