动态聚类+结构压缩,三重难题一体解。
Breaking the MoE LLM Trilemma: Dynamic Expert Clustering with Structured Compression
- 按参数与激活相似性动态聚类专家,稳定负载分配。
- 每组专家压缩至原大小1/5,吞吐提升10%-20%,负载方差降3倍以上。
- 适合追求高效部署的MoE模型研究者与工程师。
Mixture-of-Experts(MoE)大语言模型面临负载不均、参数冗余和通信开销三重困境。本文提出一种基于动态专家聚类与结构压缩的统一框架,通过融合参数与激活相似性的在线聚类机制,周期性重组专家以稳定使用率。首次利用路由器的语义嵌入能力,在训练中动态重构模型架构,实现显著效率提升。在每组聚类内,将专家权重分解为共享基矩阵与极低秩残差适配器,每组参数减少约80%且保留专业化能力。采用两级分层路由:先分配至聚类,再选择具体专家,大幅降低路由搜索空间与全对全通信量。结合异构精度存储(共享基用FP16,残差因子用INT4)与动态卸载非活跃聚类,峰值内存降至接近稠密模型水平。在GLUE与WikiText-103上评估,模型性能媲美标准MoE,总参数减少约80%,吞吐提升10%-20%,专家负载方差下降超3倍。结果表明,结构重组织是实现可扩展、高效、内存友好的MoE LLM的可行路径。代码已开源。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) Large Language Models (LLMs) face a trilemma of load imbalance, parameter redundancy, and communication overhead. We introduce a unified framework based on dynamic expert clustering and structured compression to address these issues cohesively. Our method employs an online clustering procedure that periodically regroups experts using a fused metric of parameter and activation similarity, which stabilizes expert utilization. To our knowledge, this is one of the first frameworks to leverage the semantic embedding capability of the router to dynamically reconfigure the model's architecture during training for substantial efficiency gains. Within each cluster, we decompose expert weights into a shared base matrix and extremely low-rank residual adapters, achieving up to fivefold parameter reduction per group while preserving specialization. This structure enables a two-stage hierarchical routing strategy: tokens are first assigned to a cluster, then to specific experts within it, drastically reducing the routing search space and the volume of all-to-all communication. Furthermore, a heterogeneous precision scheme, which stores shared bases in FP16 and residual factors in INT4, coupled with dynamic offloading of inactive clusters, reduces peak memory consumption to levels comparable to dense models. Evaluated on GLUE and WikiText-103, our framework matches the quality of standard MoE models while reducing total parameters by approximately 80%, improving throughput by 10% to 20%, and lowering expert load variance by a factor of over three. Our work demonstrates that structural reorganization is a principled path toward scalable, efficient, and memory-effective MoE LLMs. Code is available at https://github.com/szdtzpj/Breaking_the_moe_trilemma
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。