Unlocking Parallelism in Autoregressive Language Models via Speculative Decoding with Progressive Tree Drafting
摘要
Speculative decoding has significantly accelerated Large Language Model (LLM) inference by alleviating memory-bound bottlenecks. However, traditional speculative decoding typically relies on auxiliary draft modules, incurring significant training and communication overhead. Although recent methods attempt to generate drafts within the target model itself, they often fail to fully exploit its latent parallel capacity due to a lack of structural coordination. In this paper, we propose \textbf{Progressive Tree Drafting (PTD)}, which employs a structured, guided parallel drafting strategy to harness the model's parallel potential. By coupling a progressive tree structure with a stepwise pruning mechanism, PTD actively guides the LLM to explore multiple semantic paths in a single forward pass, ensuring both draft diversity and coherence. Experiments demonstrate that PTD achieves up to decoding speedup across various benchmarks while remaining training-free and model-agnostic. Our code is available at: https://github.com/MINE-USTC/PTD.
引用
@article{arxiv.2607.10661,
title = {Unlocking Parallelism in Autoregressive Language Models via Speculative Decoding with Progressive Tree Drafting},
author = {Zipeng Gao and Zhi Zheng and Qingrong Xia and Junda Lin and Ziwei Zhao and Tong Xu and Zhefeng Wang and Enhong Chen},
journal= {arXiv preprint arXiv:2607.10661},
year = {2026}
}