arXiv:2511.00351cs.LGcs.CL2025-11中稿 · NeurIPS被引 2

只拒绝影响任务性能的关键词元,实现更快推理而不降质

Reject Only Critical Tokens: Pivot-Aware Speculative Decoding

  • 仅拒绝可能降低最终输出效用的关键词元,放宽传统采样匹配要求
  • 在多个数据集上实现最高2.5倍加速,且任务性能与原模型相当
  • 适合追求高推理速度又需保证输出质量的落地场景

推测解码(Speculative Decoding, SD)能精确匹配目标模型的输出分布,但这种严格要求导致接受率过低,限制了加速潜力。我们提出新视角:解码目标应是匹配目标模型的预期效用(即任务特定性能),而非严格分布一致。基于此,我们提出枢轴感知推测解码(Pivot-Aware Speculative Decoding),仅拒绝那些会导致最终输出效用下降的关键词元(称为枢轴词元)。我们设计方法标注词元是否为关键,并训练轻量分类器识别。该方法可视为标准SD的宽松版本,显著提高接受率并保留效用。在多个数据集上的实验表明,可实现最高2.5倍加速,且效用相当。源代码已公开于https://github.com/amir-zsh/PAD。

原文摘要 · Abstract (English)

Speculative Decoding (SD) ensures that the output matches the target model's distribution exactly. However, we argue that this distribution matching requirement is too stringent and results in unnecessarily low acceptance rates, limiting potential speedups. Instead, we advocate a reformulation of the decoding objective: the proposed decoding strategy should match the expected utility, i.e., the task-specific performance, of the target model. This perspective also aligns better with real-world use cases of LLMs, where utility (e.g., code correctness, factual accuracy) is often more important than sampling distribution. Based on this reformulation, we propose a novel decoding strategy: Pivot-Aware Speculative Decoding, which rejects only those tokens that would lead to a utility drop in the final output. We refer to these critical tokens as pivot tokens. We propose a method for labeling tokens as pivotal or non-pivotal and train a lightweight classifier to detect them. This method can be viewed as a relaxed version of standard SD, which offers much higher acceptance while preserving utility. We evaluate our method across various datasets, demonstrating that we can achieve up to $2.5\times$ speedup with comparable utility. Source code is available at https://github.com/amir-zsh/PAD.

推理加速推测解码效用优化

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