μ-Parametrization for Mixture of Experts
On the map: Mixture-of-experts → MoE theory and gating functions
Abstract (arXiv)
Recent years have seen a growing interest and adoption of LLMs, with Mixture-of-Experts (MoE) emerging as a leading architecture in extremely large models. Currently, the largest open-source models reach over 1T parameters. At such scales, hyperparameter tuning becomes prohibitively expensive. Precisely for this reason, the μTransfer is becoming a key technique. It allows for seamless transfer of optimal hyperparameters across model scales, resulting in a huge reduction in tuning costs. However, existing work has primarily focused on dense LLMs, leaving MoE architectures unexplored. In this work, we derive a μ-Parameterization for MoE, providing theoretical guarantees for feature learning across model widths. Our experiments demonstrate that the optimal learning rate reliably transfers across model sizes, establishing a foundation for efficient hyperparameter tuning in large-scale MoE models.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in MoE theory and gating functions, one of 12 directions in Mixture-of-experts. Builds on 14 of its 18 references that are in the map. First posted on arXiv 13 Aug 2025.
Builds on
14 of 18 references.
Closest ancestors · 1
Same direction, newest first
Other references · 13
- 2017Attention Is All You Needcited by 59,932
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2022Mixture-of-Experts with Expert Choice Routingcited by 611
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 391
- 2022Unified Scaling Laws for Routed Language Modelscited by 249
- 2020Feature Learning in Infinite-Width Neural Networkscited by 182
- 2024Scaling Laws for Fine-Grained Mixture of Expertscited by 166
- 2023A Spectral Condition for Feature Learningcited by 101
- 2023Tensor Programs VI: Feature Learning in Infinite-Depth Neural Networkscited by 94
1 more
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Complete-muE: Optimal Hyperparameter Transfer and Scaling for MoE Modelscited by 1
- 2026How to Scale Mixture-of-Experts: From muP to the Maximally Scale-Stable Parameterizationcited by 0
- 2026Let's Scale Step by Step: Compute-Efficient Hyperparameter Transfer for Large-Scale Mixture-of-Expertscited by 0
- 2026Let's Scale Step by Step: Compute-Efficient Hyperparameter Transfer for Large-Scale Mixture-of-Experts20 Aug 2026
- 2026Complete-muE: Optimal Hyperparameter Transfer and Scaling for MoE Models22 May 2026
- 2026How to Scale Mixture-of-Experts: From muP to the Maximally Scale-Stable Parameterization13 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Hyperparameter Transfer with Mixture-of-Expert Layers
- 2026Hyperparameter Scaling Laws Across MoE Sparsity
- 2025Symphony-MoE: Harmonizing Disparate Pre-trained Models into a Coherent Mixture-of-Experts
- 2025Mixture-of-Experts Can Surpass Dense LLMs Under Strictly Equal Resource
- 2023Pushing Mixture of Experts to the Limit: Extremely Parameter Efficient MoE for Instruction Tuning
- 2026ACE: Adapter Consolidation across Experts for Parameter-Efficient Fine-Tuning of MoE LLMs
- 2024Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
- 2026MoE-DisCo:Low Economy Cost Training Mixture-of-Experts Models