让视觉专家模型更懂语义,路由更精准高效
Guiding the Experts: Semantic Priors for Efficient and Focused MoE Routing
- 用语义引导增强软专家路由,让专家激活对齐图像前景区域
- 在ImageNet-1K上提升分类准确率,且路由结果更可解释
- 仅需微小改动,适配现有软专家架构,易集成
混合专家(MoE)模型是高效扩展视觉架构的有前途方向。其中,软MoE通过连续调度权重将每个令牌分配给所有专家,提升了训练稳定性。然而,现有设计忽略了这些权重中隐含的语义结构,导致专家路由不理想。本文发现,软MoE中的调度权重天然呈现分割样模式,但未显式对齐语义区域。为此,我们提出一种前景引导增强策略:引入空间感知的辅助损失,促使专家激活对齐语义前景;同时融合轻量级LayerScale机制,改善跳跃连接中的信息流动与优化稳定性。该方法仅需微小架构调整,可无缝集成至主流软MoE框架。在ImageNet-1K及多个小规模分类基准上的实验表明,性能持续提升,并揭示出更具可解释性的专家路由机制。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) models have emerged as a promising direction for scaling vision architectures efficiently. Among them, Soft MoE improves training stability by assigning each token to all experts via continuous dispatch weights. However, current designs overlook the semantic structure which is implicitly encoded in these weights, resulting in suboptimal expert routing. In this paper, we discover that dispatch weights in Soft MoE inherently exhibit segmentation-like patterns but are not explicitly aligned with semantic regions. Motivated by this observation, we propose a foreground-guided enhancement strategy. Specifically, we introduce a spatially aware auxiliary loss that encourages expert activation to align with semantic foreground regions. To further reinforce this supervision, we integrate a lightweight LayerScale mechanism that improves information flow and stabilizes optimization in skip connections. Our method necessitates only minor architectural adjustments and can be seamlessly integrated into prevailing Soft MoE frameworks. Comprehensive experiments on ImageNet-1K and multiple smaller-scale classification benchmarks not only showcase consistent performance enhancements but also reveal more interpretable expert routing mechanisms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。