研究高温生成对推测解码的影响,提出温度一致性优化方案。
Temperature-Centric Investigation of Speculative Decoding with Knowledge Distillation
- 用知识蒸馏保持温度一致,缓解高温下推测失败问题
- 在高温和跨域测试下,模型性能下降明显,需针对性优化
- 适合关注推理加速与生成质量平衡的研究者
推测解码是加速自回归大语言模型推理的关键技术,通过小型草稿模型预估一串标记,由目标模型评估是否接受。尽管已有大量研究提升其效率,但生成配置(尤其是解码温度)对解码过程的影响仍不清晰。本文深入探究解码温度对推测解码有效性的影响。首先,基于知识蒸馏(KD),揭示了高温下解码面临的挑战,并证明在统一温度设置下进行知识蒸馏可有效缓解该问题。同时,研究了在分布外测试集和超出范围温度下的表现。基于上述发现,本文初步探索了在高温生成场景下的进一步提速方法。工作揭示了生成配置对推测解码性能的显著影响,强调需发展针对多样化解码配置的优化方法。代码已公开于 https://github.com/ozyyshr/TempSpec。
原文摘要 · Abstract (English)
Speculative decoding stands as a pivotal technique to expedite inference in autoregressive (large) language models. This method employs a smaller draft model to speculate a block of tokens, which the target model then evaluates for acceptance. Despite a wealth of studies aimed at increasing the efficiency of speculative decoding, the influence of generation configurations on the decoding process remains poorly understood, especially concerning decoding temperatures. This paper delves into the effects of decoding temperatures on speculative decoding's efficacy. Beginning with knowledge distillation (KD), we first highlight the challenge of decoding at higher temperatures, and demonstrate KD in a consistent temperature setting could be a remedy. We also investigate the effects of out-of-domain testing sets with out-of-range temperatures. Building upon these findings, we take an initial step to further the speedup for speculative decoding, particularly in a high-temperature generation setting. Our work offers new insights into how generation configurations drastically affect the performance of speculative decoding, and underscores the need for developing methods that focus on diverse decoding configurations. Code is publically available at https://github.com/ozyyshr/TempSpec.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。