mlsanityData to 14 Oct 2026

Three Phases of Expert Routing: How Load Balance Evolves During Mixture-of-Experts Training

arXiv 2604.04230 ↗First posted 5 Apr 2026cs.LG cs.AI cs.MAcited by 4

On the map: Mixture-of-experts → Expert routing strategies

Abstract (arXiv)

We model Mixture-of-Experts (MoE) token routing as a congestion game with a single effective parameter, the congestion coefficient gamma_eff, that quantifies the balance-quality tradeoff. Tracking gamma_eff across training checkpoints of two open-source MoE models, OLMoE-1B-7B (20 checkpoints, with dense sampling in the surge region) and OpenMoE-8B (6 checkpoints), reveals a three-phase trajectory: a surge phase where the router learns to balance load (gamma_eff: 14 to 36-39, peaking in the step 30K-40K region), a stabilization phase where experts specialize under steady balance (B_0: 2.4 to 2.3, steps 100K-400K), and a relaxation phase where the router trades balance for quality as experts differentiate (gamma_eff: 27 to 9, steps 400K-1.2M). This non-monotone trajectory, invisible to post-hoc analysis of converged models, reveals that early MoE training prioritizes balance while late training prioritizes quality. The theoretical framework is honest about its limits: the single-type equilibrium reduces to temperature-scaled softmax (held-out L1: MFG = 0.199 vs. softmax = 0.200). The game is not a better predictor; it reveals what the temperature means and, critically, how that temperature evolves. We complement the dynamics with an effective congestion decomposition, a multi-type extension that improves load prediction via token clustering on all 16 layers (mean: 30%), scope diagnostics (K/M, epsilon_l), and robustness verification across four independent quality estimators (r >= 0.89). All confidence intervals are from bootstrap resampling over 50 independent text batches.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 9 of its 20 references that are in the map. First posted on arXiv 5 Apr 2026.

Builds on

9 of 20 references.

Closest ancestors · 3

Same direction, newest first

  1. 2024Auxiliary-Loss-Free Load Balancing Strategy for Mixture-of-Experts28 Aug
  2. 2022Mixture-of-Experts with Expert Choice Routing18 Feb
  3. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan

Other references · 6

  1. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  2. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,631
  3. 2021BASE Layers: Simplifying Training of Large, Sparse Modelscited by 320
  4. 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
  5. 2024OpenMoE: An Early Effort on Open Mixture-of-Experts Language Modelscited by 187
  6. 2024Toward Inference-optimal Mixture-of-Expert Large Language Modelscited by 19

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Routing-Free Mixture-of-Experts
  2. 2026When Load-Balancing Goes Too Far: Expert Pruning in Over-Dispersed Mixture-of-Experts Models
  3. 2026Router Prior Bias: Preserving Base Routing Structure in MoE Post-Training
  4. 2026Share First, Route What Remains: A Unified Framework for Token-Adaptive MoE Computation
  5. 2026Hierarchical Mixture-of-Experts with Two-Stage Optimization
  6. 2026Beyond Routing: Decoupling Expert Dispatch and Aggregation in Sparse Mixture-of-Experts
  7. 2026ReLibra: Routing-Replay-Guided Load Balancing for MoE Training in Reinforcement Learning
  8. 2026Distributionally Robust Mixture-of-Experts Training