arXiv:2411.02829cs.DCcs.LG2024-11被引 49

通过云边协同降低大模型推理延迟,提升边缘部署效率。

CE-CoLLM: Efficient and Adaptive Large Language Models Through Cloud-Edge Collaboration

  • 设计云边协同框架,优化上下文数据传输与计算分配。
  • 推理延迟降低13.81%,云端计算负载减少超84.53%。
  • 支持低延迟独立推理与高精度协同推理,适应多样边缘环境。

大型语言模型(LLMs)具备类人预测能力,但将其高效、自适应地部署于边缘端仍具挑战。本文提出一种新型云边协同框架CE-CoLLM以应对该问题。首先,识别出云边间上下文数据传输是主要性能瓶颈,导致显著通信开销并主导整体推理延迟。其次,引入时延感知的早期退出机制与高效的云端上下文管理技术,有效降低通信开销并保持推理准确率。第三,设计两种自适应推理模式:(1) 低延迟独立边缘推理模式,在网络不稳定时仍可实现可靠边缘侧独立推理;(2) 高精度云边协同推理模式,根据需求动态调用云端资源以提升预测准确性。在多个基准数据集上的实验表明,相比传统云端部署,CE-CoLLM将整体推理时间减少最高达13.81%,并将超过84.53%的计算任务从云端卸载至边缘,且不损失预测准确率。代码已开源于GitHub。

原文摘要 · Abstract (English)

Large Language Models (LLMs) exhibit remarkable human-like predictive capabilities. However, it is challenging to deploy LLMs to provide efficient and adaptive inference services at the edge. This paper proposes a novel Cloud-Edge Collaboration framework for LLMs (CE-CoLLM) to tackle these challenges. First, we identify the transmission of LLM contextual data between the cloud and edge as a key performance bottleneck, which introduces substantial communication overhead that dominates overall inference latency and makes naïve cloud-edge collaboration for LLMs inefficient. Second, we introduce a suite of novel techniques, including a latency-aware early exit mechanism and efficient cloud context management, into CE-CoLLM, which collectively reduce communication overhead and preserve LLM inference accuracy. Third, we design two adaptive inference modes to accommodate diverse edge environments: (1) a low-latency standalone edge inference mode that enables reliable edge-side independent LLM inference even under unstable network conditions, and (2) a high-accuracy cloud-edge collaborative inference mode that adaptively leverages cloud resources to enhance prediction accuracy. Extensive experiments on multiple benchmark datasets demonstrate that CE-CoLLM reduces overall inference time by up to 13.81% and offloads over 84.53% of the computational workload from the cloud to the edge, compared to conventional cloud-based LLM deployment, without sacrificing prediction accuracy. The code is provided on GitHub at https://github.com/mlsysx/CE-CoLLM.

云边协同大模型推理边缘计算LLM优化

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