arXiv:2605.08738cs.LGcs.AI2026-05被引 2

通过剪枝与蒸馏压缩MoE大模型,提升训练效率与性能。

SlimQwen: Exploring the Pruning and Distillation in Large MoE Model Pre-training

论文配图:SlimQwen: Exploring the Pruning and Distillation in Large MoE Model Pre-training
图 1 · 摘自论文原文
  • 先预训练再剪枝,比从零训练效果更好。
  • 不同剪枝方式经持续预训练后表现相近,合并专家可提升下游性能。
  • 结合语言建模损失的蒸馏更有效,尤其在知识密集任务中。

结构化剪枝和知识蒸馏是压缩大型语言模型的常用方法,但在大规模预训练场景下,特别是针对近期的混合专家(MoE)模型,其应用方式仍不明确。本文系统研究了大规模预训练中MoE模型的压缩策略,聚焦三个核心问题:剪枝能否提供优于从零开始训练的初始化?专家压缩方式如何影响最终模型性能?何种训练策略最有效?研究发现:在深度、宽度和专家压缩层面,对预训练后的MoE模型进行剪枝,均能在相同训练预算下优于从零开始训练;不同一次性专家压缩方法在大规模持续预训练后收敛至相似性能;基于此,我们提出一种简单部分保留的专家合并策略,显著提升多数基准测试表现;结合知识蒸馏与语言建模损失优于仅使用蒸馏,尤其在知识密集型任务中;进一步提出多标记预测蒸馏(MTP),带来稳定收益;此外,在相同训练令牌数下,渐进式剪枝策略优于一次性压缩,表明渐进架构调整能获得更优优化轨迹。最终,我们将Qwen3-Next-80A3B压缩为23A2B模型,仍保持竞争力。这些结果为大规模高效MoE压缩提供了实用指导。

原文摘要 · Abstract (English)

Structured pruning and knowledge distillation (KD) are typical techniques for compressing large language models, but it remains unclear how they should be applied at pretraining scale, especially to recent mixture-of-experts (MoE) models. In this work, we systematically study MoE compression in large-scale pretraining, focusing on three key questions: whether pruning provides a better initialization than training from scratch, how expert compression choices affect the final model after continued training, and which training strategy is most effective. We have the following findings: First, across depth, width, and expert compression, pruning a pretrained MoE consistently outperforms training the target architecture from scratch under the same training budget. Second, different one-shot expert compression methods converge to similar final performance after large-scale continual pretraining. Motivated by this, we introduce a simple partial-preservation expert merging strategy that improves downstream performance across most benchmarks. Third, combining KD with the language modeling loss outperforms KD alone, particularly on knowledge-intensive tasks. We further propose multi-token prediction (MTP) distillation, which yields consistent gains. Finally, given the same training tokens, progressive pruning schedules outperform one-shot compression, suggesting that gradual architecture transitions lead to better optimization trajectories. Putting it all together, we compress Qwen3-Next-80A3B to a 23A2B model that retains competitive performance. These results offer practical guidance for efficient MoE compression at scale.

MoE模型模型压缩知识蒸馏剪枝

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。