Skip to yearly menu bar Skip to main content


Poster

Break the Sequential Dependency of LLM Inference Using Lookahead Decoding

Yichao Fu · Peter Bailis · Ion Stoica · Hao Zhang


Abstract:

Autoregressive decoding of large language models (LLMs) is memory bandwidth bounded, resulting in high latency and significant wastes of the parallel processing power of modern accelerators. Existing methods for accelerating LLM decoding often require a draft model (e.g., speculative decoding), which is nontrivial to obtain and unable to generalize. In this paper, we introduce Lookahead decoding, an exact, parallel decoding algorithm that accelerates LLM decoding without needing auxiliary models or data stores. It allows trading per-step log(FLOPs) to reduce the number of total decoding steps, is more parallelizable on single or multiple modern accelerators, and is compatible with concurrent memory-efficient attention (e.g., FlashAttention). Our implementation of Lookahead decoding can speed up autoregressive decoding by up to 1.8x on MT-bench and 4x with strong scaling on multiple GPUs in code completion tasks. Our code is available at https://anonymous.4open.science/r/LookaheadDecoding-E8C1

Live content is unavailable. Log in and register to view live content