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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Mathematical Reasoning | GSM8K | Accuracy83 | 1424 | |
| Creative Writing | WildBench | WildBench Score40.2 | 49 | |
| Multiple-Choice QA | Multiple-Choice Suite | MC Avg0.681 | 49 | |
| Multiple-choice Question Answering | MC (test) | MC Avg72.3 | 46 | |
| Code Generation | Coding Eval+ LiveCode (test) | Eval+ Score84.5 | 32 | |
| Code Generation | EvalPlus (test) | Eval+73.4 | 23 | |
| Mathematical Reasoning | MATH 500 | MATH-500 Score69.2 | 23 | |
| Creative Writing | WildBench (test) | WildBench Score60.4 | 15 | |
| Math | GSM8K MATH-500 (test) | GSM8K Accuracy89.5 | 15 | |
| Expert Pruning Efficiency | MoE Models | Calibration Time (h)0.22 | 6 |