How to Score Experts for One-Shot MoE Expert Pruning: A Unified Formulation and Selection Principle
About
Mixture-of-Experts (MoE) language models reduce per-token computation through sparse expert activation, yet deployment still requires storing the full expert pool, making one-shot expert pruning a practical approach for reducing memory usage. Although effective, existing criteria are largely heuristic, and no single criterion is universally optimal. Thus, establishing a principle for selecting pruning criteria suited to different deployment objectives remains an important yet largely underexplored problem in one-shot expert pruning. To this end, we introduce a unified formulation for one-shot MoE expert pruning organized around three factors: routing frequency, gate weighting, and activation strength. The formulation yields a criteria selection principle: task-agnostic pruning should favor routed-token-averaged, gate-free activation-based criteria, whereas task-specific pruning can benefit from retaining routing-frequency and gate-weight information. Beyond this principle, the formulation also provides a systematic view of existing heuristic criteria and gives rise to two new task-agnostic criteria, Mean Activation Norm (MAN) and Mean Squared Activation Norm (MSAN). Across four representative MoE models and 16 diverse benchmarks, MAN and MSAN are consistently strong in the task-agnostic setting, obtain the top-two average ranks, and improve average performance by up to 8.8 points over the strongest baseline.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Math | GSM8K | Accuracy0.937 | 248 | |
| Coding | Eval+ | Eval+ Score85 | 47 | |
| Mathematics | MATH-500 (test) | Score (%)79.6 | 41 | |
| Coding | Eval+ (test) | Score86.8 | 32 | |
| Coding | LiveCode (test) | Score34.6 | 32 | |
| General Performance | Overall Aggregate (test) | Average Score70.7 | 32 | |
| Multiple-Choice | Multiple Choice MC Avg (test) | Average Accuracy72.7 | 32 | |
| Writing | WildBench (test) | Score0.57 | 32 | |
| Coding | LiveCode | LiveCode Score34.6 | 31 | |
| Math | MATH 500 | MATH 500 Score67.6 | 25 |