arXiv:2605.14978cs.CL2026-05

用强化学习优化生成窗口,提升大模型推理速度

Performance-Driven Policy Optimization for Speculative Decoding with Adaptive Windowing

论文配图:Performance-Driven Policy Optimization for Speculative Decoding with Adaptive Windowing
图 1 · 摘自论文原文
  • 改用窗口级奖励机制,不再仅模仿单个词
  • 平均接受长度达6.29-6.52,提速3.39-4.36倍
  • 适合追求高效推理的部署场景

推测解码通过轻量级草稿模型提出候选词窗口,由更大的目标模型并行验证以加速大语言模型推理。实践中,因难生成位置导致早期不匹配会截断有效前缀,使整个推测窗口失效。现有基于学习的草稿模型仍采用逐词监督目标,但推测效用本质上是窗口级且对前缀敏感。本文提出PPOW(性能驱动的策略优化与自适应窗口机制),一种将草稿模型优化从逐词模仿转向窗口级优化的强化学习框架。PPOW结合代价感知加速奖励、基于分布的贴近度奖励及自适应分歧感知窗口机制,优先选择高置信度差异的有信息量窗口。在统一解码协议下,多个模型族和基准测试中,平均接受长度达6.29-6.52,加速比为3.39-4.36×。结果表明,性能驱动的窗口级优化是提升推测解码效率的实用方法。

原文摘要 · Abstract (English)

Speculative decoding accelerates LLM inference by having a lightweight draft model propose speculative windows of candidate tokens for parallel verification by a larger target model. In practice, speculative efficiency is often bottlenecked by hard-to-draft positions, where an early mismatch truncates the accepted prefix and invalidates the rest of the speculative window. Most learning-based drafters are still optimized with token-level supervised objectives, even though speculative utility is inherently window-level and prefix-sensitive. We propose PPOW (Performance-Driven Policy Optimization with Adaptive Windowing), a reinforcement learning framework that shifts drafter optimization from token-level imitation to window-level optimization. PPOW combines a Cost-Aware Speedup Reward, a Distribution-Based Proximity Reward, and Adaptive Divergence-Aware Windowing, which prioritizes informative windows with high confidence-weighted draft-target divergence. PPOW achieves average acceptance lengths of 6.29-6.52 and speedups of 3.39-4.36$\times$ across multiple model families and benchmarks under a unified decoding protocol. These results show that performance-driven window-level optimization is a practical approach to improving speculative decoding efficiency.

推理加速强化学习推测解码

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