English

Multi-Step Likelihood-Ratio Correction for Reinforcement Learning with Verifiable Rewards

Machine Learning 2026-05-21 v1 Artificial Intelligence

Abstract

Reinforcement learning with verifiable rewards (RLVR) plays a pivotal role in improving the reasoning ability of large language models. However, widely used PPO surrogate objectives are fundamentally local, as they rely on a local approximation of the exact policy gradient objective. While this approximation improves stability by reducing the variance induced by importance sampling, it also introduces structural bias into the surrogate objective, which must be controlled through trust region mechanisms. In this work, we introduce the NN-step forward trace, which augments the PPO surrogate objective using the cumulative likelihood ratio of the next N1N-1 tokens. Building on this idea, we propose NN-Step Forward-Trace Policy Optimization (NFPO), a practical RLVR algorithm that integrates the NN-step forward trace into the masked policy gradient framework. NFPO provides a continuous bridge between the PPO surrogate objective and the exact policy gradient objective, offering a principled mechanism for controlling the bias-variance trade-off. Our theoretical analysis shows that, with an appropriate choice of NN, the proposed objective yields a tighter policy-improvement bound than the standard PPO surrogate. Experiments on comprehensive reasoning benchmarks demonstrate that NFPO consistently improves performance, supporting our theoretical findings.

Keywords

Cite

@article{arxiv.2605.20865,
  title  = {Multi-Step Likelihood-Ratio Correction for Reinforcement Learning with Verifiable Rewards},
  author = {Deokgyu Yoon and Hyungkyu Kang and Joongkyu Lee and Byeongchan Kim and Gyungin Shin and Sungrae Park and Min-hwan Oh},
  journal= {arXiv preprint arXiv:2605.20865},
  year   = {2026}
}
R2 v1 2026-07-22T07:23:27.899Z