AI & ML
ForeSpark: Joint Draft and Verification Budget Scheduling for Parallel Speculative Decoding
arXiv cs.DC0 views
arXiv:2611.13442v1 Announce Type: new Abstract: Parallel drafters accelerate LLM decoding by proposing a whole block of draft tokens in a single forward pass, making them attractive for LLM serving. Recent systems optimize goodput (i.e., accepted tokens per second) because rejected drafts steal compute from other requests. However, these systems control only how many tokens are verified, so every request still drafts a full block. At high serving load, we find that drafting consumes up to 40% of step time, of which a majority is discarded because it is never verified. Yet, draft-length control from autoregressive drafters does not carry over, because a parallel drafter commits to its length before its single pass. Moreover, draft and verification lengths are coupled across the batch, so they must be selected jointly for every request while keeping the decision computationally cheap at every step. In this paper, we propose work-conserving speculation (WCS), a design that jointly allocates draft and verification budgets per request in LLM serving, and present ForeSpark as its prototype. ForeSpark realizes WCS with (i) a context-aware pre-draft estimator that predicts acceptance for every candidate length from the target model's context features, (ii) a lightweight joint draft-verify scheduler that selects both lengths through a global sort, and (iii) a post-draft budget distribution that spends verification on the most promising tokens. Implemented on vLLM, ForeSpark improves throughput by up to 1.77x over the state-of-the-art DSpark and by 2.49x over autoregressive decoding at a matched per-request generation rate, without changing the output distribution. It also tolerates higher load, sustaining up to 1.51x the request rate of DSpark under latency service level objectives (SLOs).
Source:https://arxiv.org/abs/2611.13442PreviousPlacing LLM Lookup Tables by Per-Layer Prefetch DeadlinesNextMulti-Objective Scheduling of LLM-Based Multi-Agent Workflows in Cloud Computing
Related posts
Disentangling Forced and Internal Climate Variability in Single Realizations using Dynamic Mode Decomposition with Control
0arXiv stat.ML
Uniform-in-Time Weak Propagation-of-Chaos in Shallow Neural Networks
0arXiv stat.ML
Density-Ratio Losses for Post-Hoc Learning to Defer
0arXiv stat.ML
Affine Tracing: A New Paradigm for Probabilistic Linear Solvers
0arXiv stat.ML
The Sampling Complexity of Condorcet Winner Identification in Dueling Bandits
0arXiv stat.ML
SVP Is NP-Hard for Some Rank-2 Cyclotomic Modules
0arXiv cs.CR