Rewarding the Unlikely: Lifting GRPO Beyond Distribution Sharpening
Andre He, Daniel Fried, Sean Welleck
cs.LG
Jun 3, 2025 · v2
TL;DR
Trains LLM provers with GRPO using Lean 4 REPL verification as reward, on Lean Workbook and miniF2F theorem statements.
Abstract
Reinforcement learning is emerging as a primary driver for improving language model reasoning capabilities. A fundamental question is whether current reinforcement learning algorithms – such as Group Relative Policy Optimization (GRPO), the de facto standard algorithm used to improve language model reasoning – merely sharpen the base model's distribution around problems it can already solve. We investigate this question in the context of formal theorem proving, which has access to a perfect verifier. We identify a degenerate rank bias in GRPO in which highly probable trajectories are reinforced and rare ones are neglected. This results in distribution sharpening: the model can solve some problems with fewer samples, but underperforms simply sampling more solutions from the original model. To overcome GRPO's rank bias we introduce unlikeliness reward, a simple method for explicitly up-weighting rare but correct solutions. We show that unlikeliness reward mitigates rank bias and improves pass@$N$ across a large range of $N$ in both synthetic and real theorem proving settings. We also uncover an unexpected link between rank bias and a seemingly mundane hyperparameter – the number of updates per batch – that leads to a second, complementary mitigation. We combine our insights into a revised GRPO training recipe for formal theorem proving, yielding an open pipeline that achieves competitive performance to DeepSeek-Prover-V1.5-RL on the miniF2F-test benchmark. We release our implementation at
https://github.com/AndreHe02/rewarding-unlikely-release
Problem
GRPO, the standard RL algorithm for LLM reasoning, tends to sharpen the base model's distribution. It improves pass@1 but fails to improve, or even harms, pass@N at large N in formal theorem proving.
Approach
The authors identify a rank bias in GRPO: high-probability correct trajectories are reinforced while rare correct ones are neglected. They introduce an unlikeliness reward that down-weights high-probability correct samples within a group. They also show that increasing the number of PPO epochs per batch mitigates the bias. Training uses Lean 4 REPL verifier rewards on a curated Lean Workbook subset plus miniF2F-valid.
Results
Both the unlikeliness reward and extra PPO epochs improve pass@N at large N on a held-out validation set. GRPO-Unlikeliness-2 solved the most training problems (8065/9600 versus 7707 for the static model). The revised recipe yields an open pipeline competitive with DeepSeek-Prover-V1.5-RL on miniF2F-test.
| Model | Solved | Δ |
|---|
| Static (V1.5-SFT) | 7707 / 9600 | – |
| GRPO-Default | 7860 / 9600 | +153 |
| GRPO-Epochs-2 | 8008 / 9600 | +301 |
| GRPO-Unlikeliness-1 | 8023 / 9600 | +316 |
| GRPO-Unlikeliness-2 | 8065 / 9600 | +358 |
Training problems solved during one epoch on D_train