Reinforcement learning with rule-based, binary-valued reward functions has been shown to significantly enhance the reasoning capabilities of large language models. These models typically rely on REINFORCE-based policy optimization methods, such as REINFORCE with baseline and group relative policy optimization (GRPO). However, a key limitation is that existing methods either ignore reward normalization or use static normalization schemes, which fail to adapt to the evolving policy during training. This mismatch can lead to unstable gradient estimates and hinder training stability. To address this issue, we propose Beta Normalization Policy Optimization (BNPO), a novel method that adaptively normalizes rewards using a Beta distribution with dynamically updated parameters. By aligning normalization with the changing policy distribution, BNPO enables more accurate, lower-variance gradient estimation and promotes stable training dynamics. We provide theoretical analysis showing that BNPO reduces variance and generalizes both REINFORCE and GRPO under binary-valued rewards. In addition, we introduce an advantage decomposition mechanism to extend BNPO to more complex reward settings. Experimental results demonstrate that BNPO achieves state-of-the-art performance among policy optimization methods on reasoning tasks.
Figures & tables
Figure 1: Probability density function of Beta distribution. The values are the coefficients (α,β) .
Methods
MATH500
AMC23
AIME2024
AIME2025
Average
Qwen2.5 - Math - 1.5B
Base
28.0
27.3
6.0
3.1
16.1
REINFORCE
72.2
53.6
18.3
11.5
38.9
ReMax
73.2
53.3
17.1
9.6
38.3
GRPO
75.0
52.0
15.6
11.0
38.4
REINFORCE++
73.8
52.0
16.7
9.8
38.1
Table 1: The performance of different policy optimization methods on math datasets.
Step
100
200
300
400
500
600
700
800
900
1000
1100
REINFORCE
354
233
127
143
162
118
133
140
196
180
256
REMAX
702
475
385
269
251
108
130
172
200
404
423
GRPO
2652
1552
1183
505
552
673
537
610
807
606
862
REINFORCE++
764
1455
1407
612
741
722
1402
739
1060
936
1146
BNPO
235
243
71
123
133
71
115
102
156
131
163
Table 2: Gradient variance at different training steps.
Figure 2: The norm of gradient during training.
Figure 3: The values of (E[p(q)],Var[p(q)],α,β) during training.
Methods
Average
Qwen2.5-1.5B-Instruct
Base
5.8
GRPO
25.2
REINFORCE++
24.6
AD-GRPO
25.6
AD-REINFORCE++
24.9
Table 3: The performance of policy optimization methods with and without advantage decomposition.
Method
Time
REINFORCE
23.9h
REINFORCE++
23.9h
GRPO
24.2h
ReMax
30.4h
BNPO
23.9h
Table 4: Training time comparison of different methods.
On-policy reinforcement learning has become a central paradigm for improving the reasoning abilities of large language models. However, its effectiveness is often limited by reward sparsity: when a model fails to discover correct trajectories for difficult problems, the optimization process receives little useful signal and may stagnate. Existing approaches mitigate this issue by incorporating off-policy demonstrations, expert traces, or model-generated solutions, but they typically require the auxiliary data to match the format of the reinforcement-learning task, often relying on rejection sampling from stronger models to obtain suitable training trajectories. We introduce Rationale-Guided Policy Optimization (RGPO), a framework that adaptively leverages ground-truth rationale information according to the model's current capability while preserving its freedom to explore. Rather than treating reference solutions as fixed imitation targets, RGPO uses them as temporary scaffolds: rationales help the model generate improved responses, after which only higher-reward, model-generated solutions are transferred back to the original unguided setting. This design allows training to exploit available ground-truth information without requiring off-policy data to follow the same format as the RL task. Across both language-only and vision-language reasoning settings, RGPO consistently improves performance over RLVR baselines, and ablation studies show that adaptive rationale guidance is a key contributor to these gains. These results suggest that RGPO offers a practical and general approach for reducing reward sparsity, stabilizing reinforcement learning, and improving reasoning performance in both text-only and multimodal models.
Hoang Phan, Minh Pham, Chau Pham +3
New York University · University at Buffalo · Monash University
Reinforcement learning with verifiable rewards (RLVR) improves the reasoning capabilities of large language models (LLMs). We introduce Bellman Policy Optimization (BPO), a critic-free method derived from Policy Mirror Descent (PMD). For autoregressive generation with terminal rewards, BPO uses the Bellman equations to reformulate PMD as a trajectory-level objective. The reformulation avoids estimating state values at intermediate states. We prove that it has the same unique optimal solution as the original PMD objective. We derive the practical BPO loss by approximating this objective. Its mismatch-correction weight is a smoothed ratio of complementary token probabilities. Experiments on mathematical reasoning benchmarks demonstrate the effectiveness of BPO.
Reinforcement learning with verifiable rewards (RLVR) has emerged as a powerful paradigm for eliciting long-chain reasoning in large language models. However, existing methods based on Group Relative Policy Optimization (GRPO) rely on a binary outcome reward, which induces two structural failure modes: Zero-Advantage Collapse, in which all rollouts in a group share the same outcome and the gradient vanishes, and Hallucinated Certainty, in which the model becomes increasingly confident on incorrect rollouts late in training. We address both modes by densifying the reward with intrinsic signals computed entirely from the policy's own conditional probabilities, and propose ISPO (Intrinsic Signal Policy Optimization, which combines a sequence-level signal measuring how informative the thinking trajectory is for the final answer, with a token-level directional reward whose hallucinated-certainty hinge penalizes confidently-wrong predictions at critical decision tokens. Across three base models and five mathematical reasoning benchmarks, ISPO consistently outperforms competitive baselines, with the largest gains on the hardest benchmarks where zero-advantage collapse is most frequent, and training-dynamics diagnostics confirm that both failure modes are decreased.
Hao Chen, Zhanming Shen, Liyao Li +8
Zhejiang University · The Chinese University of Hong Kong · Eastern Institute of Technology