论文

结构保持式专家剪枝方法 PEEP 与 DCEP 发布

Experts Exhibit Structured Redundancy: Structure-Preserving Expert Pruning for Mixture-of-Experts Large Language Models

精选理由

一篇讲怎么给 MoE 模型砍专家省显存的论文,在 DeepSeek-V2-Lite 和 Qwen3-30B-A3B 上砍掉一半专家还能跑赢基线,做部署的可以看看。

arXiv 论文发现 MoE 层的专家输出存在结构化冗余:稳定秩和熵有效秩远低于名义专家数。作者将专家剪枝重构为结构保持的子集选择问题,提出基于贪心策略的两种方法 PEEP 和 DCEP。在 DeepSeek-V2-Lite 和 Qwen3-30B-A3B 上,专家保留率 75% 和 50% 时,两种方法在多数基准上超过现有最优基线。

原文 · arXiv cs.AI

Experts Exhibit Structured Redundancy: Structure-Preserving Expert Pruning for Mixture-of-Experts Large Language Models

Sparse MoE mechanism significantly improves the trade-off between model capacity and inference efficiency. However, practical deployment still requires storing all experts, resulting in substantial memory overhead. We show that expert outputs in MoE layers exhibit structured redundancy. Across different models and task categories, the stable rank and entropy effective rank of expert outputs are consistently far below the nominal number of experts, while expert output distributions also exhibit fewer effective modes. Motivated by these observations, we reformulate expert pruning as a structure-preserving expert subset selection problem and efficiently optimize it via a greedy strategy, rather than only independently scoring and ranking individual experts or exhaustively searching over expert combinations. We propose two expert pruning methods under the unified structure-preserving framework: Projection Energy Expert Pruning (PEEP), which preserves the functional subspace by maximizing weighted projection energy, and Distributional Coverage Expert Pruning (DCEP), which preserves the distributional geometry by minimizing weighted coverage error. Both methods incorporate router-derived weights as priors, enabling them to jointly preserve the global structural relationships among experts while accounting for the local importance of individual experts. Experiments on DeepSeek-V2-Lite and Qwen3-30B-A3B demonstrate that, at the expert retention ratios of 75% and 50%, our methods outperform the state-of-the-art baselines on the majority of benchmarks and consistently achieve the best average performance across both general tasks and mathematical reasoning tasks.