arXiv:2607.14647cs.CL2026-07被引 1

动态裁剪草案验证深度,提升高并发下大模型推理速度

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

论文配图:D-cut: Adaptive Verification Depth Pruning for Batched Speculative Decoding
图 1 · 摘自论文原文
  • 跨请求联合裁剪,根据接受概率分配验证资源
  • 高并发下平均提速从1.26倍提升至1.65倍,MoE模型最高达3.0倍
  • 自适应适配硬件环境,适合部署在多请求并发场景

推测解码可加速大语言模型推理且不降低输出质量。近期并行草稿方法通过解耦草稿长度与草稿延迟,支持更长草稿,提升平均接受标记数(MAT)。但在高请求并发下,长草稿会浪费大量计算在被拒绝的标记上,增加验证开销,可能导致推测解码慢于自回归解码。我们提出D-Cut,一种自适应裁剪方法,跨批次联合选择草稿标记,将验证预算集中于最可能被接受的标记。D-Cut基于两点观察:其一,不同并发请求的接受长度差异显著,因此实施跨请求裁剪,按草稿置信度动态分配验证预算;其二,验证成本受部署环境(如GPU架构、并行策略)影响大,D-Cut引入运行时成本模型,自适应调整裁剪深度。在密集模型和混合专家(MoE)模型上的实验表明,在高并发下,D-Cut将平均加速比从1.26倍提升至1.65倍,恢复了密集模型配置中因长草稿导致的加速失效问题,并在MoE模型上实现高达3.0倍于自回归解码的速度提升。

原文摘要 · Abstract (English)

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.

推理加速大模型批处理自适应

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。