arXiv cs.AI· Liang He, SiYuan Ma, Qishi Zhan, Yongqi Fan, Mingyu Cao, Qizhen Lan, Zhaolu Kang, Xilu Wang·· 11 小时前AI 评分34
ValuSpec:树状投机解码在目标验证前的即插即用候选评估机制
ValuSpec: Plug-and-Play Candidate Valuation before Target Verification for Tree-Based Speculative Decoding
AI 导读
ValuSpec 是一种即插即用的候选评估机制,可在目标模型验证前过滤候选树中的低质量分支,无需参数训练,仅用校准集上目标模型的贪心轨迹估计过滤阈值。在 HumanEval、GSM8K、Dolly 和 SpecBench 上,检索式树设置下实现最高 2.38x 端到端加速;接入 EAGLE-3 动态树框架后最高达 2.78x,且不改动其树构建策略。
正文 · 原文
Abstract:Tree-based speculative decoding accelerates large language model inference by verifying multiple candidate branches in parallel. However, in practical serving scenarios, similar requests repeatedly induce overlapping candidate branches. While retrieval-based methods mitigate this by reusing historical drafts, the resulting candidate trees often contain many low-quality branches as the candidate tree grows, which reduces verification efficiency and wastes target-side computation. To address this, we propose ValuSpec, a plug-and-play candidate valuation mechanism that filters low-quality branches from a supplied candidate tree before expensive target-side verification. ValuSpec introduces transfer-based supervision to align filtering with the target model's behavior. A filtering threshold is estimated from the target model's greedy trajectories on a calibration split, without any parameter training. The target model then verifies the filtered tree under the output-identical greedy rule, so accepted tokens are guaranteed to match standard autoregressive decoding exactly. Experimental results on HumanEval, GSM8K, Dolly, and SpecBench show that, in our main evaluation, ValuSpec achieves up to 2.38x end-to-end speedup in retrieval-based tree settings. Furthermore, integrating ValuSpec into the EAGLE-3 dynamic tree framework yields up to 2.78x end-to-end speedup without modifying its tree construction strategy, demonstrating that ValuSpec serves as a generally applicable and plug-and-play candidate valuation module.
| Comments: | NeurIPS 2026 poster |
| Subjects: | Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.31338 [cs.AI] |
| (or arXiv:2610.31338v2 [cs.AI] for this version) | |
| https://doi.org/10.48550/arXiv.2610.31338 arXiv-issued DOI via DataCite |
Submission history
From: Liang He [view email]
[v1]
Thu, 22 Oct 2026 14:23:26 UTC (4,560 KB)
[v2]
Thu, 5 Nov 2026 15:09:32 UTC (4,560 KB)
来源:arXiv cs.AI · arxiv.org