让百亿参数大模型通过线性路径无缝连接并合并
Scaling Linear Mode Connectivity and Merging to Billion Parameter Pretrained Transformers

- 双向优化双模型变换,对齐功能等价参数
- 语言模型在WikiText上近乎零损失屏障
- 适合研究大模型合并与参数对齐的学者
线性模式连通性(LMC)为理解与合并独立训练的神经网络提供了有前景的基础,但现有方法通常仅从一个模型端点优化插值路径,限制了其在大规模预训练变压器中的可扩展性和有效性。我们提出一种新型且可扩展的框架,实现基于LMC的模型合并应用于百亿参数预训练变换器。该方法应用适当的、保持功能的权重变换来对齐功能等价解,并引入双学习过程,使两个模型共同学习各自的变换以逼近共享的线性插值路径。这种双向优化显著降低了插值障碍,实现了大规模架构上的更可靠合并。实验表明,我们的方法在中等参数量语言模型上于WikiText实现了近乎零的损失屏障,据我们所知,这是该规模下首次实现近无屏障的线性连通性。在视觉领域,ViT-L在整个插值路径中保持超过69%的ImageNet top-1准确率,而现代百亿参数大语言模型仅表现出微小的损失屏障。结果表明,合理解决参数对称性可使大型预训练变压器通过简单线性路径实现高效连接与合并。代码:https://github.com/VILA-Lab/Dual-Learned-Matching。
原文摘要 · Abstract (English)
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 .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。