arXiv:2410.01610cs.CLcs.AI2024-10ACL被引 6

用少量数据将普通模型升级为专家混合模型,提升效率与性能。

Upcycling Instruction Tuning from Dense to Mixture-of-Experts via Parameter Merging

  • 利用指令微调中的中间检查点构建专用专家。
  • 通过参数融合与遗传算法扩展专家,实现灵活数量配置。
  • 仅需少量种子数据优化路由,适合资源有限场景。

混合专家(MoE)在大语言模型中表现优异,但现有从密集模型转换为MoE的方法需大量数据且依赖大规模后训练。本文提出一种高效方法UpIT,将密集预训练模型转化为MoE指令模型。首先指出指令微调过程中的中间检查点天然适合作为专用专家,进而引入专家扩展阶段,通过遗传算法和参数融合确保新扩展专家的多样性。为保障每个专家正常工作,使用少量各专家擅长的种子数据对路由器进行预优化。在多种数据规模和升级设置下的实验表明,UpIT具有卓越性能与数据效率,并在专家或数据扩展时保持稳定提升。进一步分析揭示了在升级过程中保持专家多样性的关键作用。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) shines brightly in large language models (LLMs) and demonstrates outstanding performance in plentiful natural language processing tasks. However, existing methods transforming LLMs from dense to MoE face significant data requirements and typically rely on large-scale post-training. In this paper, we propose Upcycling Instruction Tuning (UpIT), a data-efficient approach for tuning a dense pre-trained model into a MoE instruction model. Specifically, we first point out that intermediate checkpoints during instruction tuning of the dense model are naturally suitable for specialized experts, and then propose an expert expansion stage to flexibly achieve models with flexible numbers of experts, where genetic algorithm and parameter merging are introduced to ensure sufficient diversity of new extended experts. To ensure that each specialized expert in the MoE model works as expected, we select a small amount of seed data that each expert excels to pre-optimize the router. Extensive experiments with various data scales and upcycling settings demonstrate the outstanding performance and data efficiency of UpIT, as well as stable improvement in expert or data scaling. Further analysis reveals the importance of ensuring expert diversity in upcycling.

MoE模型升级数据高效专家融合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。