Experiments on MoE models with 55M and 108M active parameters show that the Muon optimizer roughly doubles the span of routing biases in the first MoE layer. This indicates that Muon makes the load balancing of the first layer significantly harder.
HOW THIS AFFECTS YOU
●
researcherYou should account for increased routing imbalance when using Muon in Mixture-of-Experts training recipes.