arXiv:2504.08242cs.DCcs.AI2025-04中稿 · IEEE International…被引 38

Jupiter让边缘设备协同运行大模型,速度提升26倍且不损失生成质量。

Jupiter: Fast and Resource-Efficient Collaborative Inference of Generative LLMs on Edge Devices

  • 分阶段优化:预填充用流水线并行,解码用推测性流水线加速
  • 实测端到端延迟降低26.1倍,生成质量与主流方法相当
  • 适合资源受限的边缘部署场景,尤其关注生成式LLM推理效率

生成式大语言模型(LLMs)因其在各类AI任务中的卓越能力而备受关注。传统上部署在云数据中心,如今正逐步向更易访问的边缘平台迁移,以保护敏感用户数据并保障隐私。然而,单个边缘设备有限的计算资源会导致推理延迟过长和内存占用过高。尽管已有研究探索协同边缘计算以突破单设备资源瓶颈,但这些方案仍存在通信开销大、边缘资源利用率低的问题,且仅聚焦于预填充阶段,忽视了生成式LLM中至关重要的自回归解码阶段。为此,我们提出Jupiter,一种快速、可扩展且资源高效的生成式LLM边缘协同推理系统。Jupiter采用灵活的流水线架构作为核心原则,根据预填充与解码阶段的不同特性进行差异化设计:针对预填充阶段,提出新型序列内流水线并行,并设计精细的并行规划策略以最大化资源效率;针对解码阶段,设计基于轮廓的流水线并行解码机制,结合推测性解码,进一步提升推理速度。基于真实实现的广泛评估表明,Jupiter在多种边缘环境配置下显著优于现有最佳方法,端到端延迟最高降低26.1倍,同时保持相近的生成质量。

原文摘要 · Abstract (English)

Generative large language models (LLMs) have garnered significant attention due to their exceptional capabilities in various AI tasks. Traditionally deployed in cloud datacenters, LLMs are now increasingly moving towards more accessible edge platforms to protect sensitive user data and ensure privacy preservation. The limited computational resources of individual edge devices, however, can result in excessively prolonged inference latency and overwhelmed memory usage. While existing research has explored collaborative edge computing to break the resource wall of individual devices, these solutions yet suffer from massive communication overhead and under-utilization of edge resources. Furthermore, they focus exclusively on optimizing the prefill phase, neglecting the crucial autoregressive decoding phase for generative LLMs. To address that, we propose Jupiter, a fast, scalable, and resource-efficient collaborative edge AI system for generative LLM inference. Jupiter introduces a flexible pipelined architecture as a principle and differentiates its system design according to the differentiated characteristics of the prefill and decoding phases. For prefill phase, Jupiter submits a novel intra-sequence pipeline parallelism and develops a meticulous parallelism planning strategy to maximize resource efficiency; For decoding, Jupiter devises an effective outline-based pipeline parallel decoding mechanism combined with speculative decoding, which further magnifies inference acceleration. Extensive evaluation based on realistic implementation demonstrates that Jupiter remarkably outperforms state-of-the-art approaches under various edge environment setups, achieving up to 26.1x end-to-end latency reduction while rendering on-par generation quality.

边缘计算大模型推理流水线并行推测解码

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