GRAFT: Adaptive DLM-Based Draft Tree Construction with Target-Distilled Edge Scoring
GRAFT 面向扩散语言模型草稿器的树式推测解码,提出 TDES 目标模型蒸馏边打分,显式学习父子 token 兼容性;SABA 则按每轮状态在草稿收益与验证成本间动态分配树预算。多模型、多任务实验相对自回归解码取得 2.13–6.36 倍端到端加速,每轮额外开销低于 0.5 ms,约为验证延迟的 1.4%。
INSIGHT推测解码的关键不只是提高 draft 概率,而是让草稿树的父子关系与 target 分布一致,并随解码状态调整预算。对搜索重排或 RAG 生成服务,可把 TDES/SABA 迁移到多分支候选验证;上线时重点监控接受 token 数、验证 GPU 时间和尾延迟,而不是只报平均加速比。