arXiv:2606.18967cs.LG2026-06被引 1

提升大模型强化学习推演效率,实测延迟降低近20%。

EfficientRollout: System-Aware Self-Speculative Decoding for RL Rollouts

论文配图:EfficientRollout: System-Aware Self-Speculative Decoding for RL Rollouts
图 1 · 摘自论文原文
  • 自洽式推测解码:从目标模型生成量化草稿,无需额外训练。
  • 动态调节推测策略,在计算资源充足时启用,避免性能浪费。
  • 适用于长序列、高温度的强化学习推演,适合追求低延迟的部署场景。

强化学习(RL)已成为大语言模型后训练的关键范式,赋予其强大推理与智能体能力。然而,推演生成仍是主要延迟瓶颈,因自回归采样需逐词解码,且少数长尾生成决定整体耗时。推测解码(SD)可通过并行验证快速生成候选词,有效降低延迟,但其在RL推演中效果有限:(i)目标策略持续演化,固定草稿模型与策略分布逐渐失配;(ii)活跃批大小随推演过程下降,使解码由计算密集转为内存受限,导致并行验证无法充分利用算力。为此,本文提出EfficientRollout,一种系统感知的自推测解码框架。该框架通过目标模型直接生成量化草稿(即自推测),保持草稿与策略同步,无需额外预训练或在线调整。同时,引入系统感知的开关策略与接受感知的草稿长度调节机制,仅在有利条件下启用推测,并根据草稿质量动态匹配预算。实验显示,相比加速后的自回归基线,EfficientRollout可将推演延迟降低19.6%,端到端延迟降低12.7%,且不损失最终模型质量。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has become a representative post-training paradigm for LLMs, enabling strong reasoning and agentic capabilities. However, rollout generation remains a dominant latency bottleneck because autoregressive sampling decodes responses sequentially and a small number of long-tailed generations often determine completion time. Speculative decoding (SD) offers a natural way to address this bottleneck, as it is a well-established technique for serving fixed LLMs that reduces latency by rapidly drafting tokens and accepting them through parallel verification while preserving the target-model distribution. However, its practical speedups do not directly carry over to RL rollouts: (i) the evolving target policy makes any fixed drafter increasingly mismatched with the policy's output distribution; and (ii) active batch sizes shrink throughout rollout decoding, shifting decoding from compute-bound to memory-bound regimes where parallel verification can exploit underutilized compute. Therefore, accelerating RL rollouts requires both a drafter that remains effective under long, high-temperature generations from an evolving policy and system-aware use of SD that avoids compute-bound regimes. We present EfficientRollout, a system-aware self-SD framework designed to address this gap for RL rollouts. EfficientRollout induces a quantized drafter from the target model (i.e. self-speculative decoding), keeping it coupled to the evolving policy without separate drafter pretraining or online adaptation. It further coordinates a system-aware SD toggle policy with acceptance-aware draft-length adaptation, enabling speculation only in beneficial regimes while matching the drafting budget to evolving drafter quality. EfficientRollout reduces rollout and end-to-end latency by up to 19.6% and 12.7%, respectively, over an accelerated AR rollout baseline, while preserving final model quality.

强化学习推理加速推测解码大模型

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