无需训练即可融合多个专家模型,动态构建高性能统一语言模型。
Training-Free Dynamic Upcycling of Expert Language Models
- 用岭回归闭式解直接组合已训专家,避免额外微调
- 在因果建模中保留97.6%原专家性能,推理任务超102.1%
- 支持动态添加专家且不破坏原有表现,适合多领域应用
大型语言模型在多种专业任务中表现卓越,但训练成本高且缺乏领域专长。虽可通过微调提升专精能力,却易导致过度专业化及多任务冲突。现有方法采用混合专家(MoE)架构融合密集专家,但仍需多任务微调。为此,我们提出动态升级MoE(DUME),通过岭回归闭式解复用已训练的跨领域专家,构建统一模型而无需新增训练。该方法兼具成本效益与可扩展性,支持动态添加专家并保持原始性能。实验表明,在因果语言建模中,DUME可保留高达97.6%的单一领域专家性能;在推理任务中甚至超越原模型,达102.1%。代码已开源。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have achieved remarkable performance on a wide range of specialized tasks, exhibiting strong problem-solving capabilities. However, training these models is prohibitively expensive, and they often lack domain-specific expertise because they rely on general knowledge datasets. Expertise finetuning can address this issue; however, it often leads to overspecialization, and developing a single multi-domain expert remains difficult due to diverging objectives. Furthermore, multitask training is challenging due to interference and catastrophic forgetting. Existing work proposes combining the expertise of dense models within a Mixture of Experts (MoE) architecture, although this approach still requires multitask finetuning. To address these issues, we introduce Dynamic Upcycling MoE (DUME), a novel approach that reuses dense experts trained on different domains to construct a unified MoE model. Our method builds a single multitask model that preserves the capabilities of the original dense experts without requiring additional training. DUME is both cost-efficient and scalable: by leveraging the closed-form solution of ridge regression, it eliminates the need for further optimization and enables experts to be added dynamically while maintaining the model's original performance. We demonstrate that DUME consistently outperforms baseline approaches in both causal language modeling and reasoning settings. Finally, we also show that the DUME model can be fine-tuned to further improve performance. We show that, in the causal language modeling setting, DUME can retain up to 97.6% of a dense expert model specialized in one particular domain, and that it can also surpass it in the reasoning setting, where it can achieve 102.1% of the dense expert performance. Our code is available at: github.com/gensyn-ai/dume.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。