Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Zongfang Liu, Jinghui Zhang, Zijian Ma, Guangyi Chen, Xin Yuan• 2026

Related benchmarks

TaskDatasetResultRank
MathGSM8K
Accuracy0.937
248
CodingEval+
Eval+ Score85
47
MathematicsMATH-500 (test)
Score (%)79.6
41
CodingEval+ (test)
Score86.8
32
CodingLiveCode (test)
Score34.6
32
General PerformanceOverall Aggregate (test)
Average Score70.7
32
Multiple-ChoiceMultiple Choice MC Avg (test)
Average Accuracy72.7
32
WritingWildBench (test)
Score0.57
32
CodingLiveCode
LiveCode Score34.6
31
MathMATH 500
MATH 500 Score67.6
25
Showing 10 of 12 rows

Other info

Follow for update