arXiv:2505.06481cs.LGcs.DC2025-05ICML被引 5

共享相似专家降低内存占用,实现多MoE模型高效共用单卡

QoS-Efficient Serving of Multiple Mixture-of-Expert LLMs Using Partial Runtime Reconfiguration

  • 基于专家相似性合并,减少多模型内存开销
  • 单卡上吞吐接近单模型,首令牌延迟增加可忽略
  • 适合资源受限环境下部署多个微调后的MoE模型

混合专家(MoE)大语言模型部署面临高内存需求挑战,尤其在多租户环境中,共享资源使传统虚拟化效果受限。本文提出一种服务系统,通过相似性驱动的专家合并,降低多细调MoE-LLM的整体内存占用。为保障输出质量,引入运行时部分重配置机制,在处理不同模型请求时动态替换非专家层。实验在配备单块NVIDIA A100 GPU(80GB)的服务器上进行,使用Mixtral-8x7B模型,相比NVIDIA多实例GPU(MIG),平均周转时间缩短85%。对Google Switch Transformer Base-8模型最多四类变体的测试表明,该方法在保持输出质量方面优于其他模型合并基线,具备良好可扩展性和鲁棒性。

原文摘要 · Abstract (English)

The deployment of mixture-of-experts (MoE) large language models (LLMs) presents significant challenges due to their high memory demands. These challenges become even more pronounced in multi-tenant environments, where shared resources must accommodate multiple models, limiting the effectiveness of conventional virtualization techniques. This paper addresses the problem of efficiently serving multiple fine-tuned MoE-LLMs on a single-GPU. We propose a serving system that employs \textit{similarity-based expert consolidation} to reduce the overall memory footprint by sharing similar experts across models. To ensure output quality, we introduce \textit{runtime partial reconfiguration}, dynamically replacing non-expert layers when processing requests from different models. As a result, our approach achieves a competitive output quality while maintaining throughput comparable to serving a single model while incurring a negligible increase in time-to-first-token (TTFT). Experiments on a server with a single NVIDIA A100 GPU (80GB) using Mixtral-8x7B models demonstrate an 85\% average reduction in turnaround time compared to NVIDIA's multi-instance GPU (MIG). Furthermore, experiments on Google's Switch Transformer Base-8 model with up to four variants demonstrate the scalability and resilience of our approach in maintaining output quality compared to other model merging baselines, highlighting its effectiveness.

MoE模型内存优化多任务部署单卡推理

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