mlsanityData to 7 Oct 2026

How Do LLMs Use Their Depth?

arXiv 2510.18871 ↗First posted 21 Oct 2025cs.CL cs.AIcited by 15

Not on the map. Why?

Abstract (arXiv)

Growing evidence suggests that large language models do not use their depth uniformly, yet we still lack a fine-grained understanding of their layer-wise prediction dynamics. In this paper, we trace the intermediate representations of several open-weight models during inference and reveal a structured and nuanced use of depth. Specifically, we propose a "Guess-then-Refine" framework that explains how LLMs internally structure their computations to make predictions. We first show that the top-ranked predictions in early LLM layers are composed primarily of high-frequency tokens, which act as statistical guesses proposed by the model due to the lack of contextual information. As contextual information develops deeper into the model, these initial guesses get refined into contextually appropriate tokens. We then examine the dynamic usage of layer depth through three case studies. (i) Multiple-choice task analysis shows that the model identifies appropriate options within the first half of the model and finalizes the response in the latter half. (ii) Fact recall task analysis shows that in a multi-token answer, the first token requires more computational depth than the rest. (iii) Part-of-speech analysis shows that function words are, on average, the earliest to be predicted correctly. To validate our results, we supplement probe-based analyses with causal manipulations in the form of activation patching and early-exiting experiments. Together, our results provide a detailed view of depth usage in LLMs, shedding light on the layer-by-layer computations that underlie successful predictions and providing insights for future works to improve computational efficiency in transformer-based models.

Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Builds on 31 of its 43 references that are in the map. First posted on arXiv 21 Oct 2025.

Builds on

31 of 43 references.

References · 31

  1. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  2. 2016Layer Normalizationcited by 5,579
  3. 2022Locating and Editing Factual Associations in GPTcited by 2,609
  4. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  5. 2019Root Mean Square Layer Normalizationcited by 1,425
  6. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
  7. 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
  8. 2022Transformers learn in-context by gradient descentcited by 714
  9. 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
  10. 2021Datasets: A Community Library for Natural Language Processingcited by 526
  11. 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
  12. 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 512
19 more
  1. 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 369
  2. 2023MQuAKE: Assessing Knowledge Editing in Language Models via Multi-Hop Questionscited by 292
  3. 2023Label Words are Anchors: An Information Flow Perspective for Understanding In-Context Learningcited by 235
  4. 2019Depth-Adaptive Transformercited by 227
  5. 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 219
  6. 2022Why Can GPT Learn In-Context? Language Models Implicitly Perform Gradient Descent as Meta-Optimizerscited by 204
  7. 2023Linearity of Relation Decoding in Transformer Language Modelscited by 190
  8. 2017Adaptive Neural Networks for Efficient Inferencecited by 161
  9. 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden Statecited by 116
  10. 2023Language Models Implement Simple Word2Vec-style Vector Arithmeticcited by 97
  11. 2024Not All Layers of LLMs Are Necessary During Inferencecited by 95
  12. 2023Jump to Conclusions: Short-Cutting Transformers With Linear Transformationscited by 89
  13. 2025Do Language Models Use Their Depth Efficiently?cited by 48
  14. 2024Do language models plan ahead for future tokens?cited by 47
  15. 2024Evidence of Learned Look-Ahead in a Chess-Playing Neural Networkcited by 33
  16. 2023DecoderLens: Layerwise Interpretation of Encoder-Decoder Transformerscited by 22
  17. 2024Unlocking the Future: Exploring Look-Ahead Planning Mechanistic Interpretability in Large Language Modelscited by 21
  18. 2024Looking Beyond The Top-1: Transformers Determine Top Tokens In Ordercited by 12
  19. 2024Racing Thoughts: Explaining Contextualization Errors in Large Language Modelscited by 11

Built on by

Cited by year (2026 to 7 Oct)

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

15 papers.

Sort:

By area:

Similar, not cited

Close in wording, no citation either way

  1. 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoning
  2. 2025What Affects the Effective Depth of Large Language Models?
  3. 2025On the Fundamental Limits of LLMs at Scale
  4. 2024Does Representation Matter? Exploring Intermediate Layers in Large Language Models
  5. 2024Do LLMs Overcome Shortcut Learning? An Evaluation of Shortcut Challenges in Large Language Models
  6. 2025Pitfalls of Scale: Investigating the Inverse Task of Redefinition in Large Language Models
  7. 2024Understanding the role of FFNs in driving multilingual behaviour in LLMs
  8. 2026Understanding Large Language Models