English

DSDE: Dynamic Speculative Decoding with KLD Stability for Real-World Serving

Distributed, Parallel, and Cluster Computing 2025-10-31 v3 Artificial Intelligence Information Theory math.IT

Abstract

Speculative decoding accelerates large language model inference, but its reliance on a fixed speculation length is suboptimal in large-batch serving environments with diverse requests. This paper explores a new direction for dynamic adaptation by investigating a novel class of post-hoc, diagnostic signals. We propose Dynamic Speculative Decoding Engine (DSDE), a training-free framework built on two primary components: (1) a predictive signal based on the variance of the Kullback-Leibler (KLD) divergence, which diagnoses the generation's regional stability, and (2) an adaptive speculation length cap to mitigate the straggler problem in per-sequence decoding. Experiments demonstrate the potential of using KLD-based stability signals for dynamic adaptation. An algorithm guided by these signals achieves end-to-end latency competitive with leading baselines and exhibits superior robustness across diverse workloads. This robustness is particularly valuable in challenging low-acceptance-rate regimes, where the proposed signal maintains its diagnostic utility. Collectively, these findings validate post-hoc signals as a valuable component for building more robust and intelligent LLM inference systems, and highlight a promising direction for future research on dynamic speculation length adaptation.

Keywords

Cite

@article{arxiv.2509.01083,
  title  = {DSDE: Dynamic Speculative Decoding with KLD Stability for Real-World Serving},
  author = {Mingyu Yang and Jae-Young Choi and Kihyo Moon and Minsung Jang and Eunjoo Jeon},
  journal= {arXiv preprint arXiv:2509.01083},
  year   = {2025}
}

Comments

Accepted for presentation at the IEEE BigData 2025 Workshop (Special Session on Intelligent Data Mining). This v2 updates formatting and adds IEEE copyright notice

R2 v1 2026-07-01T05:14:33.595Z