Layer Importance for Mathematical Reasoning is Forged in Pre-Training and Invariant after Post-Training
On the map: LLM reasoning → Mathematical reasoning
also in 1 more
Abstract (arXiv)
Large language models improve at math after instruction tuning, reinforcement learning, or knowledge distillation. We ask whether these gains come from major changes in the transformer layers or from smaller adjustments that keep the original structure. Using layer-wise ablation on base and trained variants, we find that math reasoning depends on a few critical layers, which stay important across all post-training methods. Removing these layers reduces math accuracy by as much as 80%, whereas factual recall tasks only show relatively smaller drops. This suggests that specialized layers for mathematical tasks form during pre-training and remain stable afterward. As measured by Normalized Mutual Information (NMI), we find that near these critical layers, tokens drift from their original syntactic clusters toward representations aligned with tokens less syntactically related but potentially more useful for downstream task.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Placed in Mathematical reasoning, one of 17 directions in LLM reasoning. Builds on 8 of its 33 references that are in the map. First posted on arXiv 27 Jun 2025.
Builds on
8 of 33 references.
References · 8
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2023Qwen Technical Reportcited by 3,344
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 712
- 2023Linear Representations of Sentiment in Large Language Modelscited by 163
- 2024The Remarkable Robustness of LLMs: Stages of Inference?cited by 146
- 2024Streamlining Redundant Layers to Compress Large Language Modelscited by 35
Built on by
Citing papers per year: 2025: 1; 2026: 8.
9 papers.
- 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoningcited by 15
- 2026Towards a Mechanistic Understanding of Large Reasoning Models: A Survey of Training, Inference, and Failurescited by 6
- 2026What does RL improve for Visual Reasoning? A Frankenstein-Style Analysiscited by 4
- 2026On the Limits of Layer Pruning for Generative Reasoning in Large Language Modelscited by 3
- 2026Is One Layer Enough? Training A Single Transformer Layer Can Match Full-Parameter RL Trainingcited by 3
- 2026Diagnosing FP4 inference: a layer-wise and block-wise sensitivity analysis of NVFP4 and MXFP4cited by 2
- 2026Distributed Multi-Layer Editing for Rule-Level Knowledge in Large Language Modelscited by 0
- 2026What Does Layer-Importance Reveal About Transformers and State-Space Models?cited by 0
- 2026Persistent Depth Ordering amid Shifting Block-Bypass Responses in Language Model Pretrainingcited by 0
- 2026Persistent Depth Ordering amid Shifting Block-Bypass Responses in Language Model Pretraining1 Oct 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
- 2026Distributed Multi-Layer Editing for Rule-Level Knowledge in Large Language Models9 Apr 2026
- 2026Diagnosing FP4 inference: a layer-wise and block-wise sensitivity analysis of NVFP4 and MXFP45 Mar 2026
- 2026What does RL improve for Visual Reasoning? A Frankenstein-Style Analysis12 Feb 2026
- 2026On the Limits of Layer Pruning for Generative Reasoning in Large Language Models2 Feb 2026
- 2026Towards a Mechanistic Understanding of Large Reasoning Models: A Survey of Training, Inference, and Failures11 Jan 2026
- 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoning2 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2026Unveiling the Reasoning Process of Large Language Models
- 2024From Next-Token to Mathematics: The Learning Dynamics of Mathematical Reasoning in Language Models
- 2024Revisiting the Superficial Alignment Hypothesis
- 2026Beyond Compression: Diagnosing How Post-Training Changes Mathematical Reasoning
- 2025Scaling Reasoning, Losing Control: Evaluating Instruction Following in Large Reasoning Models
- 2026Disentangling Mathematical Reasoning in LLMs: A Methodological Investigation of Internal Mechanisms
- 2026NeuReasoner: Theory-grounded Mapping of Reasoning Elicitation Boundaries
- 2026Reasoning as State Transition: A Representational Analysis of Reasoning Evolution in Large Language Models