wikillm

Concept · 39 papers · updated 2026-11-01

Policy gradient

Training a policy by raising the probability of sampled actions in proportion to a return or advantage, so the reward need not be differentiable. Variants differ in the baseline: a learned critic with GAE, a group of rollouts per prompt, a running mean, a greedy rollout, or none. Costs: extra rollouts, a critic, or a reward model queried at every step.

reinforcement-learningoptimizationtext-generation

Markdown19 min read

AI-compiled overview, updated 2026-11-01, not yet reviewed by a human. Every claim cites its source; check it when a detail matters.

Overview

A policy-gradient method samples an action or a whole output from the current policy, scores it, and raises the log-probability of that output in proportion to its weight 1609.05473 eq. 6. The reward enters only as a multiplier, so it need not be differentiable. The cards use discriminator scores 1701.06547 eq. 1, simulator returns 2108.10470 §6.1.1, exact-match verifiers 2503.16219 §2, solver execution results 2505.11792 eq. 6, learned video-text similarity 2206.08853 §4.2 and ROUGE-L overlap with approved headlines 2607.06818 §4.1.

The estimator is noisy, so every variant subtracts a typical value from the return, or normalises it, before the update. Variants differ in four places:

  1. The baseline or advantage. A learned value function [arxiv:2102.02202 app. A], the mean over a group of rollouts for the same prompt 2503.01785 eq. 4, a running average of past returns 1808.09637 eq. 1, a greedy rollout 2012.10638 eq. 16–18, a per-token KL term 2605.07865 eq. 10, or nothing sequence_gan.py.
  2. What counts as an action. A token 1609.05473 eq. 6, a denoising step of a diffusion policy 2409.00588 app. C.10, an act-level decision above a word generator 1808.09637 §3.4, a guide chosen per turn 2508.04451 eq. 1, a candidate architecture 1812.00332 eq. 8, or the parameters of a masked-diffusion language model 2507.08838 §3.1.
  3. Stabilisers. A probability-ratio clip 2102.02202 tab. 2, a KL penalty to a reference model 2505.11792 tab. 6, an entropy term 2102.02202 tab. 2, or a replay buffer of past successes 2510.02245 §4.2.
  4. Where the rollouts come from. The real environment, a world model 2301.04104 tab. 4, a cached tool sandbox 2509.21842 tab. 7, or a teacher's per-token log-probabilities 2605.07865 §3.1.

Four cards are not method sources for this page. Two are surveys or position papers that discuss the method without an experiment 2003.01200 §3.3 2105.01767 §5.1. Two are simulation frameworks that use PPO as a tool 2301.04195 §5.1 2108.10470 §4. One is a textbook chapter 2610.17391 app. B. Their content is used only for the throughput and training-setup notes in the Variants section.

Variants

Learned value baseline with GAE (PPO-style)

The policy is updated with a clipped surrogate and a learned value function that supplies the advantage through generalised advantage estimation. This is the most common setting in robot learning and simulation. Settings from the cards:

Group-relative baseline without a critic (GRPO family)

For each prompt, sample a group of G outputs, score them, and normalise the rewards inside the group. No value network is trained. The advantage is the group-normalised reward 2503.01785 eq. 4 2504.06958 eq. 1. This is the main variant for language and vision-language models with a verifiable reward.

Single rollout per prompt

Some agentic tasks cannot afford a group of long episodes. SAO uses one rollout per prompt and corrects for the policy lag with the rollout engine's own log-probabilities 2607.07508 §3.1. It updates a critic twice per batch, with attention frozen 2607.07508 §3.2. On AIME2025 it reports 97.3, against 84.2 for GRPO with a Python tool and 93.5 for GRPO with DIS 2607.07508 tab. 1. Its ablation reports 95.0 with one critic update per batch 2607.07508 tab. 3. The value model needs a large pretraining corpus 2607.07508 §3.2, and throughput and memory were not measured 2607.07508 §1. The results come from one training seed [arxiv:2607.07508 §4.1]. Card: Single-Rollout Asynchronous Optimization for Agentic Reinforcement Learning.

Running mean, greedy rollout or teacher-based baselines

No baseline (plain REINFORCE on sampled returns)

Variance-reduced, Bayesian and theory variants

Not a score-function method

The DDAC actor for lane keeping takes the gradient of the critic with respect to the action and multiplies it by the actor's gradient 1704.02532 §3.2. It is a deterministic actor-critic, not a policy gradient in the sense used here. The paper reports no metric values 1704.02532 §5. Card: Deep Reinforcement Learning framework for Autonomous Driving.

Model-based value expansion is also a deterministic actor-critic, built on the deterministic policy gradient 1803.00101 §2.1 1803.00101 §2.2. It rolls a learned one-step model forward H steps and bootstraps from the critic 1803.00101 eq. 2. Its horizon is set by hand per task: H = 10 on cheetah and swimmer, H = 3 on walker 1803.00101 fig. 3 caption. The results are learning curves on three MuJoCo tasks with 4 seeds for the learned model 1803.00101 §4 1803.00101 app. A. The paper does not state a discount. Card: Model-Based Value Estimation for Efficient Model-Free Reinforcement Learning.

When to use which

Choose by what you can afford per update and what the reward looks like.

SituationUseWhy
Dense reward, many cheap parallel simulator environmentsPPO with GAE and a learned valueIsaac Gym uses 4096 to 16384 environments 2108.10470 tab. 17
Verifiable answer, no room for a critic in memoryGRPO-style group baselineG = 6 ran in 24 hours on four GPUs 2503.16219 §3
One long episode costs as much as a groupSingle rollout with a critic97.3 vs 84.2 on AIME2025 in one comparison 2607.07508 tab. 1, one seed
Real environment is slow and a world model is feasibleLearned-model imagination8.9 A100 GPU-days for 100 million steps 2301.04104 tab. 2
Per-step learned reward must be queriedBudget for the reward modelAbout 150 million parameters, queried at every step 2206.08853 §4.2
Discrete construction with a cheap greedy rolloutREINFORCE with a greedy baselineTSP100 gap 0.38% at batch 50 2012.10638 tab. 1
Pretrained diffusion policy to fine-tune onlineDiffusion PPOK network evaluations per environment step 2409.00588 app. D
Teacher with per-token log-probabilitiesControl-variate on-policy distillationBaseline without a critic 2605.07865 §3.1
Diffusion language model, no likelihood ratiowd1-style softmax weights1280 vs 30000 rollouts to best model 2507.08838 tab. 8
Reward is a discriminator on long sequencesHierarchical goal (LeakGAN) or plain Monte Carlo (SeqGAN)LeakGAN NLL 7.191 vs 10.310 at length 40 1709.08624 tab. 1
Need a convergence boundTheory onlyDEF-PG has no experiments 2610.03165 §7
Masked or non-autoregressive text generator with a cheap greedy decode and a metric rewardSelf-critical baseline from the greedy outputROUGE-L 6.33 vs 5.08 without the RL step, author-reported 2607.06818 tab. 1; proprietary data
Learned model is only accurate for a few stepsShort-horizon value expansion with a deterministic actorH = 3 needed on walker, H = 10 on cheetah and swimmer 1803.00101 fig. 3 caption
Thinking tokens dominate inference costGRPO with skip-count and length terms in the rewardWithout them, MATH500 accuracy is 51.00 against 85.00 2508.03346 tab. 5

Two cautions apply across rows. Group and single-rollout results come from one training seed or a few seeds, and several cards flag that [arxiv:2607.07508 §4.1] 2503.16219 §3. Most headline numbers in these cards are author-reported, and the comparison baselines often differ in compute.

Implementation sketch

The loop below covers the common core. The lines that change between variants are the advantage and the optional clip and KL terms.

Pseudocode
# Sample a group of G outputs for each prompt [arxiv:2503.01785 eq. 4]
outputs = [policy.sample(prompt) for _ in range(G)]
# Score each output with the reward: verifier, judge or learned model
rewards = [reward(prompt, o) for o in outputs]
# Choose the advantage: group mean, critic with GAE, running mean or none
adv = advantage(rewards, outputs)  # see Variants
# Optionally divide by the standard deviation [arxiv:2504.06958 eq. 1]
adv = adv / (std(rewards) + 1e-4)
# Score the sampled tokens under the current policy
logp = policy.logprob(outputs)
# Add a KL penalty to a reference model if the variant uses one
loss = -mean(adv * logp) + beta * kl(policy, ref)
# PPO-style variants clip the probability ratio here [arxiv:2102.02202 tab. 2]

Contradictions

No two cards report incompatible results on the same comparison. Several pairs differ in setting, so the cards do not test each other's claims.

  • KL to a reference. SIRL finds that a partial KL term helps, 51.7 against 47.3 2505.11792 tab. 3. Conductor trains with KL weight 0 2512.04388 app. A. The two cards use different tasks and different reward designs, so neither result tests the other.
  • Critic against group baseline. SAO reports that a single rollout with a critic beats GRPO on AIME2025 2607.07508 tab. 1. The group-relative cards use no critic and do not run this comparison 2503.16219 §3 2512.04388 app. A.
  • Within-paper conflicts. Several cards show conflicts inside one paper, and the page does not resolve them. These include the DPPO batch-size tables 2409.00588 app. C.10, the DreamerV3 horizon [arxiv:2301.04104 tab. 4], and the wd1 Sudoku gain in the text and table 2507.08838 §5.1.

Papers

Source and license

Compiled from the papers listed on this page, all published under CC BY, CC BY-SA or CC0, with citations. See each paper's page for its authors and license.