菜单

关于 🐙 GitHub
arXiv 提交日期: 2026-05-27
📄 Abstract - Pruning and Distilling Mixture-of-Experts into Dense Language Models

Mixture-of-Experts (MoE) is now the dominant architecture for frontier language models, yet it requires all expert parameters to be loaded in memory, making it less preferable for memory-constrained deployment. Existing compression methods reduce the number of experts but the output remains an MoE model with the same fundamental limitation. We present the first systematic framework for converting a trained MoE into a standard fully dense architecture: experts are scored, selected, and grouped, then concatenated into a dense FFN and refined by knowledge distillation from the MoE teacher. We evaluate 7 scoring, 5 grouping, and 2 magnitude scaling methods across a range of selected expert counts on Qwen3-30B-A3B, yielding 350 configurations. We find that the choice of scoring method is the most impactful, with our novel diversity-aware scoring consistently outperforming prior methods on Qwen3-30B-A3B, DeepSeek-V2-Lite, and GPT-OSS-20B. Under a controlled comparison at matched parameter count, MoE-to-dense outperforms dense-to-dense pruning by +6.3 pp in average downstream accuracy after ~4B-token distillation at 1.6x faster training wall-clock speed.

顶级标签: model training model evaluation machine learning
详细标签: mixture of experts knowledge distillation pruning dense model conversion scoring methods 或 搜索:

将混合专家模型剪枝与蒸馏为密集语言模型 / Pruning and Distilling Mixture-of-Experts into Dense Language Models


1️⃣ 一句话总结

本文提出了一种将训练好的混合专家(MoE)模型系统性地转换为标准密集模型的方法,通过专家评分、选择、分组、拼接成密集前馈网络,再结合知识蒸馏来提升性能;实验表明,该方法在同等参数量下比传统的密集模型剪枝平均准确率高6.3个百分点,且训练速度快1.6倍。

源自 arXiv: 2605.28207