跳到正文
arXiv cs.AI· Ekagra Ranjan, Bharat Venkitesh·· 2 小时前AI 评分41

Hybrid Spark:让内容自选起草者的自适应投机解码框架

Hybrid Spark: Adaptive Speculative Decoding That Lets the Content Choose the Drafter

AI 导读

arXiv 论文提出 Hybrid Spark 框架,在每步同时运行基于模型和无模型的 drafter(如 n-gram),为每个请求选择更优的草稿链,并完全在 GPU 上运行。

正文 · 原文

View PDF HTML (experimental)

Abstract:Speculative decoding (SD) speeds up large language model (LLM) serving, but its gain depends on the workload's \emph{shape}, such as batch size and context length, and on its \emph{content}. A model-based drafter proposes a short block on any content, while a model-free drafter such as n-gram lookup proposes long copies at almost no cost when the output repeats its input, as in code editing. Adaptive verification spends a batch-wide verification budget on the draft tokens most likely to be accepted, but it supports only one model-based drafter. Model-free drafters provide no per-token confidence, so they cannot take part in budget allocation, and deployments must either fix one drafter or split their GPUs between drafters. We present \sys, a framework that runs a model-based and a model-free drafter on every step and gives each request the better of their chains, within adaptive verification and entirely on the GPU, which keeps asynchronous scheduling intact. The key idea is a price per verified draft token: the goodput that slot would earn elsewhere in the batch. Once the model-free drafter's tokens carry calibrated acceptance estimates, fit once on general text, this price puts both drafters' chains on one scale. Instantiated with \dspark and an n-gram drafter in vLLM, \sys achieves 1.10--1.62\x the output throughput of \dspark on code editing at batch size~4 and 1.04--1.27\x at batch size~128, while staying at 0.97--1.01\x of it on chat at temperature~0. A single deployment can therefore serve both workloads.
Comments: Speculative Decoding
Subjects: Artificial Intelligence (cs.AI)
Cite as: arXiv:2611.05611 [cs.AI]
  (or arXiv:2611.05611v1 [cs.AI] for this version)
  https://doi.org/10.48550/arXiv.2611.05611

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Ekagra Ranjan [view email]
[v1] Tue, 3 Nov 2026 20:55:26 UTC (191 KB)

来源:arXiv cs.AI · arxiv.org