arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2603.02217cs.LGcs.AI

无需重新训练是否足够?高效MoE压缩中路由校准的必要性

Is Retraining-Free Enough? The Necessity of Router Calibration for Efficient MoE Compression

Department of Electrical and Computer Engineering, Seoul National University, Seoul, South Korea(电气与计算机工程系,首尔国立大学,首尔,韩国);Interdisciplinary Program in Artificial Intelligence, Seoul National University, Seoul, South Korea(人工智能跨学科项目,首尔国立大学,首尔,韩国)

Sieun Hyeon, Jaeyoung Do

AI总结:

本文提出路由知识蒸馏方法,通过轻量级路由校准实现无需重新训练的MoE压缩,有效解决路由-专家不匹配问题,提升细粒度MoE的性能表现。

摘要:

混合专家(MoE)模型能够高效地扩展容量,但其庞大的参数足迹在部署时造成了内存瓶颈。我们将无需重新训练的MoE压缩分为三种范式——专家剪枝、专家编辑和专家合并,并表明持续压缩后的退化问题主要源于一个被忽视的因素:当专家被更改但路由器未被调整时的路由-专家不匹配。我们认为有效的无需重新训练压缩应避免更新专家参数,同时允许轻量级的路由校准。为此,我们提出了路由知识蒸馏(Router KD),仅更新极小比例的参数(路由器)通过在未标记的校准数据上蒸馏原始模型的下一个令牌分布。在所有三种范式中代表性方法的实验中,展示了持续的性能恢复,由于其更复杂的路由决策边界,在细粒度MoE(许多小专家)中获得了显著更大的收益。

英文摘要:

Mixture-of-Experts (MoE) models scale capacity efficiently, but their massive parameter footprint creates a deployment-time memory bottleneck. We organize retraining-free MoE compression into three paradigms - Expert Pruning, Expert Editing, and Expert Merging - and show that persistent post-compression degradation largely stems from a neglected factor: router-expert mismatch when experts are changed but the router is left untouched. We argue that effective retraining-free compression should avoid updating expert parameters while allowing lightweight router calibration. To this end, we propose Router Knowledge Distillation (Router KD), which updates only a tiny fraction of parameters (the router) by distilling the original model's next-token distribution on unlabeled calibration data. Experiments across representative methods in all three paradigms demonstrate consistent performance recovery, with substantially larger gains in fine-grained MoEs (many small experts) than in coarse-grained MoEs due to their more complex routing decision boundaries.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑