Scaling Laws for Speculative Decoding

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Yan, Siyuan, Zhu, Mo, Jiang, Guo-qing, Wang, Jianfei, Chen, Jiaxing, Zhang, Wentai, Liao, Xiang, Cui, Xiao, Zhang, Chen, Song, Zhuoran, Zhu, Ran
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915285870575616
author Yan, Siyuan
Zhu, Mo
Jiang, Guo-qing
Wang, Jianfei
Chen, Jiaxing
Zhang, Wentai
Liao, Xiang
Cui, Xiao
Zhang, Chen
Song, Zhuoran
Zhu, Ran
author_facet Yan, Siyuan
Zhu, Mo
Jiang, Guo-qing
Wang, Jianfei
Chen, Jiaxing
Zhang, Wentai
Liao, Xiang
Cui, Xiao
Zhang, Chen
Song, Zhuoran
Zhu, Ran
contents The escalating demand for efficient decoding in large language models (LLMs) is particularly critical for reasoning-intensive architectures like OpenAI-o3 and DeepSeek-R1, which depend on extended chain-of-thought reasoning. This study investigates speculative decoding techniques through dense LLM architectures to establish foundational insights for accelerating reasoning tasks. While speculative decoding methods leveraging parallel draft-verification cycles have emerged as promising acceleration techniques, the scaling laws governing decoding efficiency remain under-explored compared to conventional backbone LLMs developed through Pretraining->SFT->RLHF training paradigms. In this work, we discover Log-linear Scaling Laws (Theorem 1.1, 1.2 and 1.3) governing draft model acceptance rate (or decoding speed) across three dimensions: pretraining token volume, draft model capacity, and decoding batch size. Building on these laws, we achieve Scylla, which coordinates multi-dimensional scaling for popular LLMs (Llama2/3, Qwen2.5). Empirical validation shows Scylla achieves 1.5-2.2 higher acceptance rate than EAGLE2 and 0.3 higher than EAGLE3 at temperature T = 0, with peak performance gains on summarization and QA tasks (Figure 2). Industrial inference engine deployments demonstrate 2X decoding throughput improvements over EAGLE2 (Table 5), validating the transformative potential of systematic scaling for efficient LLM inference. Code will be released later.
format Preprint
id arxiv_https___arxiv_org_abs_2505_07858
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Scaling Laws for Speculative Decoding
Yan, Siyuan
Zhu, Mo
Jiang, Guo-qing
Wang, Jianfei
Chen, Jiaxing
Zhang, Wentai
Liao, Xiang
Cui, Xiao
Zhang, Chen
Song, Zhuoran
Zhu, Ran
Computation and Language
Artificial Intelligence
The escalating demand for efficient decoding in large language models (LLMs) is particularly critical for reasoning-intensive architectures like OpenAI-o3 and DeepSeek-R1, which depend on extended chain-of-thought reasoning. This study investigates speculative decoding techniques through dense LLM architectures to establish foundational insights for accelerating reasoning tasks. While speculative decoding methods leveraging parallel draft-verification cycles have emerged as promising acceleration techniques, the scaling laws governing decoding efficiency remain under-explored compared to conventional backbone LLMs developed through Pretraining->SFT->RLHF training paradigms. In this work, we discover Log-linear Scaling Laws (Theorem 1.1, 1.2 and 1.3) governing draft model acceptance rate (or decoding speed) across three dimensions: pretraining token volume, draft model capacity, and decoding batch size. Building on these laws, we achieve Scylla, which coordinates multi-dimensional scaling for popular LLMs (Llama2/3, Qwen2.5). Empirical validation shows Scylla achieves 1.5-2.2 higher acceptance rate than EAGLE2 and 0.3 higher than EAGLE3 at temperature T = 0, with peak performance gains on summarization and QA tasks (Figure 2). Industrial inference engine deployments demonstrate 2X decoding throughput improvements over EAGLE2 (Table 5), validating the transformative potential of systematic scaling for efficient LLM inference. Code will be released later.
title Scaling Laws for Speculative Decoding
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2505.07858