Understanding Layer Patching in Model Size Interpolation
Not on the map. Why?
Abstract (arXiv)
Zero-shot model size interpolation aims to create new models of intermediate target sizes by combining existing models without additional training. Recent work on boomerang distillation [Kangaslahti et al., 2026] shows that a student language model distilled from a larger teacher can be expanded by iteratively patching its layers, replacing student layers with contiguous blocks of teacher layers to obtain models whose size and performance interpolate between the student and the teacher. In this work, we provide the first systematic study of student-layer selection for model size interpolation. We cast finding the optimal layer subset for each model size as an optimization problem and prove it can be viewed as a shortest-path problem in a certain acyclic graph. In experiments, we show that patching strongly shapes interpolation behavior, with effects that vary substantially across model families. We find that simple sequential strategies--patching either from the first layer to the last or from the last to the first--often achieve surprisingly strong performance in practice. We further introduce KLPatch, a greedy patching algorithm based on KL divergence, which often improves over last-to-first patching and approximately solves the optimization problem. Together, our results provide a principled understanding of how layer patching affects model size interpolation and offer practical guidance for constructing near-optimal interpolated models.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 44 of its 53 references that are in the map. First posted on arXiv 9 Jul 2026.
Builds on
44 of 53 references.
References · 44
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2025Qwen3 Technical Reportcited by 8,195
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
32 more
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,201
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2023Instruction-Following Evaluation for Large Language Modelscited by 1,143
- 2017RACE: Large-scale ReAding Comprehension Dataset From Examinationscited by 851
- 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 776
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2026Kimi K2.5: Visual Agentic Intelligencecited by 449
- 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
- 2018AMC: AutoML for Model Compression and Acceleration on Mobile Devicescited by 375
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 369
- 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 285
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2019Uniform convergence may be unable to explain generalization in deep learningcited by 223
- 2019Model Fusion via Optimal Transportcited by 217
- 2024Compact Language Models via Pruning and Knowledge Distillationcited by 142
- 2022Same Pre-training Loss, Better Downstream: Implicit Bias Matters for Language Modelscited by 84
- 2024LLM Pruning and Distillation in Practice: The Minitron Approachcited by 82
- 2025A Comprehensive Survey on Knowledge Distillationcited by 49
- 2024FinerCut: Finer-grained Interpretable Layer Pruning for Large Language Modelscited by 26
- 2025Instruction-Following Pruning for Large Language Modelscited by 14
- 2024100K or 100 Days: Trade-offs when Pre-Training with Academic Resourcescited by 6
- 2026Rethinking Layer Relevance in Large Language Models Beyond Cosine Similaritycited by 4
- 2025Boomerang Distillation Enables Zero-Shot Model Size Interpolationcited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.