mlsanityData to 7 Oct 2026

Extra-Merge: Tracing the Rank-1 Subspace of Model Merging in Language Model Pre-Training

arXiv 2605.26484 ↗First posted 26 May 2026cs.LGcited by 2

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  3. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  4. 2020Scaling Laws for Neural Language Modelscited by 6,411
  5. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  6. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  7. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  8. 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
  9. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  10. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  11. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
  12. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
37 more
  1. 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
  2. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
  3. 2022Editing Models with Task Arithmeticcited by 1,076
  4. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
  5. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
  6. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
  7. 2023Baichuan 2: Open Large-scale Language Modelscited by 664
  8. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
  9. 2021Merging Models with Fisher-Weighted Averagingcited by 613
  10. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
  11. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
  12. 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 384
  13. 2025Muon is Scalable for LLM Trainingcited by 381
  14. 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 364
  15. 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 248
  16. 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
  17. 2018Gradient Descent Happens in a Tiny Subspacecited by 208
  18. 2024Evolutionary Optimization of Model Merging Recipescited by 204
  19. 2014Qualitatively characterizing neural network optimization problemscited by 186
  20. 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 160
  21. 2023Composing Parameter-Efficient Modules with Arithmetic Operationscited by 155
  22. 2022Model Ratatouille: Recycling Diverse Models for Out-of-Distribution Generalizationcited by 97
  23. 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 75
  24. 2022Stop Wasting My Time! Saving Days of ImageNet and BERT Training with Latest Weight Averagingcited by 54
  25. 2019Emergent properties of the local geometry of neural loss landscapescited by 46
  26. 2023Early Weight Averaging meets High Learning Rates for LLM Pre-trainingcited by 36
  27. 2025Model Merging in Pre-training of Large Language Modelscited by 36
  28. 2024Does SGD really happen in tiny subspaces?cited by 35
  29. 2021Analyzing Monotonic Linear Interpolation in Neural Network Loss Landscapescited by 27
  30. 2021What can linear interpolation of neural network loss landscapes tell us?cited by 27
  31. 2024Checkpoint Merging via Bayesian Optimization in LLM Pretrainingcited by 27
  32. 2025The Sharpness Disparity Principle in Transformers for Accelerating Language Model Pre-Trainingcited by 27
  33. 2017Checkpoint Ensembles: Ensemble Methods from a Single Training Processcited by 24
  34. 2025WSM: Decay-Free Learning Rate Schedule via Checkpoint Merging for LLM Pre-trainingcited by 21
  35. 2018Using Mode Connectivity for Loss Landscape Analysiscited by 19
  36. 2024Improving Generalization and Convergence by Enhancing Implicit Regularizationcited by 14
  37. 2025How Learning Rate Decay Wastes Your Best Data in Curriculum-Based LLM Pretrainingcited by 12

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.