mlsanityData to 7 Oct 2026

Future-KL Regularized GRPO: Process-Level Credit Assignment from f-Divergence Regularization

arXiv 2601.10201 ↗First posted 15 Jan 2026cs.LG cs.AI cs.CLcited by 11

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

  1. 2026Your Group-Relative Advantage Is Biased13 Jan
  2. 2026GDPO: Group reward-Decoupled Normalization Policy Optimization for Multi-reward RL Optimization8 Jan
  3. 2026Ratio-Variance Regularized Policy Optimization for Efficient LLM Fine-tuning6 Jan
  4. 2025Reinforce-Ada: An Adaptive Sampling Framework under Non-linear RL Objectives6 Oct
  5. 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoning30 May
  6. 2025On the Design of KL-Regularized Policy Gradient Algorithms for LLM Reasoning23 May
  7. 2025Learning Like Humans: Advancing LLM Reasoning Capabilities via Adaptive Difficulty Curriculum Learning and Expert-Guided Self-Reformulation13 May
  8. 2025A Minimalist Approach to LLM Reasoning: from Rejection Sampling to Reinforce15 Apr
  9. 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
  10. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
  11. 2025Process Reinforcement through Implicit Rewards3 Feb
  12. 2025REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalization4 Jan
4 more
  1. 2024Free Process Rewards without Process Labels2 Dec
  2. 2023Let's Verify Step by Step31 May
  3. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
  4. 2017Proximal Policy Optimization Algorithms20 Jul

Other references · 28

  1. 2022Training language models to follow instructions with human feedbackcited by 15,712
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  4. 2025Qwen3 Technical Reportcited by 8,195
  5. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  6. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  7. 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,634
  8. 2015Trust Region Policy Optimizationcited by 2,503
  9. 2022Emergent Abilities of Large Language Modelscited by 2,483
  10. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  11. 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
  12. 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
16 more
  1. 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
  2. 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
  3. 2025Group Sequence Policy Optimizationcited by 788
  4. 2018Diversity is All You Need: Learning Skills without a Reward Functioncited by 734
  5. 2024Training Language Models to Self-Correct via Reinforcement Learningcited by 405
  6. 2023Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-Constraintcited by 346
  7. 2024Large Language Models for Mathematical Reasoning: Progresses and Challengescited by 283
  8. 2024Correcting the Mythos of KL-Regularization: Direct Alignment without Overoptimization via Chi-Squared Preference Optimizationcited by 78
  9. 2025Defeating the Training-Inference Mismatch via FP16cited by 62
  10. 2023Generative Flow Networks as Entropy-Regularized RLcited by 60
  11. 2024Sharp Analysis for KL-Regularized Contextual Bandits and RLHFcited by 32
  12. 2025Optimizing Chain-of-Thought Reasoners via Gradient Variance Minimization in Rejection Sampling and RLcited by 23
  13. 2019Momentum in Reinforcement Learningcited by 19
  14. 2022KL-Entropy-Regularized RL with a Generative Model is Minimax Optimalcited by 13
  15. 2024Entropy-Regularized Token-Level Policy Optimization for Language Agent Reinforcementcited by 12
  16. 2022Entropy Regularized Reinforcement Learning with Cascading Networkscited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 11.

11 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Rethinking KL Regularization in RLHF: From Value Estimation to Gradient Optimization
  2. 2026On KL-Regularized Policy Optimization
  3. 2026Why GRPO Needs Normalization: A Local-Curvature Perspective on Adaptive Gradients
  4. 2025Reinforcement Learning with Verifiable Rewards: GRPO's Effective Loss, Dynamics, and Success Amplification
  5. 2025TIC-GRPO: Provable and Efficient Optimization for Reinforcement Learning from Human Feedback
  6. 2025Group-Relative REINFORCE Is Secretly an Off-Policy Algorithm: Demystifying Some Myths About GRPO and Its Friends
  7. 2026Constrained Group Relative Policy Optimization
  8. 2025GRPO-RM: Fine-Tuning Representation Models via GRPO-Driven Reinforcement Learning