arXiv:2603.18297cs.LG2026-03被引 1

通过约束专家路径提升稀疏MoE模型效率

Path-Constrained Mixture-of-Experts

  • 将每个token的专家选择看作跨层路径,发现实际使用路径集中于少数
  • 新模型 extit{PathMoE}共享多层路由参数,使路径更集中且更稳定
  • 在0.9B与16B模型上均优于独立路由,无需辅助损失

稀疏混合专家(MoE)架构在每一层独立地将每个标记路由到一组专家。我们从‘专家路径’视角重新审视MoE计算——即一个标记在各层专家选择的序列。这一视角揭示:尽管存在 $N^L$ 种可能路径($N$ 个专家、$L$ 层),但实际中标记仅集中在少数符合语言功能的路径上,多数路径未被探索,造成统计效率低下。为此,我们提出约束有效路径空间以增强这种自然聚集。作为实例,我们引入 extit{PathMoE},其在连续层块间共享路由器参数。分析表明, extit{PathMoE}强化了路径结构:产生更集中的路径簇、更好的跨层一致性,并对路由扰动更具鲁棒性。在0.9B和16B参数的 extit{PathMoE} 模型上,实验显示其在困惑度和下游任务上持续优于独立路由,且无需辅助损失。这些结果确立了专家路径作为MoE架构设计的新维度,与现有独立路由机制互补。

原文摘要 · Abstract (English)

Sparse Mixture-of-Experts (MoE) architectures route each token through a subset of experts at each layer independently. We propose viewing MoE computation through the lens of \emph{expert paths} -- the sequence of expert selections a token makes across all layers. This perspective reveals that, despite $N^L$ possible paths for $N$ experts across $L$ layers, tokens in practice cluster into a small fraction of paths that align with linguistic function, yet the vast majority of paths remain unexplored, representing a statistical inefficiency. This motivates architectures that constrain the effective path space to amplify this natural concentration. As one instantiation, we introduce \pathmoe{}, which shares router parameters across blocks of consecutive layers. Analysis confirms that \pathmoe{} amplifies the emergent path structure: it produces more concentrated path clusters, better cross-layer consistency, and greater robustness to routing perturbations. Experiments on 0.9B and 16B parameter \pathmoe{} models demonstrate consistent improvements on perplexity and downstream tasks over independent routing, while eliminating the need for auxiliary losses. These results establish expert paths as a useful design axis for MoE architectures, complementary to existing work on independent routing mechanisms.

MoE专家路径稀疏架构路由优化

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