arXiv:2602.16961cs.ITcs.LG2026-02

提出贪心多路径验证,让大模型生成更快更准。

Greedy Multi-Path Block Verification for Faster Decoding in Speculative Sampling

  • 用多条候选路径并行验证,提升生成效率
  • 实测比现有方法快15%以上,吞吐提升超15%
  • 适合追求高速推理的LLM部署场景

L步推测解码旨在通过廉价草稿模型生成长度为L的候选词序列,以加速目标模型的自回归解码。标准方法在每一步独立验证,而块验证(BV)则联合考虑路径上所有概率,被证明是仅使用路径内概率的最优验证策略。本文进一步证明:即使引入路径外概率,块验证仍是最优的,基于信息无关线性规划(LP)构造。将该框架扩展至多路径情形,提出自然的多路径块验证推广。由于最优算法不可行,采用贪心策略设计高效方法——贪心多路径块验证(GBV)。实验表明,相比标准块验证,GBV可提升30%以上的块效率,解码耗时减少15%以上;在Llama-3 70B上,端到端吞吐率超越当前最佳多路径验证方法超过15%。

原文摘要 · Abstract (English)

The goal of $L$-step speculative decoding is to accelerate autoregressive decoding of a target model by using a cheaper draft model to generate a candidate path of $L$ tokens. Based on a verification algorithm involving target and draft model probabilities, a prefix of the candidate sequence is accepted, and an additional correction token is sampled from a residual distribution to ensure that the final output adheres to the target distribution. While standard speculative decoding uses a verification algorithm which is independent at each token on the path, a recent extension called block verification uses a joint condition involving all sampled on-path probabilities. Block verification (BV) was shown to be optimal over all verification algorithms which use only on-path probabilities, improving on standard speculative decoding. In this work, we first show that block verification is optimal even over verification algorithms that use off-path probabilities, by constructing an information-agnostic linear program (LP). Further, we can extend our LP to the setting where the draft model samples multiple candidate paths, and use it to construct a natural class of multi-path block verification generalizations. While computing the optimal algorithm in this class is not tractable, by considering a stricter class of greedy algorithms, we can formulate an efficient method called greedy multi-path block verification (GBV). Empirically, GBV can improve block efficiency by over 30% and reduce decoding walltimes by over 15% relative to BV. On Llama-3 70B, GBV can improve the end-to-end decoding throughput over SOTA multi-path verification methods by more than 15%.

推理加速大模型生成优化

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