mlsanityData to 14 Oct 2026

Freeze Deep, Train Shallow: Interpretable Layer Allocation for Continued Pre-Training

arXiv 2605.11416 ↗First posted 12 May 2026cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Selective layer-wise updates are essential for low-cost continued pre-training of Large Language Models (LLMs), yet determining which layers to freeze or train remains an empirical black-box problem due to the lack of interpretable guidance. To address this issue, we propose LayerTracer, an architecture-agnostic diagnostic framework that reveals the evolution patterns of layer-wise representations and stability by locating task execution positions and quantifying layer sensitivity. Analysis results reveal that deep layers act as critical regions for task execution and maintain high stability against disruptive updates. Guided by this finding, we conduct three controlled continued pre-training trials to compare diverse freeze-train strategies, demonstrating that training shallow layers while freezing deep layers consistently outperforms full-parameter fine-tuning and the opposite allocation on both C-Eval and CMMLU benchmarks. We further present a hybrid model case study, which validates that placing high-quality pre-trained modules in deep layers effectively preserves inherent knowledge of the model. This work delivers a low-cost and interpretable solution for resource-constrained teams, offering actionable guidance for layer-wise parameter allocation in continued pre-training and hybrid model construction.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 30 of its 41 references that are in the map. First posted on arXiv 12 May 2026.

Builds on

30 of 41 references.

References · 30

  1. 2017Attention Is All You Needcited by 60,361
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2024The Llama 3 Herd of Modelscited by 14,715
  4. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  5. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  6. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
  7. 2023Qwen Technical Reportcited by 3,344
  8. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  9. 2023Code Llama: Open Foundation Models for Codecited by 2,139
  10. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  11. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
  12. 2022Mass-Editing Memory in a Transformercited by 1,001
18 more
  1. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 876
  2. 2023C-Eval: A Multi-Level Multi-Discipline Chinese Evaluation Suite for Foundation Modelscited by 708
  3. 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 532
  4. 2023CMMLU: Measuring massive multitask language understanding in Chinesecited by 449
  5. 2024Jamba: A Hybrid Transformer-Mamba Language Modelcited by 401
  6. 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 334
  7. 2024BioMistral: A Collection of Open-Source Pretrained Large Language Models for Medical Domainscited by 301
  8. 2020Information-Theoretic Probing for Linguistic Structurecited by 207
  9. 2023Linearity of Relation Decoding in Transformer Language Modelscited by 200
  10. 2023Linear attention is (maybe) all you need (to understand transformer optimization)cited by 98
  11. 2023HuatuoGPT-II, One-stage Training for Medical Adaption of LLMscited by 92
  12. 2023DISC-MedLLM: Bridging General Large Language Models and Real-World Medical Consultationcited by 90
  13. 2024How Large Language Models Encode Context Knowledge? A Layer-Wise Probing Studycited by 64
  14. 2026Locate, Steer, and Improve: A Practical Survey of Actionable Mechanistic Interpretability in Large Language Modelscited by 27
  15. 2025Overclocking LLM Reasoning: Monitoring and Controlling Thinking Path Lengths in LLMscited by 22
  16. 2017Distinguishing Antonyms and Synonyms in a Pattern-based Neural Networkcited by 19
  17. 2025Parameter-Efficient Fine-Tuning of Large Language Models via Deconvolution in Subspacecited by 11
  18. 2025Analyzing LLMs' Knowledge Boundary Cognition Across Languages Through the Lens of Internal Representationscited by 6

Built on by

Not cited yet.