arXiv:2503.02354cs.DCcs.AI2025-03被引 5

通过专家依赖关系优化,实现小内存下多专家模型高效推理。

CoServe: Efficient Collaboration-of-Experts (CoE) Model Inference with Limited Memory

  • 利用专家依赖性减少不必要的模型切换,提升推理效率。
  • 在真实制造场景中,吞吐量比现有系统高4.5至12倍。
  • 适合资源受限的工业级智能推理任务,如电路板质检。

大型语言模型如GPT-4资源消耗大,而小型专业化专家模型在特定任务上表现更优。协作专家(CoE)方法融合多个专家模型,提升生成结果精度,适用于电路板质量检测等高精度场景。然而,部署CoE服务面临内存瓶颈,因需调用大量专家,频繁跨内存与存储层级切换导致显著性能开销。本文提出CoServe,一种在异构CPU/GPU、内存受限环境下的高效CoE模型推理系统。CoServe通过利用专家依赖性,减少冗余切换,引入依赖感知请求调度器与专家管理机制,并配备离线分析器自动优化不同设备上的资源分配。在真实智能制造负载下,相比最先进系统,吞吐量提升4.5×至12×。

原文摘要 · Abstract (English)

Large language models like GPT-4 are resource-intensive, but recent advancements suggest that smaller, specialized experts can outperform the monolithic models on specific tasks. The Collaboration-of-Experts (CoE) approach integrates multiple expert models, improving the accuracy of generated results and offering great potential for precision-critical applications, such as automatic circuit board quality inspection. However, deploying CoE serving systems presents challenges to memory capacity due to the large number of experts required, which can lead to significant performance overhead from frequent expert switching across different memory and storage tiers. We propose CoServe, an efficient CoE model serving system on heterogeneous CPU and GPU with limited memory. CoServe reduces unnecessary expert switching by leveraging expert dependency, a key property of CoE inference. CoServe introduces a dependency-aware request scheduler and dependency-aware expert management for efficient inference. It also introduces an offline profiler to automatically find optimal resource allocation on various processors and devices. In real-world intelligent manufacturing workloads, CoServe achieves 4.5$\times$ to 12$\times$ higher throughput compared to state-of-the-art systems.

模型推理专家协作高效调度工业AI

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