arXiv AI

Bayes-Adaptive MDPs for Rollout Allocation to Enhance GRPO Training

The paper introduces AdaRoll-GRPO, a method that allocates rollout budgets adaptively during Group Relative Policy Optimization (GRPO) training for large language models. By modeling rollout allocation as a Bayes‑adaptive Markov decision process, the approach learns an offline policy that assigns more rollouts to difficult questions and fewer to easier ones, all while keeping the total budget fixed. Experiments across several reasoning datasets and LLM architectures show that this adaptive strategy consistently improves GRPO training performance compared to the uniform allocation used in standard GRPO.

arXiv Machine Learning
Jun 9

The Easy, the Hard, and the Learnable: Confidence and Difficulty-Adaptive Policy Optimization for LLM Reasoning

arXiv:2606. 07950v1 Announce Type: new Abstract: RL with verifiable rewards can substantially improve LLM reasoning, yet standard GRPO-style training often treats easy, hard, and learnable questions alike through uniform sampling and weighting, leading to inefficient compute allocation.

By Zhanke Zhou, Xiangyu Lu, Chentao Cao, Brando Miranda, Tongliang Liu, Bo Han, Sanmi Koyejo
arXiv AI
3d ago

When Does Closed-Loop Curriculum Adaptation Help GRPO? An Analysis on Countdown

The paper introduces CLOSEDLOOPGAUSSIAN, an adaptive curriculum strategy for Group Relative Policy Optimization (GRPO) that dynamically adjusts the pace and spread of an Easy-to-Hard Gaussian sampler based on probe estimates of per-difficulty accuracy. Unlike fixed easy-to-hard schedules, the method accelerates training when the model outpaces a reference curriculum, slows it down when it lags, and broadens sampling when earlier difficulties regress, thereby keeping training focused on learnable yet challenging tasks. Experiments on symbolic arithmetic reasoning demonstrate that CLOSEDLOOPGAUSSIAN outperforms vanilla GRPO and fixed open-loop curricula, especially in intermediate-difficulty regimes where group-relative learning signals are most informative.

By Yuv Raj Pant, Moshiur Rahman Tonmoy, Kaiqun Fu
arXiv AI
Oct 15

Learning with a Single Rollout via Pass@k Policy Optimization

The paper introduces Single‑Rollout Pass@k Policy Optimization (SR‑PPO), a reinforcement learning method that uses only one rollout per prompt to train language‑model agents. SR‑PPO combines Monte Carlo token‑level advantage estimation with Pass@k gradient reweighting and an online critic for baseline and weight estimation. Experiments show that SR‑PPO achieves stable learning and performance comparable to traditional Group Relative Policy Optimization (GRPO) on reasoning and logic benchmarks, even when off‑policy drift degrades standard Pass@1 PPO.

By Fengdi Che, Yang Liu, Lei Yu, Meng Cao, Tong Che, Rupam Mahmood, Dale Schuurmans
arXiv AI
Sep 30

Learn from the Gap: Differential-Aware Advantage Pruning with Adaptive Rollout Sampling for GRPO

The paper introduces FastRL, a reinforcement learning framework designed to enhance the efficiency of Group Relative Policy Optimization (GRPO) and its variants. FastRL employs an advantage-aware pruning strategy that retains high-advantage trajectories while preserving gradient diversity, and an adaptive rollout sampling mechanism that adjusts sampling scale during training based on historical pruning data. Experiments show that FastRL can be integrated into GRPO, DAPO, and GSPO, yielding a 2.07× speedup on Geometry3K and GeoQA8K-R1V and a 1.64% accuracy improvement on visual reasoning benchmarks.

By Jiahua Yang, Zhiwei Yang, Xianpeng Zhang, Dongyu Chen, Xing Chen, Tianhuang Su, Haonan Lu, Quanlong Guan, Kai Tang, Chuangchuang Wang
arXiv Machine Learning
Jul 30

Early Verdicts, Better Budgets: Sequential Adaptive Rollout Allocation for Compute-Efficient RLVR

arXiv:2607. 26253v1 Announce Type: new Abstract: Reinforcement learning with verifiable rewards (RLVR) is bottlenecked by rollout generation, yet many sampled prompts produce saturated groups (all responses correct or all incorrect) whose zero reward variance yields no policy-gradient signal.

By Pixel Nomand, Elena Voss, Marcus Hale, Sofia Reyes
arXiv Machine Learning
1d ago

Optimizing Anytime Reasoning via Budget Relative Policy Optimization

The paper introduces AnytimeReasoner, a framework that optimizes large language models for anytime reasoning by truncating the reasoning process to fit sampled token budgets and requiring the model to summarize the answer for verification. This approach adds verifiable dense rewards, improving credit assignment in reinforcement learning. The authors also propose Budget Relative Policy Optimization (BRPO), a variance reduction technique that enhances learning robustness and efficiency, and demonstrate that their method outperforms GRPO on mathematical reasoning tasks across various budgets and prior distributions.

By Penghui Qi, Zichen Liu, Tianyu Pang, Chao Du, Wee Sun Lee, Min Lin