SPEC-RL: Accelerating On-Policy Reinforcement Learning with Speculative Rollouts

Bingshuai Liu, Ante Wang, Zijun Min, Liang Yao, Haibo Zhang +5 more
9/27/2025
cs.LGcs.AIcs.CL

Official code: ShopeeLLM/Spec-RL (67 stars, Apache-2.0, last commit 2025-12-01).

Abstract

Large Language Models (LLMs) increasingly rely on reinforcement learning with verifiable rewards (RLVR) to elicit reliable chain-of-thought reasoning. However, the training process remains bottlenecked by the computationally expensive rollout stage. Existing acceleration methods-such as parallelization, objective- and data-driven modifications, and replay buffers-either incur diminishing returns, introduce bias, or overlook redundancy across iterations. We identify that rollouts from consecutive training epochs frequently share a large portion of overlapping segments, wasting computation. To address this, we propose SPEC-RL, a novel framework that integrates SPECulative decoding with the RL rollout process. SPEC-RL reuses prior trajectory segments as speculative prefixes and extends them via a draft-and-verify mechanism, avoiding redundant generation while ensuring policy consistency. Experiments on diverse math reasoning and generalization benchmarks, including AIME24, MATH-500, OlympiadBench, MMLU-STEM, and others, demonstrate that SPEC-RL reduces rollout time by 2-3x without compromising policy quality. As a purely rollout-stage enhancement, SPEC-RL integrates seamlessly with mainstream algorithms (e.g., PPO, GRPO, DAPO), offering a general and practical path to scale RLVR for large reasoning models. Our code is available at https://github.com/ShopeeLLM/Spec-RL

View on arXivView PDF

Code Implementations(1)

ShopeeLLM/Spec-RLOfficial100%
Apache-2.0

SPEC-RL: Accelerating On-Policy Reinforcement Learning via Speculative Rollouts

673Roff, ANTLRSep 19, 202510 months agoApache-2.0

Similar Papers

Cite this paper

@article{liu2025specrl,
  title  = {SPEC-RL: Accelerating On-Policy Reinforcement Learning with Speculative Rollouts},
  author = {Bingshuai Liu and Ante Wang and Zijun Min and Liang Yao and Haibo Zhang and Yang Liu and Xu Han and Peng Li and Anxiang Zeng and Jinsong Su},
  year   = {2025},
  eprint = {2509.23232},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2509.23232v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2509.23232.svg)](https://codeofpaper.com/paper/2509.23232)

Discussion