将专家混合机制引入视觉Transformer,提升模型容量与效率
ViMoE: An Empirical Study of Designing Vision Mixture-of-Experts
- 在ViT中引入MoE结构,通过共享专家增强路由稳定性
- 发现专家层配置影响性能,不当设计导致信息获取失效
- 可分析专家路由行为,帮助精简冗余层提升效率
Mixture-of-Experts(MoE)模型体现分而治之思想,是提升模型容量的有前途方法,在多个领域展现出优异可扩展性。本文将MoE结构集成到经典视觉Transformer(ViT)中,命名为ViMoE,通过图像分类和语义分割任务全面研究将MoE应用于视觉的潜力。然而我们发现性能对MoE层配置敏感,需精心设计才能获得最佳效果。根本原因在于不当的MoE层导致路由不可靠,阻碍专家获取有效信息。为此,我们引入共享专家以学习并捕获通用知识,有效构建稳定的ViMoE。此外,我们展示了如何分析专家路由行为,揭示哪些MoE层能专门处理特定信息,哪些不能,为保留关键层、去除冗余提供指导,从而在不牺牲精度的前提下提升ViMoE效率。本工作旨在为视觉MoE模型设计提供新洞察,并为未来研究提供有价值的实证指导。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) models embody the divide-and-conquer concept and are a promising approach for increasing model capacity, demonstrating excellent scalability across multiple domains. In this paper, we integrate the MoE structure into the classic Vision Transformer (ViT), naming it ViMoE, and explore the potential of applying MoE to vision through a comprehensive study on image classification and semantic segmentation. However, we observe that the performance is sensitive to the configuration of MoE layers, making it challenging to obtain optimal results without careful design. The underlying cause is that inappropriate MoE layers lead to unreliable routing and hinder experts from effectively acquiring helpful information. To address this, we introduce a shared expert to learn and capture common knowledge, serving as an effective way to construct stable ViMoE. Furthermore, we demonstrate how to analyze expert routing behavior, revealing which MoE layers are capable of specializing in handling specific information and which are not. This provides guidance for retaining the critical layers while removing redundancies, thereby advancing ViMoE to be more efficient without sacrificing accuracy. We aspire for this work to offer new insights into the design of vision MoE models and provide valuable empirical guidance for future research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。