Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoning
On the map: LLM reasoning → Reasoning benchmarks and evaluation
Abstract (arXiv)
Recent studies suggest that the deeper layers of Large Language Models (LLMs) contribute little to representation learning and can often be removed without significant performance loss. However, such claims are typically drawn from narrow evaluations and may overlook important aspects of model behavior. In this work, we present a systematic study of depth utilization across diverse dimensions, including evaluation protocols, task categories, and model architectures. Our analysis confirms that very deep layers are generally less effective than earlier ones, but their contributions vary substantially with the evaluation setting. Under likelihood-based metrics without generation, pruning most layers preserves performance, with only the initial few being critical. By contrast, generation-based evaluation uncovers indispensable roles for middle and deeper layers in enabling reasoning and maintaining long-range coherence. We further find that knowledge and retrieval are concentrated in shallow components, whereas reasoning accuracy relies heavily on deeper layers -- yet can be reshaped through distillation. These results highlight that depth usage in LLMs is highly heterogeneous and context-dependent, underscoring the need for task-, metric-, and model-aware perspectives in both interpreting and compressing large models.
Key facts. Cited by 15 papers in the map (data through 14 Oct 2026). Placed in Reasoning benchmarks and evaluation, one of 17 directions in LLM reasoning. Builds on 30 of its 38 references that are in the map. First posted on arXiv 2 Oct 2025.
Builds on
30 of 38 references.
Closest ancestors · 1
Same direction, newest first
Other references · 29
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2023Qwen Technical Reportcited by 3,344
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2022Beyond the Imitation Game: Quantifying and extrapolating the capabilities of language modelscited by 1,998
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
17 more
- 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 662
- 2020KILT: a Benchmark for Knowledge Intensive Language Taskscited by 573
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
- 2025Layer by Layer: Uncovering Hidden Representations in Language Modelscited by 311
- 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 226
- 2024Compact Language Models via Pruning and Knowledge Distillationcited by 148
- 2020Low-Rank Bottleneck in Multi-head Attention Modelscited by 89
- 2025The Curse of Depth in Large Language Modelscited by 62
- 2023Spike No More: Stabilizing the Pre-training of Large Language Modelscited by 52
- 2024Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LNcited by 37
- 2024AlphaPruning: Using Heavy-Tailed Self Regularization Theory for Improved Layer-wise Pruning of Large Language Modelscited by 36
- 2022Random Fully Connected Neural Networks as Perturbatively Solvable Hierarchiescited by 24
- 2024A deeper look at depth pruning of LLMscited by 23
- 2025Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanismcited by 15
- 2024Change Is the Only Constant: Dynamic LLM Slicing based on Layer Redundancycited by 9
- 2025Layer Importance for Mathematical Reasoning is Forged in Pre-Training and Invariant after Post-Trainingcited by 9
Built on by
Citing papers per year: 2025: 1; 2026: 14.
15 papers.
Sort:
By area:
- 2025When Fewer Layers Break More Chains: Layer Pruning Harms Test-Time Scaling in LLMscited by 5
- 2026On the Limits of Layer Pruning for Generative Reasoning in Large Language Modelscited by 3
- 2026When Context Returns: Toward Robust Internalization in On-Policy Distillationcited by 3
- 2026Is One Layer Enough? Training A Single Transformer Layer Can Match Full-Parameter RL Trainingcited by 3
- 2026MTA: Multi-Granular Trajectory Alignment for Large Language Model Distillationcited by 2
- 2026From One-to-One to Many-to-Many: Dynamic Cross-Layer Injection for Deep Vision-Language Fusioncited by 1
- 2026Not All Layers Need Tuning: Selective Layer Restoration Recovers Diversitycited by 1
- 2026Layer-wise Representation Dynamics: An Empirical Investigation Across Embedders and Base LLMscited by 1
- 2026OI-Bench: An Option Injection Benchmark for Evaluating LLM Susceptibility to Directive Interferencecited by 0
- 2026AdaCultureSafe: Adaptive Cultural Safety Grounded by Cultural Knowledge in Large Language Modelscited by 0
- 2026TAPIOCA: Why Task- Aware Pruning Improves OOD model Capabilitycited by 0
- 2026Understanding helpfulness and harmless tension in reward modelscited by 0
- 2026Attention-Aware Routing: Coupling Routing and Attention in MoEs17 Sep 2026
- 2026What Does Layer-Importance Reveal About Transformers and State-Space Models?15 Sep 2026
- 2026Is One Layer Enough? Training A Single Transformer Layer Can Match Full-Parameter RL Training1 Jul 2026
- 2026Peeking Inside LLMs: Leveraging Internal Artifacts of LLMs for Enhancing Reliability in Legal Classification18 Jun 2026
- 2026Understanding helpfulness and harmless tension in reward models11 Jun 2026
- 2026When Context Returns: Toward Robust Internalization in On-Policy Distillation10 Jun 2026
- 2026TAPIOCA: Why Task- Aware Pruning Improves OOD model Capability14 May 2026
- 2026Layer-wise Representation Dynamics: An Empirical Investigation Across Embedders and Base LLMs12 May 2026
- 2026MTA: Multi-Granular Trajectory Alignment for Large Language Model Distillation2 May 2026
- 2026AdaCultureSafe: Adaptive Cultural Safety Grounded by Cultural Knowledge in Large Language Models9 Mar 2026
- 2026Not All Layers Need Tuning: Selective Layer Restoration Recovers Diversity6 Feb 2026
- 2026On the Limits of Layer Pruning for Generative Reasoning in Large Language Models2 Feb 2026
3 more
- 2026OI-Bench: An Option Injection Benchmark for Evaluating LLM Susceptibility to Directive Interference19 Jan 2026
- 2026From One-to-One to Many-to-Many: Dynamic Cross-Layer Injection for Deep Vision-Language Fusion15 Jan 2026
- 2025When Fewer Layers Break More Chains: Layer Pruning Harms Test-Time Scaling in LLMs25 Oct 2025
- 2026Is One Layer Enough? Training A Single Transformer Layer Can Match Full-Parameter RL Trainingcited by 3
- 2026Not All Layers Need Tuning: Selective Layer Restoration Recovers Diversitycited by 1
- 2026Layer-wise Representation Dynamics: An Empirical Investigation Across Embedders and Base LLMscited by 1
- 2026OI-Bench: An Option Injection Benchmark for Evaluating LLM Susceptibility to Directive Interferencecited by 0
- 2026AdaCultureSafe: Adaptive Cultural Safety Grounded by Cultural Knowledge in Large Language Modelscited by 0
- 2026TAPIOCA: Why Task- Aware Pruning Improves OOD model Capabilitycited by 0
- 2026Peeking Inside LLMs: Leveraging Internal Artifacts of LLMs for Enhancing Reliability in Legal Classificationcited by 0
- 2026Is One Layer Enough? Training A Single Transformer Layer Can Match Full-Parameter RL Training1 Jul 2026
- 2026Peeking Inside LLMs: Leveraging Internal Artifacts of LLMs for Enhancing Reliability in Legal Classification18 Jun 2026
- 2026TAPIOCA: Why Task- Aware Pruning Improves OOD model Capability14 May 2026
- 2026Layer-wise Representation Dynamics: An Empirical Investigation Across Embedders and Base LLMs12 May 2026
- 2026AdaCultureSafe: Adaptive Cultural Safety Grounded by Cultural Knowledge in Large Language Models9 Mar 2026
- 2026Not All Layers Need Tuning: Selective Layer Restoration Recovers Diversity6 Feb 2026
- 2026OI-Bench: An Option Injection Benchmark for Evaluating LLM Susceptibility to Directive Interference19 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2024Exploring Concept Depth: How Large Language Models Acquire Knowledge and Concept at Different Layers?
- 2025DeepQuestion: Systematic Generation of Real-World Challenges for Evaluating LLMs Performance
- 2025How Do LLMs Use Their Depth?
- 2024Does Representation Matter? Exploring Intermediate Layers in Large Language Models
- 2025What Affects the Effective Depth of Large Language Models?
- 2023Large Language Models Are Not Strong Abstract Reasoners
- 2026From Interpretability to Performance: Optimizing Retrieval Heads for Long-Context Language Models
- 2024Beyond Accuracy: Evaluating the Reasoning Behavior of Large Language Models -- A Survey