English

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

Machine Learning 2026-05-26 v2 Artificial Intelligence Computation and Language

Abstract

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 2arcsinp2\arcsin\sqrt 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 ff-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.

Keywords

Cite

@article{arxiv.2601.10201,
  title  = {Future-KL Regularized GRPO: Process-Level Credit Assignment from $f$-Divergence Regularization},
  author = {Jiarui Yao and Ruida Wang and Hao Bai and Tong Zhang},
  journal= {arXiv preprint arXiv:2601.10201},
  year   = {2026}
}
R2 v1 2026-07-01T09:05:31.291Z