When Does Sparse Routing Help? The Geometry of Expert Separation in Mixture of Experts
Huy Nguyen, Dung Le, Alessandro Rinaldo, Nhat Ho
Abstract
Sparse mixtures of experts combine multiple prediction models while evaluating only a small subset for each input. Although this architecture reduces computation, how expert separation and routing sparsity jointly govern expert recovery remains an open problem. In this paper, we address this question for location Gaussian mixtures of experts with a fixed number of experts by establishing how expert separation and spatial arrangement determine recovery rates. For dense models, our results reveal that when experts form well-separated clusters, the separation geometry is governed by the size of the largest cluster; without a prescribed cluster structure, this geometry is instead driven by the total number of experts. For sparse models, the separation geometry on retained routing regions is determined by the number of active experts, while the contribution of omitted regions is controlled by their total probability. Consequently, activating fewer experts reduces the separation complexity in these bounds. Furthermore, Top-1 routing yields a particularly striking result: experts can be recovered at a rate independent of the separation quantity. Our technical approach combines polynomial interpolation and Gaussian reproducing-kernel Hilbert space (RKHS) test functions with quantitative restriction inequalities that extend pointwise comparisons to expert functions.
Create a lesson
Related papers
Nearly Linear-Time Follow-the-Perturbed-Leader for Weakly Observable Graph Bandits
Jongyeong Lee, Hyundam Yoo
How Risky Is Your Data-Driven Policy?
Omar Besbes, Alon Rieger
Two-Layer Random ReLU Networks Can Separate Disjoint Classes
Yichen Liu, Jianbin Yang
Point interpolation and distributional universality in flow-based generative models
Antonio Álvarez-López, Jingpu Cheng, Liyan Zhang et al.
Nonparametric GARCH: An Iterated Neural Estimator
Ruizhi Deng, Guohao Shen, Ngai Hang Chan
Error Accumulation in Second-Order Sequential Machine Unlearning
Chen Shang, Haolin Zou, Binghe Zhu et al.