基于通用专家覆盖率的稀疏混合专家语言模型剪枝方法 / Generic Expert Coverage for Pruning SparseMixture-of-Experts Language Models
1️⃣ 一句话总结
本文提出了一种名为“通用TB覆盖”的专家剪枝方法,无需下游任务数据,仅用通用文本语料库评估每个专家的效用,并确保从不同语料中保留高效用专家,从而在削减模型规模的同时,显著提升了混合专家模型在多个零样本任务上的准确性和语言建模性能,尤其适用于大比例剪枝场景。
Sparsely activated Mixture-of-Experts (MoE) language models contain substantial structured redundancy among routed experts, but pruning them without downstream calibration data remains challenging. Existing expert-pruning methods typically rely on a single aggregated importance score, which can bias the retained set toward experts favored by dominant calibration patterns. We propose \textbf{Generic TB-Coverage}, a coverage-aware expert pruning method that uses only generic text corpora (WikiText2 and C4) for calibration. Instead of collapsing expert utility into one score, our method profiles per-expert utility separately on each corpus and enforces a fixed-budget coverage rule that preserves high-utility experts from each corpus before constructing the final pruning mask. Across Qwen1.5-MoE-A2.7B and DeepSeek-MoE-16B-Base at 25\%, 50\%, and 75\% retention budgets, our method improves average accuracy on six common zero-shot benchmarks over random pruning, REAP, and ExpertSparsity, while also reducing perplexity degradation on WikiText2 and C4. The gains are largest under aggressive pruning (25\% and 50\% retain), suggesting that preserving cross-corpus expert coverage is an effective generic-data prior for MoE pruning. Our improvements hold with fixed pruning budgets and no downstream calibration data.
基于通用专家覆盖率的稀疏混合专家语言模型剪枝方法 / Generic Expert Coverage for Pruning SparseMixture-of-Experts Language Models
本文提出了一种名为“通用TB覆盖”的专家剪枝方法,无需下游任务数据,仅用通用文本语料库评估每个专家的效用,并确保从不同语料中保留高效用专家,从而在削减模型规模的同时,显著提升了混合专家模型在多个零样本任务上的准确性和语言建模性能,尤其适用于大比例剪枝场景。
源自 arXiv: 2607.01710