Search papers, labs, and topics across Lattice.
This paper introduces the BALANCE framework, which integrates hybrid autoregressive and speculative decoding methods for efficient large language model (LLM) inference in wireless edge networks. By simultaneously utilizing both an autoregressive decoder and a small language model, BALANCE addresses the latency-memory tradeoff inherent in traditional approaches, allowing for better resource allocation and user scheduling. Experimental results show that BALANCE significantly enhances task throughput compared to conventional autoregressive and speculative decoding methods, making it a promising solution for serving heterogeneous user demands in edge computing environments.
Hybridizing autoregressive and speculative decoding, BALANCE boosts task throughput in edge LLM inference while managing latency and memory constraints.
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.
