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

AIMER: Calibration-Free Task-Agnostic MoE Expert Pruning

About

Mixture-of-Experts (MoE) language models increase parameter capacity without proportional per-token computation, yet deployment still requires storing the full expert pool, making expert pruning important for reducing memory and serving overhead. Existing task-agnostic expert-pruning methods are typically calibration-dependent: they estimate expert importance from routing or activation statistics on a calibration set, making pruning decisions sensitive to calibration-data variation while introducing substantial preprocessing cost. We propose AIMER (\textbf{A}bsolute mean over root mean square \textbf{IM}portance for \textbf{E}xpert \textbf{R}anking), a simple calibration-free criterion that identifies more distinct experts by capturing the concentration pattern of expert weights, making it well suited for task-agnostic expert pruning. Across 7B to 47B MoE language models with distinct architectures and 16 diverse benchmarks, AIMER consistently delivers stronger capability balance across diverse tasks than existing calibration-free methods. Surprisingly, AIMER also achieves better balance than strong calibration-based expert-pruning baselines calibrated on the widely used task-agnostic C4 corpus, while requiring only 0.22--2.06 seconds to score all experts.

Zongfang Liu, Guangyi Chen, Shengkun Tang, Yifan Shen, Huan Wang, Xin Yuan• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K
Accuracy83
1424
Creative WritingWildBench
WildBench Score40.2
49
Multiple-Choice QAMultiple-Choice Suite
MC Avg0.681
49
Multiple-choice Question AnsweringMC (test)
MC Avg72.3
46
Code GenerationCoding Eval+ LiveCode (test)
Eval+ Score84.5
32
Code GenerationEvalPlus (test)
Eval+73.4
23
Mathematical ReasoningMATH 500
MATH-500 Score69.2
23
Creative WritingWildBench (test)
WildBench Score60.4
15
MathGSM8K MATH-500 (test)
GSM8K Accuracy89.5
15
Expert Pruning EfficiencyMoE Models
Calibration Time (h)0.22
6
Showing 10 of 10 rows

Other info

Follow for update