Extra-Merge: Tracing the Rank-1 Subspace of Model Merging in Language Model Pre-Training
Not on the map. Why?
Abstract (arXiv)
Model merging has emerged as a lightweight paradigm for enhancing Large Language Models (LLMs), yet its underlying mechanisms remain poorly understood. In this work, we analyze late-stage pre-training trajectories and uncover a Rank-1 Subspace phenomenon: while raw optimization steps oscillate violently, consecutive merged checkpoints collapse onto a stable, approximately one-dimensional linear manifold. We theoretically ground this observation in a river-valley landscape analysis: averaging acts as a geometric low-pass filter that dampens high-curvature noise to reveal the optimal descent direction. Capitalizing on this insight, we propose Extra-Merge, a training-free strategy that extrapolates along this subspace to minimize loss without additional gradient updates. Extensive experiments across GPT-2 and LLaMA families (124M to 2B) demonstrate that Extra-Merge consistently outperforms standard merging baselines. Notably, it yields consistent zero-shot accuracy gains on Pythia-12B downstream tasks and generalizes effectively to the Muon optimizer citepjordan2024muon.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 49 of its 59 references that are in the map. First posted on arXiv 26 May 2026.
Builds on
49 of 59 references.
References · 49
- 2017Attention Is All You Needcited by 59,932
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
37 more
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
- 2023Baichuan 2: Open Large-scale Language Modelscited by 664
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
- 2021Merging Models with Fisher-Weighted Averagingcited by 613
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 384
- 2025Muon is Scalable for LLM Trainingcited by 381
- 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 364
- 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 248
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2018Gradient Descent Happens in a Tiny Subspacecited by 208
- 2024Evolutionary Optimization of Model Merging Recipescited by 204
- 2014Qualitatively characterizing neural network optimization problemscited by 186
- 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 160
- 2023Composing Parameter-Efficient Modules with Arithmetic Operationscited by 155
- 2022Model Ratatouille: Recycling Diverse Models for Out-of-Distribution Generalizationcited by 97
- 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 75
- 2022Stop Wasting My Time! Saving Days of ImageNet and BERT Training with Latest Weight Averagingcited by 54
- 2019Emergent properties of the local geometry of neural loss landscapescited by 46
- 2023Early Weight Averaging meets High Learning Rates for LLM Pre-trainingcited by 36
- 2025Model Merging in Pre-training of Large Language Modelscited by 36
- 2024Does SGD really happen in tiny subspaces?cited by 35
- 2021Analyzing Monotonic Linear Interpolation in Neural Network Loss Landscapescited by 27
- 2021What can linear interpolation of neural network loss landscapes tell us?cited by 27
- 2024Checkpoint Merging via Bayesian Optimization in LLM Pretrainingcited by 27
- 2025The Sharpness Disparity Principle in Transformers for Accelerating Language Model Pre-Trainingcited by 27
- 2017Checkpoint Ensembles: Ensemble Methods from a Single Training Processcited by 24
- 2025WSM: Decay-Free Learning Rate Schedule via Checkpoint Merging for LLM Pre-trainingcited by 21
- 2018Using Mode Connectivity for Loss Landscape Analysiscited by 19
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularizationcited by 14
- 2025How Learning Rate Decay Wastes Your Best Data in Curriculum-Based LLM Pretrainingcited by 12
Built on by
Citing papers per year: 2026: 2.
2 papers.