A First-Principles Derivation of LLM Policy Optimization: From Expected Reward to GRPO and Its Structural Extensions
2026-06-15 • Artificial Intelligence
Artificial Intelligence
AI summaryⓘ
The authors review various algorithms used to improve language models by tweaking how they learn from rewards, focusing on two main parts: the chances of certain actions happening (trajectory probability) and the rewards those actions get. Instead of grouping methods by area or time, they organize them based on which part each method changes and why. This new framework helps identify problems that can’t be fixed by adjusting only one part, showing that both the trajectory and reward aspects need to be designed together. Their work offers a clear way to understand past algorithms and guide the creation of better ones in the future.
policy gradientlanguage modelsREINFORCEPPOGRPOtrajectory probabilityreward functionpolicy optimizationreinforcement learninggradient estimator
Authors
Jianghan Shen, Siqi Luo, Yue Li, Jiyao Liu, Wanying Qu, Yi Zhang, Ziyan Huang, Tianbin Li, Ming Hu, Xiaohong Liu, Yirong Chen, Junjun He
Abstract
Policy gradient algorithms for language models optimize the same objective $J(θ) = \mathbb{E}*{τ\sim p*θ(τ)}[R(τ)]$, which has exactly two factors: the trajectory probability $p_θ(τ)$ and the reward $R(τ)$. Every method from REINFORCE to PPO to GRPO and their descendants modifies one or both factors to address a specific failure in the preceding formulation. Existing surveys organize these methods by domain or chronology, which obscures the rationale behind each design choice and the precise location of its intervention within the gradient estimator. This survey revisits the landscape of LLM policy optimization from $J(θ)$ on first principles and uses the trajectory side, induced by $p_θ(τ)$, and the reward side, induced by $R(τ)$, as the two axes along which methods are located. It covers the path from REINFORCE and PPO to GRPO, as well as post-GRPO variants, Agentic RL, and GRPO-OPD. The resulting framework is unified, diagnostic, and extensible: it analyzes methods from a shared objective, identifies which side each method modifies and why, and applies the same trajectory and reward axes across these settings. Across these settings, the framework also exposes compound failures that no single-side fix resolves and that therefore require joint design of the trajectory side and the reward side. The boundary cases and coupled failures identified by this map mark where existing solutions run out and provide a principled starting point for designing the next generation of LLM policy optimization algorithms.