Future-KL Regularized GRPO: Process-Level Credit Assignment from f-Divergence Regularization
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Group Relative Policy Optimization (GRPO) is widely used for critic-free Large Language Model (LLM) post-training, but its KL regularization is usually implemented as a local loss-side token penalty. We show that this misses the policy-gradient signal induced by autoregressive KL regularization. Unlike standard KL-regularized Reinforcement Learning (RL) objectives, GRPO's group normalization induces a non-linear prompt-level utility; for binary verifier rewards, this utility is 2arcsin√ p. As a result, reward and KL cannot be fused before normalization without changing the implicit objective. We derive the on-policy gradient of GRPO-style objectives with token-wise f-divergence regularization. The reward term recovers the standardized GRPO advantage, while the regularizer term includes a causal future-regularization return-to-go omitted by local KL losses. For reverse KL, this yields a simple future KL correction: add a reverse cumulative sum of per-token log ratios after advantage construction. The resulting method, Future-KL Regularized Policy Optimization (FRPO), requires no critic or extra model passes. On mathematical reasoning tasks, FRPO improves pass@16 in our main large-model setting while maintaining higher entropy and lower policy drift than conventional loss-side KL baselines.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 44 of its 53 references that are in the map. First posted on arXiv 15 Jan 2026.
Builds on
44 of 53 references.
Closest ancestors · 16
Same direction, newest first
- 2026Your Group-Relative Advantage Is Biased13 Jan
- 2026GDPO: Group reward-Decoupled Normalization Policy Optimization for Multi-reward RL Optimization8 Jan
- 2026Ratio-Variance Regularized Policy Optimization for Efficient LLM Fine-tuning6 Jan
- 2025Reinforce-Ada: An Adaptive Sampling Framework under Non-linear RL Objectives6 Oct
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoning30 May
- 2025On the Design of KL-Regularized Policy Gradient Algorithms for LLM Reasoning23 May
- 2025Learning Like Humans: Advancing LLM Reasoning Capabilities via Adaptive Difficulty Curriculum Learning and Expert-Guided Self-Reformulation13 May
- 2025A Minimalist Approach to LLM Reasoning: from Rejection Sampling to Reinforce15 Apr
- 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025Process Reinforcement through Implicit Rewards3 Feb
- 2025REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalization4 Jan
4 more
Other references · 28
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,634
- 2015Trust Region Policy Optimizationcited by 2,503
- 2022Emergent Abilities of Large Language Modelscited by 2,483
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
16 more
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
- 2025Group Sequence Policy Optimizationcited by 788
- 2018Diversity is All You Need: Learning Skills without a Reward Functioncited by 734
- 2024Training Language Models to Self-Correct via Reinforcement Learningcited by 405
- 2023Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-Constraintcited by 346
- 2024Large Language Models for Mathematical Reasoning: Progresses and Challengescited by 283
- 2024Correcting the Mythos of KL-Regularization: Direct Alignment without Overoptimization via Chi-Squared Preference Optimizationcited by 78
- 2025Defeating the Training-Inference Mismatch via FP16cited by 62
- 2023Generative Flow Networks as Entropy-Regularized RLcited by 60
- 2024Sharp Analysis for KL-Regularized Contextual Bandits and RLHFcited by 32
- 2025Optimizing Chain-of-Thought Reasoners via Gradient Variance Minimization in Rejection Sampling and RLcited by 23
- 2019Momentum in Reinforcement Learningcited by 19
- 2022KL-Entropy-Regularized RL with a Generative Model is Minimax Optimalcited by 13
- 2024Entropy-Regularized Token-Level Policy Optimization for Language Agent Reinforcementcited by 12
- 2022Entropy Regularized Reinforcement Learning with Cascading Networkscited by 2
Built on by
Citing papers per year: 2026: 11.
11 papers.
- 2026From Reasoning to Agentic: Credit Assignment in Reinforcement Learning for Large Language Modelscited by 19
- 2026STRIDE: Strategic Trajectory Reasoning via Discriminative Estimation for Verifiable Reinforcement Learningcited by 9
- 2026Rethinking Reward Supervision: Rubric-Conditioned Self-Distillationcited by 9
- 2026TikArt: Stabilizing Aperture-Guided Fine-Grained Visual Reasoning with Reinforcement Learningcited by 4
- 2026Beyond Alignment: Expanding Reasoning Capacity via Manifold-Reshaping Policy Optimizationcited by 3
- 2026LLM Reasoning with Process Rewards for Outcome-Guided Stepscited by 3
- 2026Which Tokens Matter? Adaptive Token Selection for RLVR with the Relative Surprisal Indexcited by 3
- 2026Unsupervised Process Reward Modelscited by 0
- 2026SCOPE-RL: Optimizing Reasoning Paths Before and After Successcited by 0
- 2026Beyond Generation and Accuracy: Diagnosing and Enhancing Visual Chain-of-Thought for Geometry Problem Solvingcited by 0
- 2026Advantage Scale Calibration Imbalance in Group-Relative Optimization under Low-Variance Rewards: Diagnosis and Bounded Recoverycited by 0
- 2026Beyond Generation and Accuracy: Diagnosing and Enhancing Visual Chain-of-Thought for Geometry Problem Solving11 Sep 2026
- 2026Advantage Scale Calibration Imbalance in Group-Relative Optimization under Low-Variance Rewards: Diagnosis and Bounded Recovery27 Aug 2026
- 2026SCOPE-RL: Optimizing Reasoning Paths Before and After Success13 Jul 2026
- 2026Which Tokens Matter? Adaptive Token Selection for RLVR with the Relative Surprisal Index30 Jun 2026
- 2026Rethinking Reward Supervision: Rubric-Conditioned Self-Distillation17 Jun 2026
- 2026STRIDE: Strategic Trajectory Reasoning via Discriminative Estimation for Verifiable Reinforcement Learning14 Jun 2026
- 2026Unsupervised Process Reward Models11 May 2026
- 2026From Reasoning to Agentic: Credit Assignment in Reinforcement Learning for Large Language Models10 Apr 2026
- 2026TikArt: Stabilizing Aperture-Guided Fine-Grained Visual Reasoning with Reinforcement Learning16 Feb 2026
- 2026LLM Reasoning with Process Rewards for Outcome-Guided Steps8 Feb 2026
- 2026Beyond Alignment: Expanding Reasoning Capacity via Manifold-Reshaping Policy Optimization30 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2025Rethinking KL Regularization in RLHF: From Value Estimation to Gradient Optimization
- 2026On KL-Regularized Policy Optimization
- 2026Why GRPO Needs Normalization: A Local-Curvature Perspective on Adaptive Gradients
- 2025Reinforcement Learning with Verifiable Rewards: GRPO's Effective Loss, Dynamics, and Success Amplification
- 2025TIC-GRPO: Provable and Efficient Optimization for Reinforcement Learning from Human Feedback
- 2025Group-Relative REINFORCE Is Secretly an Off-Policy Algorithm: Demystifying Some Myths About GRPO and Its Friends
- 2026Constrained Group Relative Policy Optimization
- 2025GRPO-RM: Fine-Tuning Representation Models via GRPO-Driven Reinforcement Learning