TALON让大模型生成更智能:动态调整候选路径,提速超5倍。
TALON: Confidence-Aware Speculative Decoding with Adaptive Token Trees
- 根据上下文难度动态扩展候选树,不固定宽度和深度。
- 在5个模型、6个数据集上实现最高5.16倍推理加速。
- 无需训练,可无缝接入现有方法,适合追求高效推理的开发者。
推测解码(SD)已成为在不损失输出质量的前提下加速大语言模型推理的标准技术。近期进展从串行链式草案生成转向树结构生成,其中草稿模型构建候选令牌的树以并行探索多种可能的草案。然而,现有树结构方法通常采用固定宽度和固定深度的草稿树,无法适应不同令牌和上下文的难度差异,导致草稿模型无法动态调整树结构以在困难令牌处提前终止,在简单情况下延长生成。为此,我们提出TALON,一种无需训练、基于预算驱动的自适应树扩展框架,可无缝集成到现有树结构方法中。与静态方法不同,TALON通过迭代构建草稿树直至达到固定令牌预算,采用混合扩展策略,自适应地将节点预算分配至草稿树的每一层。该框架自然形成‘深而窄’(确定性上下文)或‘浅而宽’(不确定性分支)的结构,有效优化给定预算下探索广度与生成深度的权衡。在5个模型和6个数据集上的大量实验表明,TALON始终优于最先进方法EAGLE-3,相较于自回归解码实现最高5.16倍端到端加速。
原文摘要 · Abstract (English)
Speculative decoding (SD) has become a standard technique for accelerating LLM inference without sacrificing output quality. Recent advances in speculative decoding have shifted from sequential chain-based drafting to tree-structured generation, where the draft model constructs a tree of candidate tokens to explore multiple possible drafts in parallel. However, existing tree-based SD methods typically build a fixed-width, fixed-depth draft tree, which fails to adapt to the varying difficulty of tokens and contexts. As a result, the draft model cannot dynamically adjust the tree structure to early stop on difficult tokens and extend generation for simple ones. To address these challenges, we introduce TALON, a training-free, budget-driven adaptive tree expansion framework that can be plugged into existing tree-based methods. Unlike static methods, TALON constructs the draft tree iteratively until a fixed token budget is met, using a hybrid expansion strategy that adaptively allocates the node budget to each layer of the draft tree. This framework naturally shapes the draft tree into a "deep-and-narrow" form for deterministic contexts and a "shallow-and-wide" form for uncertain branches, effectively optimizing the trade-off between exploration width and generation depth under a given budget. Extensive experiments across 5 models and 6 datasets demonstrate that TALON consistently outperforms state-of-the-art EAGLE-3, achieving up to 5.16x end-to-end speedup over auto-regressive decoding.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。