arXiv:2603.27965cs.CV2026-03中稿 · IEEE TMM2026

用多专家融合提升Transformer训练效率,零额外部署成本。

ExFusion: Efficient Transformer Training via Multi-Experts Fusion

  • 初始化时将FFN转为多专家结构,训练中动态融合专家
  • 训练后融合专家权重,仅增极小计算开销
  • 适合追求高效训练的视觉与语言模型研究者

混合专家(MoE)模型通过提升密集架构容量显著改善性能,但直接训练需大量计算资源,并增加参数存储与部署开销。为此,我们提出一种名为ExFusion的新预训练方法,通过多专家融合提升Transformer训练效率。具体而言,在初始化阶段,将Transformer的前馈网络(FFN)升级为多专家配置,每个专家分配一个后续用于参数融合的权重。训练过程中,这些权重使多个专家融合为一个等效于原FFN的统一专家,用于前向计算。结果,ExFusion在训练中引入多专家特性,相比标准密集训练仅增加微小计算成本。训练完成后,利用学习到的权重将多专家整合为单一专家,彻底消除存储与部署的额外开销。在多种计算机视觉与自然语言处理任务上的广泛实验验证了该方法的有效性。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) models substantially improve performance by increasing the capacity of dense architectures. However, directly training MoE models requires considerable computational resources and introduces extra overhead in parameter storage and deployment. Therefore, it is critical to develop an approach that leverages the multi-expert capability of MoE to enhance performance while incurring minimal additional cost. To this end, we propose a novel pre-training approach, termed ExFusion, which improves the efficiency of Transformer training through multi-expert fusion. Specifically, during the initialization phase, ExFusion upcycles the feed-forward network (FFN) of the Transformer into a multi-expert configuration, where each expert is assigned a weight for later parameter fusion. During training, these weights allow multiple experts to be fused into a single unified expert equivalent to the original FFN, which is subsequently used for forward computation. As a result, ExFusion introduces multi-expert characteristics into the training process while incurring only marginal computational cost compared to standard dense training. After training, the learned weights are used to integrate multi-experts into a single unified expert, thereby eliminating additional overhead in storage and deployment. Extensive experiments on a variety of computer vision and natural language processing tasks demonstrate the effectiveness of the proposed method.

Transformer多专家训练效率参数融合

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