论文精选

一种针对混合专家语言模型的专家剪枝方法 HOPE

Higher-order pruning of experts in mixture-of-experts language models

精选理由

研究团队提出的新方法 HOPE,通过考虑专家间的协同作用来优化混合专家语言模型的剪枝过程,在 122B 参数模型上测试,HOPE 在 50% 剪枝率下比 REAP 等方法性能更好,在代理任务上提升达 6.1%,证明了其有效性。

研究提出 HOPE 方法,通过考虑专家间的协同作用来优化混合专家语言模型的剪枝过程。在 122B 参数模型上测试,HOPE 在 50% 剪枝率下比 REAP 等方法性能更好,在代理任务上提升达 6.1%,证明了其有效性。

原文 · arXiv cs.AI

Higher-order pruning of experts in mixture-of-experts language models

Mixture-of-Experts (MoE) language models suffer from large parameter counts, which create a significant memory bottleneck. Expert pruning is the most direct approach for reducing this parameter count, yet existing methods make pruning decisions for each expert independently, and assume experts' contributions are purely additive. In reality, expert usage in MoEs is inherently cooperative. We derive HOPE (Higher-Order Pruning of Experts), a second-order pruning objective which provably minimizes an upper bound on the error resulting from pruning. We show that REAP (a state-of-the-art first-order pruning method) is a special case of HOPE where interaction terms are ignored. Across three frontier MoE models (up to 122B parameters), two distinct calibration sets, and multiple benchmarks (including math, instruction following, coding, and an agentic suite), we demonstrate that HOPE produces better pruning decisions than existing methods, and its advantage is most pronounced at high pruning rates and on challenging agentic workloads. At 50% pruning, HOPE outperforms all baselines and achieves an average rank of 1.58 out of 5 methods (versus 2.42 for the next-best method, REAP), with gains of up to +6.1% on agentic coding. Over all conditions, HOPE again achieves the best average rank and surpasses every other method in the majority of head-to-head comparisons. By preserving cooperative expert structure that first-order methods ignore, HOPE enables aggressive compression with minimal degradation, particularly on complex tasks where diverse expert combinations are invoked over long sequences.