arXiv:2508.05257cs.LG2025-08被引 7

提出MoBE方法压缩专家混合模型,大幅降参同时几乎不损失精度。

MoBE: Mixture-of-Basis-Experts for Compressing MoE-based LLMs

  • 将专家权重分解为专属矩阵与共享基矩阵的线性组合。
  • 在671B和1T级大模型上压缩24%-30%,精度仅降1%-2%。
  • 适合需要部署超大规模语言模型的场景,如云端推理服务。

专家混合(MoE)架构已成为扩展大型语言模型的主流范式。尽管其具备强大性能与计算效率,但如DeepSeek-V3-0324和Kimi-K2-Instruct等大型MoE模型在部署时面临严重的内存压力。现有压缩方法常导致显著精度下降(相对降低7-14%),即使在中等压缩率下亦然。本文提出一种新型混合基专家(Mixture-of-Basis-Experts, MoBE)方法,在实现模型压缩的同时保持极小精度损失。具体而言,每个专家的up/gate矩阵通过秩分解表示为W = AB,其中矩阵A专属于各专家,而较大矩阵B被重参数化为同一MoE层内所有专家共享的基矩阵{Bi}的线性组合。该分解通过最小化相对于原始权重矩阵的重构误差进行学习。实验表明,MoBE相比先前方法显著降低精度损失:例如,在Qwen3-235B-A22B-2507、DeepSeek-V3-0324(671B)和Kimi-K2-Instruct(1T)上分别实现24%-30%参数量压缩,精度相对下降仅1%-2%。

原文摘要 · Abstract (English)

The Mixture-of-Experts (MoE) architecture has become a predominant paradigm for scaling large language models (LLMs). Despite offering strong performance and computational efficiency, large MoE-based LLMs like DeepSeek-V3-0324 and Kimi-K2-Instruct present serious challenges due to substantial memory requirements in deployment. While recent works have explored MoE compression to address this issue, existing methods often suffer from considerable accuracy drops (e.g., 7-14% relatively) even at modest compression rates. This paper introduces a novel Mixture-of-Basis-Experts (MoBE) method that achieves model compression while incurring minimal accuracy drops. Specifically, each up/gate matrix in an expert is decomposed via a rank decomposition as W = AB, where matrix A is unique to each expert. The relatively larger matrix B is further re-parameterized as a linear combination of basis matrices {Bi} shared across all experts within a given MoE layer. The factorization is learned by minimizing the reconstruction error relative to the original weight matrices. Experiments demonstrate that MoBE achieves notably lower accuracy drops compared to prior works. For instance, MoBE can reduce the parameter counts of Qwen3-235B-A22B-2507, DeepSeek-V3-0324 (671B) and Kimi-K2-Instruct (1T) by 24%-30% with only 1%-2% accuracy drop (about 2% drops when measured relatively).

模型压缩MoE大模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。