arXiv:2412.14219cs.LGcs.AI2024-12综述被引 63

系统梳理MoE模型推理优化技术,助你高效部署大模型。

A Survey on Inference Optimization Techniques for Mixture of Experts Models

  • 从模型、系统到硬件分层归纳优化方法
  • 涵盖压缩、路由、调度等关键技术提升效率
  • 适合研究者与工程师快速掌握前沿优化方案

大规模混合专家(MoE)模型通过条件计算显著提升了模型容量与计算效率。然而,在实际推理中面临计算资源消耗大、延迟高和能效低的挑战。本文全面调研了贯穿整个系统栈的MoE模型推理优化技术。首先建立分类框架,将优化方法分为模型级、系统级和硬件级三类。模型级包括高效专家设计、注意力机制改进、剪枝、量化、知识蒸馏等压缩技术,以及动态路由与专家合并等算法优化。系统级聚焦分布式计算、负载均衡与高效调度算法,支持可扩展部署。硬件级则探讨专用优化与软硬件协同设计,以最大化吞吐量与能效。本综述不仅提供现有解决方案的系统性梳理,还指出现有挑战与未来研究方向。为持续更新与共享最新进展,作者建立了开源仓库:https://github.com/MoE-Inf/awesome-moe-inference/

原文摘要 · Abstract (English)

The emergence of large-scale Mixture of Experts (MoE) models represents a significant advancement in artificial intelligence, offering enhanced model capacity and computational efficiency through conditional computation. However, deploying and running inference on these models presents significant challenges in computational resources, latency, and energy efficiency. This comprehensive survey analyzes optimization techniques for MoE models across the entire system stack. We first establish a taxonomical framework that categorizes optimization approaches into model-level, system-level, and hardware-level optimizations. At the model level, we examine architectural innovations including efficient expert design, attention mechanisms, various compression techniques such as pruning, quantization, and knowledge distillation, as well as algorithm improvement including dynamic routing strategies and expert merging methods. At the system level, we investigate distributed computing approaches, load balancing mechanisms, and efficient scheduling algorithms that enable scalable deployment. Furthermore, we delve into hardware-specific optimizations and co-design strategies that maximize throughput and energy efficiency. This survey provides both a structured overview of existing solutions and identifies key challenges and promising research directions in MoE inference optimization. To facilitate ongoing updates and the sharing of cutting-edge advances in MoE inference optimization research, we have established a repository accessible at https://github.com/MoE-Inf/awesome-moe-inference/.

MoE模型推理优化系统架构大模型

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