arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2603.03333cs.CL
收藏

无需训练的Dropout采样用于推测解码中的语义标记接受

Training-free Dropout Sampling for Semantic Token Acceptance in Speculative Decoding

Jeongtae Lee, Minjung Jo, Hyunjoon Jeong, Gunho Park, Sunghyeon Woo, Joonghoon Kim, Se Jung Kwon, Dongsoo Lee

AI总结:

本文提出DropMatch,一种无需训练的数据和校准的推测解码方法,通过蒙特卡洛dropout实现基于采样的token接受,提升推理速度。

摘要:

推测解码通过提出轻量级草稿模型的token并使用目标模型选择性接受,从而加速大语言模型的推理。本文引入DropMatch,一种新颖的方法,通过仅在LM头部应用蒙特卡洛dropout,将草稿token与目标模型的预测分布进行匹配,从而实现基于采样的接受决策。通过生成多个解码路径,我们的方法形成一个经验token分布,用以评估草稿token的一致性。这种接受机制使模型能够在适当的dropout概率下自适应地控制解码路径的大小,防止目标模型预测分布的显著扭曲。所提出的方法以无需训练、无需数据和无需校准的方式运行,不需要对预训练模型进行架构修改,并可以与广泛存在的推测解码和推理加速技术正交整合。在多个基准测试中的实验表明,我们的方法在保持竞争性任务性能的同时增加了接受长度,从而在标准基线之上实现1.09x至1.33x的推理加速,当应用于EAGLE3之上时,还能进一步实现额外1.09x的加速。

英文摘要:

Speculative decoding accelerates large language model inference by proposing tokens with a lightweight draft model and selectively accepting them using a target model. This work introduces DropMatch, a novel approach that matches draft tokens to the predictive distribution of the target model via Monte Carlo dropout applied exclusively to the LM head, enabling sampling-based acceptance decisions. By generating multiple decoding paths, our method forms an empirical token distribution against which draft tokens are evaluated for consistency. This acceptance mechanism enables the model to adaptively control the size of decoding paths under an appropriate dropout probability, preventing substantial distortion of the target model predictive distribution. The proposed method operates in a training-free, data-free, and calibration-free manner, requires no architectural modification to pretrained models, and can be orthogonally integrated with a wide range of existing speculative decoding and inference acceleration techniques. Experiments across multiple benchmarks demonstrate that our approach increases acceptance length while maintaining competitive task performance, yielding inference speedups ranging from 1.09x to 1.33x over the standard baseline, and up to an additional 1.09x speedup when applied on top of EAGLE3.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑