dorsal/arxiv
View SchemaSRT: Accelerating Reinforcement Learning via Speculative Rollout with Tree-Structured Cache
| Authors | Chi-Chih Chang, Siqi Zhu, Zhichen Zeng, Haibin Lin, Jiaxuan You, Mohamed S. Abdelfattah, Ziheng Jiang, Xuehai Qian |
|---|---|
| Categories | |
| ArXiv ID | 2601.09083vv1 |
| URL | https://arxiv.org/abs/2601.09083 |
| License | http://arxiv.org/licenses/nonexclusive-distrib/1.0/ |
Abstract
We present Speculative Rollout with Tree-Structured Cache (SRT), a simple, model-free approach to accelerate on-policy reinforcement learning (RL) for language models without sacrificing distributional correctness. SRT exploits the empirical similarity of rollouts for the same prompt across training steps by storing previously generated continuations in a per-prompt tree-structured cache. During generation, the current policy uses this tree as the draft model for performing speculative decoding. To keep the cache fresh and improve draft model quality, SRT updates trees online from ongoing rollouts and proactively performs run-ahead generation during idle GPU bubbles. Integrated into standard RL pipelines (\textit{e.g.}, PPO, GRPO and DAPO) and multi-turn settings, SRT consistently reduces generation and step latency and lowers per-token inference cost, achieving up to 2.08x wall-clock time speedup during rollout.
{
"annotation_id": "536b9df6-3b8e-4308-80ed-e36fc0ea30cc",
"date_created": "2026-02-17T05:53:20.222000Z",
"date_modified": "2026-02-17T05:53:20.222000Z",
"file_hash": "298eb0c21ba9867a99fb54c6e205442be86a50208e3d4cd9b2a1b68b2740fdea",
"private": false,
"record": {
"abstract": "We present Speculative Rollout with Tree-Structured Cache (SRT), a simple, model-free approach to accelerate on-policy reinforcement learning (RL) for language models without sacrificing distributional correctness. SRT exploits the empirical similarity of rollouts for the same prompt across training steps by storing previously generated continuations in a per-prompt tree-structured cache. During generation, the current policy uses this tree as the draft model for performing speculative decoding. To keep the cache fresh and improve draft model quality, SRT updates trees online from ongoing rollouts and proactively performs run-ahead generation during idle GPU bubbles. Integrated into standard RL pipelines (\\textit{e.g.}, PPO, GRPO and DAPO) and multi-turn settings, SRT consistently reduces generation and step latency and lowers per-token inference cost, achieving up to 2.08x wall-clock time speedup during rollout.",
"arxiv_id": "2601.09083",
"authors": [
"Chi-Chih Chang",
"Siqi Zhu",
"Zhichen Zeng",
"Haibin Lin",
"Jiaxuan You",
"Mohamed S. Abdelfattah",
"Ziheng Jiang",
"Xuehai Qian"
],
"categories": [
"cs.LG"
],
"license": "http://arxiv.org/licenses/nonexclusive-distrib/1.0/",
"title": "SRT: Accelerating Reinforcement Learning via Speculative Rollout with Tree-Structured Cache",
"url": "https://arxiv.org/abs/2601.09083",
"version": "v1"
},
"schema_id": "dorsal/arxiv",
"source": {
"execution_id": "38651384-eeea-4e82-bd5c-bff653f2069a",
"id": "arXiv Dataset",
"type": "Model",
"variant": "snapshot-2026-01-17",
"version": "0.1.0"
},
"user_id": 1000002
}