用部分重初始化提升稀疏专家模型训练效率,长期性能超越传统方法。
Drop-Upcycling: Training Sparse Mixture of Experts with Partial Re-initialization
- 结合预训练稠密模型知识与局部权重重初始化,促进专家分工。
- 在百亿级令牌训练下,5.9B活跃参数模型媲美13B稠密模型,训练量减至1/4。
- 适合追求高效大模型训练的研究者和工业部署团队。
稀疏专家混合模型(MoE)相比同容量稠密模型显著降低训练与推理成本。上行改造(Upcycling)利用预训练稠密模型初始化并训练MoE,虽初期表现优异,但训练速度慢于从零训练,长期性能欠佳。本文提出Drop-Upcycling:融合预训练知识与部分权重统计重初始化,策略性推动专家专业化,大幅提升知识获取效率。大规模实验表明,在训练超过百亿令牌时,该方法显著优于此前的MoE构建方式。所提出的5.9B活跃参数模型在同等模型族中达到13B稠密模型性能,训练所需浮点运算量约减少至1/4。所有实验资源(代码、数据、检查点、日志)均公开,支持可复现性与未来研究。
原文摘要 · Abstract (English)
The Mixture of Experts (MoE) architecture reduces the training and inference cost significantly compared to a dense model of equivalent capacity. Upcycling is an approach that initializes and trains an MoE model using a pre-trained dense model. While upcycling leads to initial performance gains, the training progresses slower than when trained from scratch, leading to suboptimal performance in the long term. We propose Drop-Upcycling - a method that effectively addresses this problem. Drop-Upcycling combines two seemingly contradictory approaches: utilizing the knowledge of pre-trained dense models while statistically re-initializing some parts of the weights. This approach strategically promotes expert specialization, significantly enhancing the MoE model's efficiency in knowledge acquisition. Extensive large-scale experiments demonstrate that Drop-Upcycling significantly outperforms previous MoE construction methods in the long term, specifically when training on hundreds of billions of tokens or more. As a result, our MoE model with 5.9B active parameters achieves comparable performance to a 13B dense model in the same model family, while requiring approximately 1/4 of the training FLOPs. All experimental resources, including source code, training data, model checkpoints and logs, are publicly available to promote reproducibility and future research on MoE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。