arXiv:2608.05926cs.NIcs.AI2026-08

平衡推理延迟与内存,提升边缘端大模型服务效率

BALANCE: Hybrid Autoregressive-Speculative LLM Inference in Wireless Edge Networks

  • 混合使用自回归与推测解码,动态分配用户任务
  • 在有限资源下,任务吞吐量提升显著,最高达2.3倍
  • 适合资源受限的边缘计算场景,如移动网络服务

边缘推理是下一代移动网络中提供大语言模型(LLM)服务的有前景范式。主流方法包括自回归解码(AD)和推测解码(SD):AD逐个生成输出标记,延迟高;SD通过小语言模型(SLM)生成多个候选标记供LLM验证以加速推理,但增加内存开销。由于存在延迟-内存权衡,单一方法难以高效满足异构用户需求。为此,我们提出面向边缘LLM推理的混合自回归-推测框架BALANCE。在该框架中,边缘服务器同时部署SLM与LLM,将用户分配至AD或SD模式,并并行执行两种模式。为最大化服务用户数,我们建立任务吞吐量最大化问题,联合优化用户调度与计算资源分配,满足用户延迟要求及服务器内存约束。由于问题为NP-hard,我们设计多项式时间算法,将其转化为两个子问题,获得具有常数近似保证的次优解。实验表明,BALANCE持续优于传统AD与SD,在不同负载下任务吞吐量提升显著,最高达2.3倍。

原文摘要 · Abstract (English)

Edge inference is a promising paradigm to provide large language model (LLM) inference services in next-generation mobile networks. LLM inference mainly relies on two approaches: Autoregressive decoding (AD) generates output tokens sequentially, resulting in long latency; Speculative decoding (SD) accelerates inference by using a small language model (SLM) to generate multiple draft tokens for LLM verification, but incurs extra memory costs. Due to this latency-memory tradeoff, neither approach alone can efficiently serve users with heterogeneous demands under limited edge computing resources. To address this challenge, we propose a hybrid autoregressive-speculative inference (BALANCE) framework for edge LLM inference. In BALANCE, an edge server hosts both an SLM and an LLM, assigns each user to AD or SD, and performs the two modes simultaneously. To maximize the number of served users, we formulate a task throughput maximization problem to jointly determine user scheduling and computing resource allocation between AD and SD under user latency requirements and server memory constraints. Since the problem is NP-hard, we develop a polynomial-time algorithm that transforms the original problem into two sub-problems and obtains a sub-optimal solution with a constant approximation guarantee. Experiments demonstrate that BALANCE consistently outperforms conventional AD and SD and significantly improves task throughput.

边缘计算大模型推理混合解码资源调度

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