Skip to main content

Related fields

Wireless edge LLM inference can switch decoding modes per user

BALANCE jointly schedules users and edge compute while assigning autoregressive or speculative decoding under latency and memory constraints.

One decoding strategy does not fit every edge request

Autoregressive decoding generates large-language-model output one token at a time. It has a simple memory profile but can impose long latency. Speculative decoding uses a smaller model to draft several tokens for a larger model to verify, reducing sequential delay at the cost of additional model memory. BALANCE treats this latency–memory tension as a wireless edge resource-allocation problem rather than choosing one decoding method for every request.

The proposed edge server hosts both a small language model and a large language model. Each user is assigned either autoregressive or speculative decoding, and the server runs both modes concurrently. The optimization then decides which users to serve and how to divide computing resources between the two modes while satisfying per-user latency requirements and an overall memory constraint. This coupling matters: assigning more users to the faster-looking mode can reduce capacity if the required model state exhausts memory.

Hybrid decoding becomes a scheduling variable

The task-throughput maximization problem is NP-hard. The authors split it into two subproblems and derive a polynomial-time algorithm with a constant approximation guarantee. That guarantee is an important distinction from a purely empirical scheduler: the method is not claimed to solve the combinatorial problem exactly, but its solution quality has a stated bound under the model.

Experiments in the public abstract consistently outperform conventional all-autoregressive and all-speculative configurations and improve the number of served tasks. The abstract does not expose the numerical gains, model sizes, radio assumptions, or sensitivity to changing token lengths. It also does not establish energy use or service quality beyond the stated latency and memory constraints. Those details would determine whether the hybrid policy transfers to a particular edge platform.

The systems insight is durable even before deployment measurements are available. Edge LLM service is not only a question of compressing a model or provisioning more accelerators. The decoding algorithm itself changes the resource footprint. A scheduler that can choose the decoding mode per request gains a new control variable for matching heterogeneous users to finite compute and memory.

Research notes

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

  • Authors: Guanqiao Qu, Shuo Chen, Qian Chen, Kin K. Leung, Xianhao Chen
  • Public record: arXiv
  • What is established: BALANCE jointly selects users, assigns autoregressive or speculative decoding, and allocates compute under latency and memory constraints using a polynomial-time constant-approximation algorithm.
  • Read with care: The abstract reports qualitative throughput gains but does not provide the numerical margin, energy cost, model sizes, or deployment-specific radio conditions.