MEGA Hub

D-cut: Adaptive Verification Depth Pruning for Batched Speculative Decoding

Authors

Do you know Tianyu Liu?You can claim authorship or link another user.Do you know Yuhao Shen?You can claim authorship or link another user.Do you know Rui Cen?You can claim authorship or link another user.Do you know Junhan Shi?You can claim authorship or link another user.Do you know Jiebin Zhang?You can claim authorship or link another user.Do you know Guangshuo Qin?You can claim authorship or link another user.Do you know Hong Liu?You can claim authorship or link another user.Do you know Song Liu?You can claim authorship or link another user.Do you know Guanghua Yu?You can claim authorship or link another user.Do you know Jianchen Zhu?You can claim authorship or link another user.

Abstract

Speculative decoding accelerates large language model (LLM) inference without compromising output quality. Recent parallel drafting methods further improve single-request performance by decoupling draft length from drafting latency, enabling longer drafts and higher mean accepted tokens (MAT). However, under high request concurrency, long drafts waste substantial computation on rejected tokens, increasing verification cost and potentially making speculative decoding slower than autoregressive decoding. We present D-Cut, an adaptive pruning method that selects draft tokens jointly across the batch and concentrates the verification budget on tokens most likely to be accepted. D-Cut is motivated by two observations. First, acceptance lengths vary considerably across concurrent requests; D-Cut therefore performs cross-request pruning, allocating the verification budget adaptively according to draft confidence. Second, verification cost depends strongly on the deployment environment, including GPU architecture and parallelism strategy; D-Cut incorporates a runtime cost model to adapt its pruning depth to the target environment. Experiments on dense and mixture-of-experts (MoE) models show that, under high concurrency, D-Cut improves the average speedup from \(1.26\times\) to \(1.65\times\), restores acceleration in dense-model configurations where long-draft baselines are slower than autoregressive decoding, and achieves up to \(3.0\times\) speedup over autoregressive decoding on MoE models.

Community

00