Freeze Deep, Train Shallow: Interpretable Layer Allocation for Continued Pre-Training
This work addresses the lack of effective, interpretable, and low-cost guidance for determining layer-wise freezing or training strategies during continual pre-training of large language models. The authors propose LayerTracer, a framework that identifies task-executing layers, quantifies layer sensitivity, and analyzes the stability of representational evolution across layers. Their analysis reveals that deeper layers are primarily responsible for task execution and exhibit greater robustness to perturbations. Based on these insights, they introduce an interpretable and cost-efficient continual pre-training paradigm—freezing deeper layers while fine-tuning shallower ones—and extend it to hybrid model construction. The approach is architecture-agnostic and, through representation analysis and controlled experiments, consistently outperforms full-parameter fine-tuning and alternative strategies on C-Eval and CMMLU benchmarks, while demonstrating that embedding high-quality deep modules effectively preserves original knowledge.