JetSpec++ drafts ahead while the target checks current tokens
In plain words
Language models usually write text one token at a time. Speculative decoding lets a smaller draft propose several tokens, which the main model checks together, but making a good draft can take time.
JetSpec++ predicts what the main model’s check may find and starts preparing the next draft while that check runs. Across seven tests with Qwen3-8B, it averaged more committed tokens per check than DSpark at temperature zero and a 16-token verification budget. The result is limited to those tested conditions.
The research
Speculative decoding can speed text generation, but a learned draft that uses the target model’s internal states may need to wait until the target checks the current block. Can a system prepare the next block during that wait and still produce proposals the target accepts often enough to help?
JetSpec++ predicts the target’s unfinished verification outcome and refines the next draft in parallel, overlapping drafting with the current check.
JetSpec++ predicts the target states and likely outcomes that are still missing during verification, then prepares possible next blocks while the current block is checked. Its draft model also revises proposals in parallel, using context refresh and anchor replacement between passes.
Across seven tests, JetSpec++ averaged 8.94 committed tokens per verification round versus DSpark’s 6.89 for Qwen3-8B at temperature zero and budget 16.
Since more committed tokens per round can reduce generation work, the result supports the method’s ability to overlap drafting while retaining useful proposals in these tests.
The evaluations cover two target models, seven named tests, and the reported decoding and serving setups. The serving results use single-request runs on GB200 GPUs, and the authors say systems optimization could further reduce drafting overhead; they do not quantify that possible gain.
Analysis
Deep
- Original paper authors
- Zhaoxiang Feng, Lanxiang Hu, Haoran Yuan, Yu-Yang Qian, Jihui Yang
- Submitted to arXiv
- License
- CC BY 4.0