通过分解专家权重,让混合专家模型更省参数。
DeRS: Towards Extremely Efficient Upcycled Mixture-of-Experts Models
- 将专家拆分为共享基权重和轻量增量权重
- 压缩后参数减少90%以上,性能几乎不变
- 适合需要高效部署的模型优化场景
将预训练稠密模型中的前馈网络层转换为混合专家(MoE)层,可提升多种任务表现。但现有方法因引入多个专家仍存在显著参数效率问题。本文提出一种新范式DeRS(分解、替换、合成),基于对上翻式MoE专家冗余机制的观察:将专家分解为一个共享基权重和多个特定增量权重,并以轻量形式表示这些增量。该方法适用于两种场景:1)推理阶段的DeRS压缩,通过稀疏化或量化压缩原始上翻式MoE模型;2)训练阶段的DeRS上翻,用轻量稀疏或低秩矩阵高效将稠密模型转为MoE模型。在三个不同任务上的大量实验表明,所提方法可在保持训练与压缩性能的同时,实现极高的参数效率。
原文摘要 · Abstract (English)
Upcycled Mixture-of-Experts (MoE) models have shown great potential in various tasks by converting the original Feed-Forward Network (FFN) layers in pre-trained dense models into MoE layers. However, these models still suffer from significant parameter inefficiency due to the introduction of multiple experts. In this work, we propose a novel DeRS (Decompose, Replace, and Synthesis) paradigm to overcome this shortcoming, which is motivated by our observations about the unique redundancy mechanisms of upcycled MoE experts. Specifically, DeRS decomposes the experts into one expert-shared base weight and multiple expert-specific delta weights, and subsequently represents these delta weights in lightweight forms. Our proposed DeRS paradigm can be applied to enhance parameter efficiency in two different scenarios, including: 1) DeRS Compression for inference stage, using sparsification or quantization to compress vanilla upcycled MoE models; and 2) DeRS Upcycling for training stage, employing lightweight sparse or low-rank matrixes to efficiently upcycle dense models into MoE models. Extensive experiments across three different tasks show that the proposed methods can achieve extreme parameter efficiency while maintaining the performance for both training and compression of upcycled MoE models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。