提出SpecRoll,让大模型推理更快更稳,不改变原有训练目标。
SpecRoll: Fast-Slow Verifier-Feedback Adaptation for Speculative Reinforcement Learning Rollouts

- 快慢双轨适应:快速修正+缓慢更新,保持生成分布一致。
- 在1.5B到14B模型上提速1.26至2.15倍,端到端效率提升1.18倍。
- 适合需要高效推理的数学推理任务,尤其对资源敏感场景友好。
强化学习(RL)后训练可提升大语言模型的推理能力,但自回归生成仍是主要效率瓶颈。推测解码虽能加速生成,但在RL中应用困难,因目标策略持续演进:静态推测器会过时,频繁更新推测器又带来额外开销。本文提出SpecRoll,一种推测式生成引擎,可在保留目标模型采样分布的前提下实现双时间尺度自适应。轻量级未来词头并行生成推测结果,提出的Reflex模块利用延迟验证反馈,在无需反向传播的情况下进行局部隐状态修正。同时,慢路径仅在持续退化时才更新头部参数。SpecRoll结合并发感知的稀疏树验证与精确目标验证,维持目标生成分布和GRPO优化目标不变。在五种规模从1.5B到14B的模型及三个数学推理数据集上,SpecRoll相较原生GRPO实现1.26–2.15倍生成加速和1.21–2.04倍端到端加速;在全部15组对比中优于FastGRPO,平均端到端性能提升1.18倍。受控消融实验表明快慢适应路径具有互补优势。源代码已公开。
原文摘要 · Abstract (English)
Reinforcement learning (RL) post-training improves the reasoning capabilities of large language models, but autoregressive rollout generation remains a major efficiency bottleneck. Speculative decoding can accelerate generation, yet applying it during RL is difficult because the target policy continually evolves: static proposers become stale, while frequent drafter updates add substantial overhead. We introduce SpecRoll, a speculative rollout engine that preserves the target model's sampling distribution while adapting at two timescales. Lightweight future-token heads generate parallel proposals, while our proposed Reflex module uses delayed verifier feedback to perform bounded, trajectory-local hidden-state corrections without backpropagation. A complementary slow path updates the head parameters only when sustained degradation is detected. SpecRoll combines these mechanisms with concurrency-aware sparse-tree verification and exact target verification, leaving the target rollout distribution and GRPO objective unchanged. Across five models ranging from 1.5B to 14B and three mathematical reasoning datasets, SpecRoll achieves 1.26-2.15x generation speedup and 1.21-2.04x end-to-end speedup over vanilla GRPO. It also outperforms FastGRPO in both generation and end-to-end time across all 15 matched settings, with an average pairwise end-to-end gain of 1.18x. Controlled ablations show that the fast and slow adaptation paths provide complementary benefits. Our source code is available at https://anonymous.4open.science/r/SpecRoll-26062006.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。