Model Merging in Pre-training of Large Language Models

Jing Liu (Central University of Finance and Economics) · Yunshui Li (ByteDance) · Yiyuan Ma (ByteDance Inc.) · Shen Yan (ByteDance Seed) · Chaoyi Zhang (ByteDance Inc.) · Jianqiao Lu (University of Hong Kong) · Ziwen Xu (ByteDance Inc.) · Mengzhao Chen (University of Hong Kong) · Minrui Wang (ByteDance Inc.) · Shiyi Zhan (ByteDance Inc.) · Jin Ma (ByteDance Inc.) · Xunhao Lai (Peking University) · Yao Luo (ByteDance Inc.) · Xingyan Bin (ByteDance Inc.) · Hongbin Ren (ByteDance Inc.) · Mingji Han (ByteDance Inc.) · Wenhao Hao (ByteDance Inc.) · Bairen Yi (ByteDance Inc.) · LingJun Liu (ByteDance Inc.) · Bole Ma (ByteDance Inc.) · Xiaoying Jia (ByteDance Inc.) · zhou Xun (ByteDance Inc.) · liang xiang (ByteDance Inc.) · Yonghui Wu (Google)
ablation studiesannealing behaviorcheckpointsexperimental analysishyperparameterslearning ratesmerging strategiesmixture-of-expertsmodel developmentmodel mergingopen-source communityperformance improvementspre-trainingtraining costs

Model merging has emerged as a promising technique for enhancing large language models, though its application in large-scale pre-training remains relatively unexplored. In this paper, we present a comprehensive investigation of model merging techniques during the pre-training process. Through extensive experiments with both dense and Mixture-of-Experts (MoE) architectures ranging from millions to over 100 billion parameters, we demonstrate that merging checkpoints trained with constant learning rates not only achieves significant performance improvements but also enables accurate prediction of annealing behavior. These improvements lead to both more efficient model development and significantly lower training costs. Our detailed ablation studies on merging strategies and hyperparameters provide new insights into the underlying mechanisms while uncovering novel applications. Through comprehensive experimental analysis, we offer the open-source community practical pre-training guidelines for effective model merging.