← All papers
First page of Rewarding the Unlikely: Lifting GRPO Beyond Distribution Sharpening

Rewarding the Unlikely: Lifting GRPO Beyond Distribution Sharpening

Andre He, Daniel Fried, Sean Welleck

cs.LG Jun 3, 2025 · v2
Trains LLM provers with GRPO using Lean 4 REPL verification as reward, on Lean Workbook and miniF2F theorem statements.
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

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.

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.

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.

ModelSolvedΔ
Static (V1.5-SFT)7707 / 9600–
GRPO-Default7860 / 9600+153
GRPO-Epochs-28008 / 9600+301
GRPO-Unlikeliness-18023 / 9600+316
GRPO-Unlikeliness-28065 / 9600+358
Training problems solved during one epoch on D_train