English

Path-Constrained Mixture-of-Experts

Machine Learning 2026-04-07 v2

Abstract

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 NLN^L possible paths for NN experts across LL 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.

Keywords

Cite

@article{arxiv.2603.18297,
  title  = {Path-Constrained Mixture-of-Experts},
  author = {Zijin Gu and Tatiana Likhomanenko and Vimal Thilak and Jason Ramapuram and Navdeep Jaitly},
  journal= {arXiv preprint arXiv:2603.18297},
  year   = {2026}
}

Comments

Under review

R2 v1 2026-07-01T11:27:11.222Z