arXiv AI By Faeze Moradi Kalarde, Armin Toroghi, Nam Ho Koh, Doo Hwan Hwang, Seungjin Kim, Kee-Eung Kim, Scott Sanner

Bayes-Adaptive MDPs for Rollout Allocation to Enhance GRPO Training

Read the original on arXiv AI →

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.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

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