Dynamic K Mechanism for Accelerating Speculative Decoding in vLLM

With the rapid growth in parameter scale of large language models (LLM), the computational cost during their inference phase has become a major bottleneck for deployment. Although speculative decoding serves as a lightweight acceleration technique—generating candidate tokens via a draft model and verifying them through the main model—it effectively reduces latency. However, mainstream implementations typically adopt a fixed generation step-size K, failing to adapt to diverse input scenarios and thus limiting the upper bound of decoding performance. To address this issue, this paper proposes a dynamic K-based speculative decoding method that adaptively adjusts the generation step-size according to draft quality, and implements it for the first time in the vLLM inference framework. Experimental results demonstrate that this strategy significantly enhances the throughput ceiling of speculative decoding without compromising output quality, providing a viable system optimization path for efficient general-purpose inference engines.

Authors

Institutions

Publication Details

Published
2025-06-20
DOI
https://doi.org/10.1109/icecai66283.2025.11170671
Primary Topic
Advanced Wireless Communication Techniques
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Dynamic K Mechanism for Accelerating Speculative Decoding in vLLM

Wei Gao, Jiansheng Liang, Lin Han, Z Gao et al.
Advanced Wireless Communication Techniques
article

Dynamic K Mechanism for Accelerating Speculative Decoding in vLLM

Wei Gao, Jiansheng Liang, Lin Han, Z Gao, Jingming Xie, Chaowei Kong
article en

Abstract

With the rapid growth in parameter scale of large language models (LLM), the computational cost during their inference phase has become a major bottleneck for deployment. Although speculative decoding serves as a lightweight acceleration technique—generating candidate tokens via a draft model and verifying them through the main model—it effectively reduces latency. However, mainstream implementations typically adopt a fixed generation step-size K, failing to adapt to diverse input scenarios and thus limiting the upper bound of decoding performance. To address this issue, this paper proposes a dynamic K-based speculative decoding method that adaptively adjusts the generation step-size according to draft quality, and implements it for the first time in the vLLM inference framework. Experimental results demonstrate that this strategy significantly enhances the throughput ceiling of speculative decoding without compromising output quality, providing a viable system optimization path for efficient general-purpose inference engines.

Zhongyuan University of Technology (CN), Zhengzhou University (CN), National Supercomputing Center in Wuxi (CN)
Openalex Percentile: Top 74%
Advanced Wireless Communication Techniques
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.