How Do LLMs Use Their Depth?
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
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2016Layer Normalizationcited by 5,579
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
- 2022Transformers learn in-context by gradient descentcited by 714
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
- 2021Datasets: A Community Library for Natural Language Processingcited by 526
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 522
- 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 512
19 more
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 369
- 2023MQuAKE: Assessing Knowledge Editing in Language Models via Multi-Hop Questionscited by 292
- 2023Label Words are Anchors: An Information Flow Perspective for Understanding In-Context Learningcited by 235
- 2019Depth-Adaptive Transformercited by 227
- 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 219
- 2022Why Can GPT Learn In-Context? Language Models Implicitly Perform Gradient Descent as Meta-Optimizerscited by 204
- 2023Linearity of Relation Decoding in Transformer Language Modelscited by 190
- 2017Adaptive Neural Networks for Efficient Inferencecited by 161
- 2023Future Lens: Anticipating Subsequent Tokens from a Single Hidden Statecited by 116
- 2023Language Models Implement Simple Word2Vec-style Vector Arithmeticcited by 97
- 2024Not All Layers of LLMs Are Necessary During Inferencecited by 95
- 2023Jump to Conclusions: Short-Cutting Transformers With Linear Transformationscited by 89
- 2025Do Language Models Use Their Depth Efficiently?cited by 48
- 2024Do language models plan ahead for future tokens?cited by 47
- 2024Evidence of Learned Look-Ahead in a Chess-Playing Neural Networkcited by 33
- 2023DecoderLens: Layerwise Interpretation of Encoder-Decoder Transformerscited by 22
- 2024Unlocking the Future: Exploring Look-Ahead Planning Mechanistic Interpretability in Large Language Modelscited by 21
- 2024Looking Beyond The Top-1: Transformers Determine Top Tokens In Ordercited by 12
- 2024Racing Thoughts: Explaining Contextualization Errors in Large Language Modelscited by 11
Built on by
Citing papers per year: 2025: 1; 2026: 14.
15 papers.
Sort:
By area:
- 2026Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillationcited by 11
- 2026Inverse Depth Scaling From Most Layers Being Similarcited by 8
- 2026Good Arguments Against the People Pleasers: How Reasoning Mitigates (Yet Masks) LLM Sycophancycited by 7
- 2026Deterministic Differentiable Structured Pruning for Large Language Modelscited by 2
- 2026Reasoning-preserved Efficient Distillation of Large Language Models via Activation-aware Initializationcited by 2
- 2026Deeper is Not Always Better: Mitigating the Alignment Tax via Confident Layer Decodingcited by 2
- 2025PAS : Prelim Attention Score for Detecting Object Hallucinations in Large Vision--Language Modelscited by 1
- 2026RBCorr: Response Bias Correction in Language Modelscited by 1
- 2026OISD: On-Policy Internal Self-Distillation of Language Modelscited by 1
- 2026Attention Degradation, Function Token Anchoring, and the Limits of Attention-Based Intervention in Large Language Modelscited by 1
- 2026Do Transformers Use their Depth Adaptively? Evidence from a Relational Reasoning Taskcited by 0
- 2026Child-directed speech facilitates production, not comprehension, in BabyLMscited by 0
- 2026Deep Learning Models Also Recall Features21 Aug 2026
- 2026Attention Degradation, Function Token Anchoring, and the Limits of Attention-Based Intervention in Large Language Models9 Jul 2026
- 2026From Execution to Education: A Bloom-Aligned Framework for Measuring Educational Control in LLMs9 Jul 2026
- 2026Deeper is Not Always Better: Mitigating the Alignment Tax via Confident Layer Decoding20 Jun 2026
- 2026Child-directed speech facilitates production, not comprehension, in BabyLMs31 May 2026
- 2026Reasoning-preserved Efficient Distillation of Large Language Models via Activation-aware Initialization28 May 2026
- 2026OISD: On-Policy Internal Self-Distillation of Language Models27 May 2026
- 2026The Hard Decision Layer: Evidence for Committed Inference in Transformers25 May 2026
- 2026Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillation10 May 2026
- 2026Do Transformers Use their Depth Adaptively? Evidence from a Relational Reasoning Task14 Apr 2026
- 2026Good Arguments Against the People Pleasers: How Reasoning Mitigates (Yet Masks) LLM Sycophancy17 Mar 2026
- 2026Deterministic Differentiable Structured Pruning for Large Language Models9 Mar 2026
3 more
- 2026Good Arguments Against the People Pleasers: How Reasoning Mitigates (Yet Masks) LLM Sycophancycited by 7
- 2026Reasoning-preserved Efficient Distillation of Large Language Models via Activation-aware Initializationcited by 2
- 2026OISD: On-Policy Internal Self-Distillation of Language Modelscited by 1
- 2026Do Transformers Use their Depth Adaptively? Evidence from a Relational Reasoning Taskcited by 0
- 2026Reasoning-preserved Efficient Distillation of Large Language Models via Activation-aware Initialization28 May 2026
- 2026OISD: On-Policy Internal Self-Distillation of Language Models27 May 2026
- 2026Do Transformers Use their Depth Adaptively? Evidence from a Relational Reasoning Task14 Apr 2026
- 2026Good Arguments Against the People Pleasers: How Reasoning Mitigates (Yet Masks) LLM Sycophancy17 Mar 2026
- 2026Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillationcited by 11
- 2026Inverse Depth Scaling From Most Layers Being Similarcited by 8
- 2026Deterministic Differentiable Structured Pruning for Large Language Modelscited by 2
- 2026RBCorr: Response Bias Correction in Language Modelscited by 1
- 2026Attention Degradation, Function Token Anchoring, and the Limits of Attention-Based Intervention in Large Language Modelscited by 1
- 2026Child-directed speech facilitates production, not comprehension, in BabyLMscited by 0
- 2026From Execution to Education: A Bloom-Aligned Framework for Measuring Educational Control in LLMscited by 0
- 2026The Hard Decision Layer: Evidence for Committed Inference in Transformerscited by 0
- 2026Deep Learning Models Also Recall Featurescited by 0
- 2026Deep Learning Models Also Recall Features21 Aug 2026
- 2026Attention Degradation, Function Token Anchoring, and the Limits of Attention-Based Intervention in Large Language Models9 Jul 2026
- 2026From Execution to Education: A Bloom-Aligned Framework for Measuring Educational Control in LLMs9 Jul 2026
- 2026Child-directed speech facilitates production, not comprehension, in BabyLMs31 May 2026
- 2026The Hard Decision Layer: Evidence for Committed Inference in Transformers25 May 2026
- 2026Cornerstones or Stumbling Blocks? Deciphering the Rock Tokens in On-Policy Distillation10 May 2026
- 2026Deterministic Differentiable Structured Pruning for Large Language Models9 Mar 2026
- 2026RBCorr: Response Bias Correction in Language Models12 Feb 2026
- 2026Inverse Depth Scaling From Most Layers Being Similar5 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoning
- 2025What Affects the Effective Depth of Large Language Models?
- 2025On the Fundamental Limits of LLMs at Scale
- 2024Does Representation Matter? Exploring Intermediate Layers in Large Language Models
- 2024Do LLMs Overcome Shortcut Learning? An Evaluation of Shortcut Challenges in Large Language Models
- 2025Pitfalls of Scale: Investigating the Inverse Task of Redefinition in Large Language Models
- 2024Understanding the role of FFNs in driving multilingual behaviour in LLMs
- 2026Understanding Large Language Models