用轻量压缩专家替换冗余专家,显著降低推理开销且性能损失小。
Efficiently Editing Mixture-of-Experts Models with Compressed Experts
- 用压缩模块替代冗余激活专家,保留核心参数。
- 在多个任务上恢复超90%全专家性能,活跃参数减少30%以上。
- 适合资源受限场景部署,利于模型进一步扩大规模。
混合专家(MoE)模型通过仅激活部分专家实现大语言模型的高效扩展。通常激活专家数量需权衡计算成本与性能:专家越少越省算力,越多越强。最新研究发现,微调预训练MoE模型时,部分激活专家贡献极小,存在大量冗余参数。本文提出压缩专家概念,即作为完整专家紧凑表示的轻量模块。方法保留关键专家,用压缩专家替换其他辅助激活专家。该策略大幅降低活跃参数量,同时保持相近性能。在Phi-MoE和OLMoE等模型上的实验表明,压缩专家在多种任务中恢复超过90%的全专家性能,活跃参数减少30%以上,推理成本降低20%。该方法支持在资源受限环境下高效部署,并为构建更大模型提供可管理的扩展路径。代码已开源。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) models have become a key approach for scaling large language models efficiently by activating only a subset of experts during training and inference. Typically, the number of activated experts presents a trade-off: fewer experts reduce computational costs, while more experts improve performance. Recent studies reveal that not all activated experts contribute equally to model performance, with some providing minimal utility, particularly when finetuning pretrained MoE models for specialized downstream tasks. The co-existence of significant and redundant parameters in experts provides us an opportunity to reduce the number of activated experts while maintaining model performance. In this work, we propose the concept of compressed experts, lightweight modules that serve as compact representations of full experts. Our approach preserves the most important experts while replacing other auxiliary activated experts with compressed experts. The reduction of active parameters significantly lowers inference costs while achieving comparable performance. Extensive experiments on models including Phi-MoE and OLMoE demonstrate that compressed experts recover over 90% of full expert performance across various tasks while reducing more than 30% active parameters and saving 20% in inference costs. This approach enables efficient deployment of MoE models in resource-constrained settings and facilitates scaling to larger models with manageable overhead. Our code is available at https://github.com/yifei-he/Compressed-Experts.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。