arXiv:2604.09731cs.DCcs.AI2026-04中稿 · ECCV

SMART动态控制生成树扩展,避免计算浪费,提升推理速度。

SMART: When is it Actually Worth Expanding a Speculative Tree?

论文配图:SMART: When is it Actually Worth Expanding a Speculative Tree?
图 1 · 摘自论文原文
  • 基于硬件效率的边际收益分析,实时决定是否扩展生成树节点。
  • 在多模态和语言模型上平均提速20%(多模态)和15.4%(语言模型)。
  • 无需训练,可直接接入现有系统,适合高并发、算力受限场景。

基于树的推测解码通过单次目标模型前向传播验证分支的草稿令牌来加速自回归生成。然而,现有方法侧重于最大化令牌级似然或接受令牌数量,忽略了关键的“效率悖论”:在大规模下,草稿与验证的计算开销可能超线性增长,导致批处理规模增大或硬件饱和时出现负的墙钟速度提升。为此,我们提出SMART,一种面向系统的边际分析框架,用于运行时树结构构建。SMART将树扩展重构为一个面向硬件的优化问题,直接最大化端到端速度提升。通过在推理时应用严谨的边际收益-成本准则,仅当节点的边际收益-成本比超过树级速度提升时才扩展该节点。SMART无需训练,可作为即插即用控制器集成至MSD和EAGLE等现有框架。在三个MLLM(如LLaVA、Qwen2-VL)和四个LLM(如Llama-3.1、DeepSeek-R1)上的广泛评估表明,SMART持续优于最先进基线,在计算密集型批处理场景和多种GPU架构下,为MLLM带来平均20.0%的额外提速,为LLMs带来15.4%的额外提速,且无性能损失。

原文摘要 · Abstract (English)

Tree-based speculative decoding accelerates autoregressive generation by verifying a branching tree of draft tokens in a single target-model forward pass. However, existing methods prioritize maximizing token-level likelihood or the number of accepted tokens while ignoring a critical ``efficiency paradox'': the computational overhead of drafting and verifying big trees can grow super-linearly, particularly at scale. This often leads to negative wall-clock speedup when batch sizes increase or hardware saturation limits are reached. To address this, we propose SMART, a system-aware marginal analysis framework for runtime tree construction. SMART reformulates tree expansion as a hardware-aware optimization problem that directly maximizes end-to-end speedup. By applying a principled marginal benefit--cost rule at inference time, SMART expands a node only when its marginal benefit--cost ratio exceeds the tree-level speedup. SMART is training-free and serves as a plug-and-play controller for existing frameworks like MSD and EAGLE. Extensive evaluations across three MLLMs (e.g., LLaVA, Qwen2-VL) and four LLMs (e.g., Llama-3.1, DeepSeek-R1) demonstrate that SMART consistently outperforms state-of-the-art baselines. It delivers an average additional speedup of 20.0\% for MLLMs and 15.4\% for LLMs across compute-bound batching regimes and diverse GPU architectures without performance loss.

推理加速生成树高效生成多模态

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