Least-Loaded Expert Parallelism: Load Balancing An Imbalanced Mixture-of-Experts
Mixture-of-Experts (MoE) models are typically pre-trained with explicit load-balancing constraints to ensure statistically balanced expert routing. Despite this, we observe that even well-trained MoE models exhibit significantly imbalanced routing. This behavior is arguably natural—and even desirabl…