Double-Rollout Reinforcement Learning with Sampling-based Reward Estimation
Organizations: NVIDIA, 2788 San Tomas Expressway, Santa Clara, CA 95051, USA
Abstract
Reinforcement learning with verifiable rewards (RLVR) implemented with Group Relative Policy Optimization (GRPO) has driven substantial progress in training reasoning language models. However, verifiable outcome rewards provide an imperfect measure of reasoning quality: a correct final answer may not be supported by the reasoning trace, whereas an incorrect rollout may contain a nearly correct reasoning trajectory. We present empirical evidence for this mismatch and hypothesize that the resulting reward noise contributes to instability in reinforcement learning for both reasoning accuracy and token efficiency. To address this credit assignment challenge, this paper proposes double-rollout reinforcement learning. For each sampled rollout, we truncate its reasoning trace, sample multiple continuations from the truncated prefix, and replace the original reward with the mean continuation reward, yielding a more robust estimate of reasoning quality. This double-rollout algorithm incurs only a moderate increase in compute cost and yet reaches substantially higher accuracies across math and coding tasks and across Qwen and Nemotron models than baseline algorithms with fewer RL steps. For example, using only DAPO data it trains Nemotron-3-Nano-30B-A3B to reach 95.4% on AIME 2026, versus 91.5% by GRPO and substantially above state of the art for this model size, within 30 RL steps. When combined with length-based rewards, the same algorithm further improves the accuracy-efficiency Pareto frontier over existing efficiency optimizations. These findings suggest that the double-rollout reward estimation is an effective and cost-efficient way to improve both the accuracy and token efficiency of reasoning RL.
Figures & tables
| Start model | Algorithm | RL with DAPO prompts | RL with OCR prompts | |||
|---|---|---|---|---|---|---|
| AIME 2024 | AIME 2025 | AIME 2026 | LCB v5 | LCB v6 | ||
| Start | 90.0% | 88.3% | 89.6% | 72.4% | 67.6% | |
| Nemotron-3-Nano | GRPO | 91.7% | 89.2% | 91.5% | 72.5% | 70.9% |
| -30B-A3B | GRPO w/ large rollout | 90.8% | 89.0% | 91.9% | 70.7% | 70.6% |
| Double-rollout (ours) | 92.1% | 93.8% | 95.4% | 74.6% | 70.7% | |
| Start | 92.5% | 83.8% | 87.1% | 68.3% | 66.3% | |
| Param | RL with DAPO prompts | ||
|---|---|---|---|
| AIME 2024 | AIME 2025 | AIME 2026 | |
| 0-25% | degradation from start model | ||
| 50% | 93.5% | 91.7% | 94.8% |
| 75% | 92.5% | 91.5% | 93.8% |
| 90% | 92.1% | 93.8% | 95.4% |
| 99% | 93.5% | 90.0% | 90.6% |
| Param | RL with DAPO prompts | ||
|---|---|---|---|
| AIME 2024 | AIME 2025 | AIME 2026 | |
| 1 | 91.7% | 89.2% | 91.5% |
| 2 | 92.7% | 91.5% | 94.6% |
| 4 | 92.7% | 91.7% | 94.0% |
| 8 | 92.1% | 93.8% | 95.4% |
| 16 | 91.5% | 92.9% | 95.2% |
Reinforcement Learning
Reinforcement learning studies how agents learn sequential decisions from rewards and interaction, balancing exploration, performance, safety, and adaptation.
See all 1,661 papers