When Load-Balancing Goes Too Far: Expert Pruning in Over-Dispersed Mixture-of-Experts Models
On the map: Mixture-of-experts → Expert routing strategies
Abstract (arXiv)
Expert pruning reduces the memory and serving cost of Mixture-of-Experts (MoE) models by removing low-importance experts identified by the router, assuming router probabilities provide a reliable importance signal. We observe that this assumption breaks down under over-dispersed routing, a regime associated with aggressive load-balancing during training, in which tokens are distributed nearly uniformly across experts and importance signals collapse. In this regime, perplexity does not predict downstream task accuracy: on gpt-oss-20B, the lowest-perplexity pruning configuration yields the worst mathematical reasoning, while the highest-perplexity configuration preserves it. This does not occur under standard routing (e.g., Mixtral-8x7B-Instruct), where perplexity and accuracy degrade together. Pruning under over-dispersed routing also exposes a capability trade-off in which no single scoring metric dominates: activation-aware scoring preserves mathematical reasoning but severely degrades knowledge-intensive science (an 18-point gap on GPQA), whereas frequency-based scoring exhibits the reverse. We propose Minimax Expert Score Allocation (MESA), a domain-aware method that iteratively boosts importance scores for experts serving whichever domain is currently worst-affected, minimizing worst-case domain degradation rather than average accuracy. At 25% expert pruning MESA achieves the smallest worst-case degradation across domains, outperforming activation-aware baselines on 7 of 11 benchmarks at a correspondingly reduced memory footprint, and it generalizes to gpt-oss-120B, Gemma-4-26B-A4B, and OLMoE-1B-7B. Our results indicate that over-dispersed routing is a qualitatively distinct pruning regime in which standard assumptions fail, and that recognizing it is a prerequisite for principled expert pruning of load-balanced MoE models.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Expert routing strategies, one of 12 directions in Mixture-of-experts. Builds on 20 of its 22 references that are in the map. First posted on arXiv 3 Sep 2026.
Builds on
20 of 22 references.
Closest ancestors · 3
Same direction, newest first
- 2025Load Balancing Mixture of Experts with Similarity Preserving Routers16 Jun
- 2023Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy2 Oct
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity11 Jan
Other references · 17
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,164
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
- 2023Instruction-Following Evaluation for Large Language Modelscited by 1,228
- 2022ST-MoE: Designing Stable and Transferable Sparse Expert Modelscited by 399
- 2024Not All Experts are Equal: Efficient Expert Pruning and Skipping for Mixture-of-Experts Large Language Modelscited by 146
5 more
- 2025REAP the Experts: Why Pruning Prevails for One-Shot MoE compressioncited by 44
- 2024Retraining-Free Merging of Sparse MoE via Hierarchical Clusteringcited by 35
- 2024Towards Efficient Mixture of Experts: A Holistic Study of Compression Techniquescited by 24
- 2026EvoESAP: Non-Uniform Expert Pruning for Sparse MoEcited by 9
- 2024DRPruning: Efficient Large Language Model Pruning through Distributionally Robust Optimizationcited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026MoRA: MoE Pruning via Router Bias Learning and Expert Approximation
- 2026Router Prior Bias: Preserving Base Routing Structure in MoE Post-Training
- 2026It Takes a MAESTRO To Prune Bad Experts
- 2026FreeBalance: Pre-Routing Online Moe Load Balancing via Residual Workload Prediction
- 2026Least-Loaded Expert Parallelism: Load Balancing An Imbalanced Mixture-of-Experts
- 2026A Replicate-and-Quantize Strategy for Plug-and-Play Load Balancing of Sparse Mixture-of-Experts LLMs
- 2025ERMoE: Eigen-Reparameterized Mixture-of-Experts for Stable Routing and Interpretable Specialization
- 2026Three Phases of Expert Routing: How Load Balance Evolves During Mixture-of-Experts Training