S cubed R: Fast and Faithful RL Rollouts via Self-Sparse Speculative Decoding
cs.LG, cs.CL
Submitted: 2026-10-09
Updated: 2026-10-09
Code: https://github.com/apoorvumang/prompt-lookup-decoding
License: http://creativecommons.org/licenses/by-nc-sa/4.0/
The gist: Fast rollout generation for long-context reinforcement learning with verifiable rewards (RLVR) requires reducing KV-cache traffic without changing the response groups used for policy optimization.
Terminology
Abstract
Fast rollout generation for long-context reinforcement learning with verifiable rewards (RLVR) requires reducing KV-cache traffic without changing the response groups used for policy optimization. In our profiled RLVR workloads, attention accounts for 72--76% of autoregressive decoding kernel time. Sparse attention reduces these reads, but sampling directly from a sparse policy changes the response groups used for learning. For group-relative optimization, we show that even exact response-wise reweighting does not generally recover the expected dense-policy update because the advantage of each response depends on the rewards of other group members. We propose Self-Sparse Speculative Rollouts (S cubed R). It uses sparse attention to propose tokens, then generates responses through full-attention verification and exact speculative sampling. Drafting and verification share the same policy parameter, eliminating the need to train or update a separate drafter. The resulting response groups follow the dense-policy distribution. Under S cubed R, we characterize speedup conditions and the costs of drafting and verification. We found that dense verification still requires the complete historical KV cache. A KV-aware scheduler therefore limits KV-cache repeated offloading and restoration while maintaining rollout parallelism. % OR thrashing In a matched ablation with an aggressive sparse drafting budget, S cubed R maintains stable training while direct sparse sampling with training-side correction becomes unstable. On long-context mathematical reasoning, S cubed R achieves 1.39 times 1.35 times the end-to-end training throughput of autoregressive rollouts on Qwen3 model, with comparable task performance.
Sources
- Importance Sampling: Intrinsic Dimension and Computational Cost
- R-KV: Redundancy-aware KV Cache Compression for Reasoning Models
- SRT: Accelerating Reinforcement Learning via Speculative Rollout with Tree-Structured Cache
- Accelerating Large Language Model Decoding with Speculative Sampling
- DFlash: Block Diffusion for Flash Speculative Decoding
- ReSpec: Towards Optimizing Speculative Decoding in Reinforcement Learning Systems
- CONCUR: High-Throughput Agentic Batch Inference of LLM via Congestion-Based Concurrency Control
- Jackpot: Optimal Budgeted Rejection Sampling for Extreme Actor-Policy Mismatch Reinforcement Learning
- DSpark: Confidence-Scheduled Speculative Decoding with Semi-Autoregressive Generation
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- Beyond Benchmarks: MathArena as an Evaluation Platform for Mathematics with LLMs
- History Rhymes: Accelerating LLM Reinforcement Learning with RhymeRL
- Accelerating RL Post-Training Rollouts via System-Integrated Speculative Decoding
- EfficientRollout: System-Aware Self-Speculative Decoding for RL Rollouts
- Trust Region Masking for Long-Horizon LLM Reinforcement Learning
- QuRL: Efficient Reinforcement Learning with Quantized Rollout
- Every Step Evolves: Scaling Reinforcement Learning for Trillion-Scale Thinking Model
- SPEC-RL: Accelerating On-Policy Reinforcement Learning with Speculative Rollouts
- Speculative Decoding: Performance or Illusion?
- A Sparse Glimpse of the Whole: Train-Free Self-Speculative Decoding
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks