arXiv:2511.10971cs.CV2025-11中稿 · CVPR被引 7

提出新型稀疏专家模型,让路由更稳定、专家分工更清晰。

ERMoE: Eigen-Reparameterized Mixture-of-Experts for Stable Routing and Interpretable Specialization

  • 用特征与专家基的余弦相似度替代传统门控,实现内容感知路由
  • 在ImageNet和跨模态任务上达到顶尖准确率,专家负载更均衡
  • 无需平衡损失,专家分工可解释,适合追求高效可解释性的研究者

Mixture-of-Experts(MoE)架构通过稀疏激活专家来扩展模型容量,但面临两大挑战:路由器输出与专家内部结构不匹配导致路由不稳定和专家利用率低,负载不均引发慢速瓶颈。现有方法如辅助负载均衡损失虽能缓解负载差异,却常削弱专家专属性能并影响下游表现。为此,我们提出ERMoE,一种基于学习正交特征基重参数化专家的稀疏MoE Transformer,将学习到的门控分数替换为“特征基得分”,即输入特征与专家基之间的余弦相似度。该内容感知路由机制将样本分配直接关联至专家表示空间,稳定了专家利用率并促进可解释的专属性能,且无需显式平衡损失,避免其引入干扰梯度。实验表明,ERMoE在ImageNet分类和跨模态图像文本检索(如COCO、Flickr30K)任务上达到最先进性能,同时自然产生更平坦的专家负载分布。此外,其3D MRI变体ERMoE-ba使脑龄预测准确率提升超7%,并生成解剖学可解释的专家专属性。因此,ERMoE为稀疏专家模型引入了新范式,直接解决路由不稳定性问题,实现可扩展、可解释的专属性能提升。

原文摘要 · Abstract (English)

Mixture-of-Experts (MoE) architectures expand model capacity by sparsely activating experts but face two core challenges: misalignment between router logits and each expert's internal structure leads to unstable routing and expert underutilization, and load imbalances create straggler bottlenecks. Standard solutions, such as auxiliary load-balancing losses, can reduce load disparities but often weaken expert specialization and hurt downstream performance. To address these issues, we propose ERMoE, a sparse MoE transformer that reparameterizes each expert in a learned orthonormal eigenbasis and replaces learned gating logits with an "Eigenbasis Score", defined as the cosine similarity between input features and an expert's basis. This content-aware routing ties token assignments directly to experts' representation spaces, stabilizing utilization and promoting interpretable specialization without sacrificing sparsity. Crucially, ERMoE removes the need for explicit balancing losses and avoids the interfering gradients they introduce. We show that ERMoE achieves state-of-the-art accuracy on ImageNet classification and cross-modal image-text retrieval benchmarks (e.g., COCO, Flickr30K), while naturally producing flatter expert load distributions. Moreover, a 3D MRI variant (ERMoE-ba) improves brain age prediction accuracy by more than 7\% and yields anatomically interpretable expert specializations. ERMoE thus introduces a new architectural principle for sparse expert models that directly addresses routing instabilities and enables improved performance with scalable, interpretable specialization.

MoE专家模型路由优化可解释性

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