Generalization and Scaling Laws for Mixture-of-Experts Transformers
On the map: Mixture-of-experts → Expert routing strategies
Abstract (arXiv)
We develop a theory of generalization and scaling for Mixture-of-Experts (MoE) Transformers that cleanly separates active per-input capacity from routing combinatorics. By conditioning on fixed routing patterns and union-bounding across them, we derive a sup-norm covering-number bound whose metric entropy scales with the active parameter budget and incurs a MoE-specific routing overhead. Combined with a standard ERM analysis for squared loss, this yields a generalization bound under a d-dimensional manifold data model and Cβ targets, showing that approximation and estimation trade off as in dense networks once active parameters are accounted for appropriately. We further prove a constructive approximation theorem for MoE architectures, showing that, under the approximation construction, error can decrease either by scaling active capacity or by increasing the number of experts, depending on the dominant bottleneck. From these results we derive neural scaling laws for model size, data size, and compute-optimal tradeoffs. Overall, our results provide a transparent statistical reference point for reasoning about MoE scaling, clarifying which behaviors are certified by worst-case theory and which must arise from data-dependent routing structure or optimization dynamics.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 32 of its 43 references that are in the map. First posted on arXiv 10 Apr 2026.
Builds on
32 of 43 references.
Closest ancestors · 2
Same direction, newest first
- 2022Mixture-of-Experts with Expert Choice Routing18 Feb
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan
Other references · 30
- 2017Attention Is All You Needcited by 59,932
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2016Layer Normalizationcited by 5,579
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2020On Layer Normalization in the Transformer Architecturecited by 906
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 890
- 2021Scaling Vision with Sparse Mixture of Expertscited by 740
- 2017Spectrally-normalized margin bounds for neural networkscited by 681
- 2016Error bounds for approximations with deep ReLU networkscited by 567
18 more
- 2017Deep Learning Scaling is Predictable, Empiricallycited by 556
- 2017Nonparametric regression using deep neural networks with ReLU activation functioncited by 431
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 425
- 2019Are Transformers universal approximators of sequence-to-sequence functions?cited by 404
- 2022DeepSpeed-MoE: Advancing Mixture-of-Experts Inference and Training to Power Next-Generation AI Scalecited by 400
- 2021BASE Layers: Simplifying Training of Large, Sparse Modelscited by 315
- 2015Norm-Based Capacity Control in Neural Networkscited by 299
- 2017Size-Independent Sample Complexity of Neural Networkscited by 289
- 2018Estimating the intrinsic dimension of datasets by a minimal neighborhood informationcited by 246
- 2022Tutel: Adaptive Mixture-of-Experts at Scalecited by 199
- 2017Optimal approximation of piecewise smooth functions using deep ReLU neural networkscited by 176
- 2024Scaling Laws for Fine-Grained Mixture of Expertscited by 166
- 2019Nonparametric Regression on Low-Dimensional Manifolds using Deep ReLU Networks : Function Approximation and Statistical Recoverycited by 73
- 2025Towards Greater Leverage: Scaling Laws for Efficient Mixture-of-Experts Language Modelscited by 44
- 2024Understanding Scaling Laws with Statistical and Approximation Theory for Transformer Neural Networks on Intrinsically Low-dimensional Datacited by 30
- 2025Joint MoE Scaling Laws: Mixture of Experts Can Be Memory Efficientcited by 29
- 2025Scaling Laws for Upcycling Mixture-of-Experts Language Modelscited by 12
- 2025Scaling Fine-Grained MoE Beyond 50B Parameters: Empirical Evaluation and Practical Insightscited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026MoRE: Scaling mixture of experts with hardware-aware low-rank routing
- 2026Towards a Statistical Understanding of Mixture-of-Experts
- 2026Optimal Expert-Attention Allocation in Mixture-of-Experts: A Scalable Law for Dynamic Model Design
- 2026DECO: Sparse Mixture-of-Experts with Dense-Comparable Performance on End-Side Devices
- 2025Load Balancing Mixture of Experts with Similarity Preserving Routers
- 2025On the Expressive Power of Mixture-of-Experts for Structured Complex Tasks
- 2025Optimal Scaling Laws for Efficiency Gains in a Theoretical Transformer-Augmented Sectional MoE Framework
- 2026Synergistic Intra- and Cross-Layer Regularization Losses for MoE Expert Specialization