Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

A Gradient Perspective on RLVR Stability and Winner Advantage Policy Optimization

About

Reinforcement learning with verifiable rewards (RLVR) improves language-model reasoning, but GRPO-style optimization remains prone to collapse. We analyse this instability through token-level gradient dynamics, deriving a taxonomy that predicts how updates affect next-token probabilities and entropy. The taxonomy shows that stability depends jointly on the advantage sign and token distribution under the current policy. Motivated by this finding, we propose Winner Advantage Policy Optimization (WAPO), a simple online clipped policy-gradient objective that updates only on positive-advantage completions. Across mathematical reasoning and multi-hop QA benchmarks, WAPO improves training stability and matches or outperforms baselines across multiple model families. Full code can be found at https://github.com/layer6ai-labs/wapo.

Prasanth YSS, Zhichen Ren, Rasa Hosseinzadeh, Ilan Gofman, Yuqi Chen, Zhaoyan Liu, Guangwei Yu, Jesse C. Cresswell, Satya Krishna Gorti• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningAIME25 (test)
Pass@136.4
45
Multi-hop Question Answering2-wiki out-of-domain transfer
Exact Match34.2
12
Mathematical ReasoningMATH-500 In-Distribution (test)
Average@3278.9
12
Mathematical ReasoningNuminaMath in-distribution LEAN (test)
Average@3261.99
12
Mathematical ReasoningAIME out-of-domain evaluation 2025
Average Score @3238.43
12
Question AnsweringOTT-QA in-distribution (test)
Average@3237.44
11
Showing 6 of 6 rows

Other info

GitHub

Follow for update