mlsanityData to 14 Oct 2026

Layer Importance for Mathematical Reasoning is Forged in Pre-Training and Invariant after Post-Training

arXiv 2506.22638 ↗First posted 27 Jun 2025cs.LG cs.AIcited by 9

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

  1. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  2. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  3. 2023Qwen Technical Reportcited by 3,344
  4. 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
  5. 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 712
  6. 2023Linear Representations of Sentiment in Large Language Modelscited by 163
  7. 2024The Remarkable Robustness of LLMs: Stages of Inference?cited by 146
  8. 2024Streamlining Redundant Layers to Compress Large Language Modelscited by 35

Built on by

Cited by year (2026 to 14 Oct)

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

9 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Unveiling the Reasoning Process of Large Language Models
  2. 2024From Next-Token to Mathematics: The Learning Dynamics of Mathematical Reasoning in Language Models
  3. 2024Revisiting the Superficial Alignment Hypothesis
  4. 2026Beyond Compression: Diagnosing How Post-Training Changes Mathematical Reasoning
  5. 2025Scaling Reasoning, Losing Control: Evaluating Instruction Following in Large Reasoning Models
  6. 2026Disentangling Mathematical Reasoning in LLMs: A Methodological Investigation of Internal Mechanisms
  7. 2026NeuReasoner: Theory-grounded Mapping of Reasoning Elicitation Boundaries
  8. 2026Reasoning as State Transition: A Representational Analysis of Reasoning Evolution in Large Language Models