Train What You Deploy: Closing the MLP Reachability Gap in Low-Rank Clone Distillation
Not on the map. Why?
Abstract (arXiv)
A compressed student has two shapes that need not agree: the weight it deploys at inference and the weight family its training can reach. We show that a state-of-the-art weight-inheritance distiller, Low-Rank Clone (LRC), deploys a full-width student MLP but ties training to a teacher-induced slice, leaving 62.5-81.4% of each deployed matrix's independent linear degrees of freedom unreachable-paid for at inference, never trainable. Our principle is one line: train what you deploy. From the identical LRC warm start, we make the training object the entire deployed matrix, with no change in deployed shape, deployed parameter count, or inference FLOPs, via two mergeable realizations (Dense-LRC and CORE-LRC) that both collapse to one deployed weight. This recovers stranded capacity: taking the stronger realization per teacher, +2.36/+2.71/+10.45 Avg9 over matched-budget plain-LRC baselines across three teachers (Llama3.2-3B, Llama3.1-8B, Qwen2.5-3B), with the largest gain on the widest teacher (Qwen), where it reaches the original recipe's approx. 20B-token accuracy at 10B tokens (2x token efficiency); there the strictly same-lineage arm still recovers +6.39, the fully controlled figure. Controls strongly support attributing the gain to the enlarged reachable set, rather than to added parameters or the recipe. From approx. 10B distillation tokens plus a short SFT, a half-parameter 1.5B student matches its approx. 9T-token teacher's 9-task macro-average, within evaluation noise and with a residual MMLU deficit, and a 2.7B student beats Meta's own official compression of Llama3.1-8B at ~900x fewer compression tokens (a token count under unmatched recipes, not a compute claim). All results are from single-seed runs on the LRC backbone.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 17 of its 21 references that are in the map. First posted on arXiv 2 Sep 2026.
Builds on
17 of 21 references.
References · 17
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,463
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
- 2018Born Again Neural Networkscited by 546
- 2021RepVGG: Making VGG-style ConvNets Great Againcited by 522
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 415
- 2024Compact Language Models via Pruning and Knowledge Distillationcited by 148
- 2024MiLoRA: Harnessing Minor Singular Components for Parameter-Efficient LLM Finetuningcited by 94
- 2024Scaling Smart: Accelerating Large Language Model Pre-training with Small Model Initializationcited by 29
- 2026On the Geometry of On-Policy Distillationcited by 19
5 more
- 2025A Token is Worth over 1,000 Tokens: Efficient Knowledge Distillation through Low-Rank Clonecited by 15
- 2025Spectral Scaling Laws in Language Models: How Effectively Do Feed-Forward Networks Use Their Latent Space?cited by 8
- 2026Demystifying Low-Rank Knowledge Distillation in Large Language Models: Convergence, Generalization, and Information-Theoretic Guaranteescited by 3
- 2026Same Architecture, Different Capacity: Optimizer-Induced Spectral Scaling Lawscited by 2
- 2026Reasoning-preserved Efficient Distillation of Large Language Models via Activation-aware Initializationcited by 2
Built on by
Not cited yet.