Scaling Linear Mode Connectivity and Merging to Billion Parameter Pretrained Transformers
Not on the map. Why?
Abstract (arXiv)
Linear mode connectivity (LMC) provides a promising foundation for understanding and merging independently trained neural networks, but existing methods typically optimize the interpolation path from only one model endpoint, limiting their scalability and effectiveness for large pretrained transformers. We propose a novel and scalable framework for enabling LMC-based model merging to em billion-parameter pretrained transformers. Our method applies properly parameterized functionality-preserving weight transformations to align functionally equivalent solutions, and introduces a dual learning procedure in which both models jointly learn their corresponding transformations toward a shared linear interpolation path. This bidirectional optimization substantially reduces interpolation barriers and enables more reliable merging across large-scale architectures. Empirically, we show that our approach achieves near-zero loss barriers on WikiText for language models with medium-sized parameters, representing, to our knowledge, the first demonstration of near-barrier-free linear connectivity at this scale. In the vision domain, ViT-L maintains above 69% ImageNet top-1 accuracy throughout the interpolation path, while modern billion-parameter LLMs exhibit only small loss barriers. These results suggest that properly resolving parameter symmetries enables large pretrained Transformers to be connected and merged through simple linear paths with substantially improved interpolation performance. Code: https://github.com/VILA-Lab/Dual-Learned-Matching .
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 31 of its 35 references that are in the map. First posted on arXiv 22 Jun 2026.
Builds on
31 of 35 references.
References · 31
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2022Editing Models with Task Arithmeticcited by 1,120
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
- 2024OLMo: Accelerating the Science of Language Modelscited by 613
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 588
- 2024Large Language Models: A Surveycited by 525
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
- 2024SliceGPT: Compress Large Language Models by Deleting Rows and Columnscited by 385
19 more
- 2024The Platonic Representation Hypothesiscited by 382
- 2018Essentially No Barriers in Neural Network Energy Landscapecited by 330
- 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 291
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 254
- 2015Convergent Learning: Do different neural networks learn the same representations?cited by 220
- 2019Model Fusion via Optimal Transportcited by 219
- 2020Optimizing Mode Connectivity via Neuron Alignmentcited by 86
- 2019Weight-space symmetry in deep networks gives rise to permutation saddles, connected by equal-loss valleys across the loss landscapecited by 63
- 2023Going Beyond Linear Mode Connectivity: The Layerwise Linear Feature Connectivitycited by 50
- 2024Neural Network Diffusioncited by 42
- 2025Symmetry in Neural Network Parameter Spacescited by 36
- 2023Layer-wise Linear Mode Connectivitycited by 22
- 2026A Survey of Weight Space Learning: Understanding, Representation, and Generationcited by 18
- 2024Merging Text Transformer Models from Different Initializationscited by 16
- 2025Beyond the Permutation Symmetry of Transformers: The Role of Rotation for Model Fusioncited by 16
- 2025Generalized Linear Mode Connectivity for Transformerscited by 16
- 2025Generative Modeling of Weights: Generalization or Memorization?cited by 14
- 2025The Butterfly Effect: Neural Network Training Trajectories Are Highly Sensitive to Initial Conditionscited by 9
- 2025On Linear Mode Connectivity of Mixture-of-Experts Architecturescited by 8
Built on by
Citing papers per year: 2026: 2.
2 papers.