arXiv:2503.02495cs.LGcs.AI2025-03被引 1

提出新型专家融合架构,提升模型效率与性能。

Union of Experts: Adapting Hierarchical Routing to Equivalently Decomposed Transformer

  • 将Transformer分解为功能等价的专家组,实现分层路由分配
  • 语言建模中困惑度降低2.38,仅需76%计算量
  • 适用于注意力模块,适合追求高效推理的开发者

Mixture-of-Experts (MoE) 在保持计算效率的同时提升模型性能,适用于大规模应用。传统MoE架构存在专家间协调不佳、易过拟合等问题,且未有效扩展至注意力模块,限制了进一步效率提升。为此,本文提出联合专家(Union-of-Experts, UoE),将Transformer模型等效分解为专家组,并采用分层路由机制将输入子空间分配给专业专家。UoE在设计上实现四大创新:(1) 通过非MoE模型的功能等价分解构建专家组;(2) 提出融合块级数据选择与专家选择的分层路由策略;(3) 将MoE设计扩展至注意力模块;(4) 设计硬件优化的并行化方案,利用批量矩阵乘法高效计算专家。实验表明,UoE在图像与自然语言多个任务中超越全注意力模型、先进MoE及高效Transformer。在语言建模任务中,相比最优MoE方法,困惑度平均降低2.38,仅需其76%的浮点运算量(FLOPs);在长程任务基准(Long Range Arena)中,平均得分至少高于所有对比模型0.68%,仅需最优MoE方法50%的FLOPs;在图像分类中,平均准确率提升1.75%,计算量相当。代码已开源:https://github.com/YujiaoYang-work/UoE。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) enhances model performance while maintaining computational efficiency, making it well-suited for large-scale applications. Conventional mixture-of-experts (MoE) architectures suffer from suboptimal coordination dynamics, where isolated expert operations expose the model to overfitting risks. Moreover, they have not been effectively extended to attention blocks, which limits further efficiency improvements. To tackle these issues, we propose Union-of-Experts (UoE), which decomposes the transformer model into an equivalent group of experts and applies a hierarchical routing mechanism to allocate input subspaces to specialized experts. Our approach advances MoE design with four key innovations: (1) Constructing expert groups by partitioning non-MoE models into functionally equivalent specialists (2) Developing a hierarchical routing paradigm that integrates patch-wise data selection and expert selection strategies. (3) Extending the MoE design to attention blocks. (4) Proposing a hardware-optimized parallelization scheme that exploits batched matrix multiplications for efficient expert computation. The experiments demonstrate that our UoE model surpasses Full Attention, state-of-the-art MoEs and efficient transformers in several tasks across image and natural language domains. In language modeling tasks, UoE achieves an average reduction of 2.38 in perplexity compared to the best-performing MoE method with only 76% of its FLOPs. In the Long Range Arena benchmark, it demonstrates an average score at least 0.68% higher than all comparison models, with only 50% of the FLOPs of the best MoE method. In image classification, it yields an average accuracy improvement of 1.75% over the best model while maintaining comparable FLOPs. The source codes are available at https://github.com/YujiaoYang-work/UoE.

MoETransformer高效模型分层路由

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