Eliminating Off-Policy Bias in Asynchronous RLVR: Proximal KL Enables Token-Wise Importance Sampling, Beating GRPO on Qwen3 with Zero Training Collapse
Off-policy data in reinforcement learning with verifiable rewards (RLVR) causes severe policy gradient bias during asynchronous generation and response reuse. Researchers from UIUC and HKUST introduce Proximal KL, theoretically proving that full future KL credit renders the exact proximal optimizer the unique stationary policy without requiring full-response weights or fitted critics. Evaluated across Qwen3-1.7B and 4B models, Proximal KL outperforms GRPO on 8 math benchmarks by up to 2.58% while completely averting training collapse.
Key Takeaways
- Theoretical breakthrough for off-policy gradient bias: proves full future KL credit guarantees a unique stationary policy, decoupling reference and sampling policies without sequence-level variance explosion
- Decisive outperformance over GRPO: yields up to +2.58% micro-averaged accuracy gains across 8 mathematics benchmarks on Qwen3-1.7B and 4B, sharply improving AIME24 pass@16
- Eliminates training collapses during off-policy refresh: permanently avoids catastrophic policy degradation seen in standard PG and CISPO, reaching peak accuracy with fewer update steps