Source-linked AI summary

On the Direction of RLVR Updates for LLM Reasoning: Identification and Exploitation

Kexin Huang, Haoming Meng, Junkang Wu, Jinda Lu, Chiyu Ma, Ziqian Chen, Xue Wang, Bolin Ding, Jiancan Wu, Xiang Wang, Xiangnan He, Guoyin Wang, Jingren Zhou

arXiv:2603.22117v1cs.LGcs.AI

TL;DR

Prior analyses of RLVR-induced changes emphasize their magnitude, leaving update direction comparatively underexamined. This paper studies signed token-level log-probability changes and develops test-time and training-time methods based on them, reporting improved reasoning performance across settings.

  • Problem

    Prior analyses find RLVR-induced changes sparse but primarily emphasize their magnitude, leaving their direction comparatively understudied.

    Source: §1 · p. 1
  • Method

    The paper analyzes signed token-level Δlog p and develops test-time extrapolation and training-time reweighting focused on low-probability tokens.

    Source: §1 · p. 2
  • Results

    Both methods improve reasoning performance across different settings, and Δlog p identifies reasoning-critical updates more effectively than magnitude-based metrics.

    Source: §6 · p. 10
  • Takeaways & Limitations

    The findings support update direction as a principle for analyzing and improving RLVR.

    Source: §6 · p. 10

    The extrapolation method requires two models and introduces additional hyperparameters.

    Source: §6 · p. 11

Abstract

from arXiv · show

Reinforcement learning with verifiable rewards (RLVR) has substantially improved the reasoning capabilities of large language models. While existing analyses identify that RLVR-induced changes are sparse, they primarily focus on the \textbf{magnitude} of these updates, largely overlooking their \textbf{direction}. In this work, we argue that the direction of updates is a more critical lens for understanding RLVR's effects, which can be captured by the signed, token-level log probability difference $Δ\log p$ between the base and final RLVR models. Through statistical analysis and token-replacement interventions, we demonstrate that $Δ\log p$ more effectively identifies sparse, yet reasoning-critical updates than magnitude-based metrics (\eg divergence or entropy). Building on this insight, we propose two practical applications: (1) a \textit{test-time extrapolation} method that amplifies the policy along the learned $Δ\log p$ direction to improve reasoning accuracy without further training; (2) a \textit{training-time reweighting} method that focuses learning on low-probability (corresponding to higher $Δ\log p$) tokens, which improves reasoning performance across models and benchmarks. Our work establishes the direction of change as a key principle for analyzing and improving RLVR.

1 Introduction

The paper argues that RLVR’s update direction, captured by signed token-level Δlog p, better reveals reasoning-critical changes than update magnitude alone. It uses this diagnostic to motivate test-time extrapolation and training-time reweighting for reasoning enhancement.

1 Introduction

  • Prior analyses found sparse RLVR changes but emphasized magnitude; base and final-model histograms for magnitude metrics are nearly identical, limiting their ability to characterize the transformation.The introduction cites high-entropy tokens, KL divergence, and selective gradient updates as prior accounts of sparsity.Source: §1 · p. 1, Figure 1 · p. 2
  • Signed token-level Δlog p records probability increases and decreases, reveals a directional signature absent from magnitude metrics, and reaches RLVR-level performance with the fewest token substitutions.The Δlog p histograms are bimodal, while magnitude-based metrics show nearly identical base and RLVR distributions.Source: §1 · pp. 1–2, Figure 1 · p. 2
  • Test-time extrapolation selectively amplifies the RLVR policy along the Δlog p direction for reasoning-critical tokens, improving accuracy without additional training.The method extrapolates the learned distributional updates rather than requiring further model training.Source: §1 · p. 2
  • Training-time reweighting upweights policy-gradient contributions from low-probability tokens, which correspond to the largest Δlog p, and yields gains over methods such as DAPO across models and benchmarks.The reweighting focuses learning on positions identified as reasoning-critical by Δlog p.Source: §1 · p. 2

2 Preliminaries

The section sets out critic-free policy-optimization preliminaries and introduces token-level metrics for analyzing changes between base and RL-finetuned policies.

  • GRPO (Shao et al., 2024), a critic-free PPO variant, estimates advantages across G sampled responses per QA pair and optimizes a clipped policy-ratio objective regularized toward a reference policy.Source: §2 · pp. 2–3
  • DAPO (Yu et al., 2025) refines GRPO with clip-higher, dynamic sampling, token-level loss aggregation, overlong punishment, and removal of the KL penalty; it is the primary empirical baseline.Source: §2 · p. 3
  • Entropy is calculated for both RLVR and base models, following the observation that RLVR-induced changes are sparse and tend to concentrate on high-entropy tokens (Wang et al., 2025b).Source: §2 · p. 3
  • Token-level KL divergence measures distributional shift; Huan et al. (2025) also reported sparse changes, and the analysis includes the reversed KL variant.Source: §2 · p. 3

3 Dissecting the Token-Level Changes Introduced by RLVR

The section argues that signed ∆log p isolates sparse, reasoning-critical RLVR changes more precisely than magnitude-only metrics and explains their sparsity through updates to low-probability tokens.

3.2 Recovering RLVR Performance via Selective Token Replacement

  • ∆log p-based replacement matched RLVR accuracy with around 10% of tokens, requiring fewer substitutions than divergence or entropy across model pairs and metric variants.Divergence consistently outperformed entropy, but both required more replacements than ∆log p.Source: Figure 2 · p. 4, §3.2 · p. 5

3.1 Statistical Analysis: Directional vs. Magnitude-Based Metrics

  • Across ORZ, DAPO, and UniReason outputs on AIME-24, metric distributions were near zero; entropy and KL overlapped across models, while ∆log p showed distinct signed tails.The pattern held across tested model pairs and entropy/divergence variants, with ∆log p consistently bimodal.Source: §3.1 · p. 4

3.2 Recovering RLVR Performance via Selective Token Replacement

  • Replacing 5-30% of base-model tokens with metric-selected RLVR tokens recovered final RLVR accuracy, while random replacement improved performance more slowly.This indicates that RLVR-modified tokens are sparse but disproportionately important for reasoning.Source: §3.2 · p. 5

3.3 A Gradient-Based Explanation for the Sparse Updates

  • Low-probability tokens receive larger gradients and account for most gradient mass; high-∆log p tokens are also low-probability in both models, with probability increasing under RLVR.This links sparse RLVR changes to concentrated updates on rare, high-impact tokens.Source: §3.3 · p. 6
  • Filtering low-probability tokens during Qwen2.5-Math-7B DAPO rollouts impaired performance: even top-p=0.95 fell below top-p=1.0, and stricter filtering degraded it further.The intervention supports the importance of low-probability tokens to RLVR reasoning gains.Source: §3.3 · p. 6

4 Exploiting RLVR’s Directional Updates to Boost Reasoning

The section proposes test-time selective extrapolation along RLVR’s learned Δlog p direction and training-time reweighting of low-probability tokens to strengthen reasoning.

4.1 Test-Time Enhancement via Extrapolation

  • The method adds γ times the learned Δlog p to RLVR log probabilities, then selectively samples from the extrapolated policy at positions gated by large negative Δlog p.Because most positions have negligible Δlog p, selective gating aims to avoid distorting well-calibrated tokens.Source: §4.1 · p. 7
  • On AIME-24, selective extrapolation achieves higher Avg@32 than the RLVR baseline across ORZ-32B, DAPO-32B, and UniReason-14B, while selective replacement does not surpass it.Replacement and extrapolation use matched selection criteria; the difference is the extrapolation.Source: Figure 4 · p. 8, §4.1 · p. 8
  • In a simplified tabular softmax setting with natural policy gradient updates, the analysis indicates extrapolation along the learned difference direction can improve expected reward.The proof relies on idealized monotonic updates along the reward direction; the authors therefore motivate selective extrapolation at positions with significant Δlog p.Source: §4.1 · p. 8
  • Extrapolating directly on DAPO-32B improves AIME-24 performance as the intervention ratio rises to a moderate level, after which gains plateau.The pattern is consistent with amplifying a limited set of reasoning-critical tokens rather than intervening aggressively.Source: Table 1 · p. 8, §4.1 · p. 8

4.2 Training-Time Enhancement via Advantage Reweighting

  • Training-time reweighting modifies DAPO’s advantage to prioritize low-probability tokens, which are associated with high Δlog p and identified as important for RLVR training.The reweighting strength is controlled by α.Source: §4.2 · pp. 8–9
  • Across Qwen2.5-Math-7B and Qwen3-8B-Base, low-probability-token reweighting consistently improves reasoning accuracy (Avg@32) across tested datasets, while average Pass@16 also increases over the DAPO baseline.Source: §4.2 · p. 9
  • On Qwen2.5-Math-7B, low-probability-token reweighting achieves the best overall Avg@32 and Pass@16 among the compared methods, with response length steadily increasing during training.Dominate has lower training entropy under its tighter clip-higher ratio, which the authors link to reduced Pass@k.Source: Table 3 · p. 9, §4.2 · pp. 9–10, Figure 5 · p. 10

5 Related Work

Related work traces RL’s shift from alignment to reasoning and highlights evidence that RLVR updates are sparse, while identifying update direction—not just magnitude—as this paper’s focus.

5 Related Work

  • LLM reinforcement learning shifted from RLHF alignment (Ouyang et al., 2022) toward reasoning models, with this study adopting DAPO as its RLVR baseline (Yu et al., 2025).OpenAI o1 (Jaech et al., 2024) and DeepSeek R1 helped inspire later industrial and academic reasoning systems.Source: §5 · p. 10
  • Prior RLVR analyses found sparse updates but emphasized change magnitude; this paper extends update direction to the full distributional shift and proposes extrapolating along it.Earlier work quantified entropy changes (Cui et al., 2025), token gradient norms (Yang et al., 2025b), and token-replacement effects (Meng et al., 2026); sparsity was measured using entropy, KL divergence, and sparse gradient norms.Source: §5 · p. 10

6 Conclusion · A Selective Token Replacement & Extraploation · A.1 Implementation Details

The paper argues that signed token-level Δlog p reveals reasoning-critical RLVR updates and supports test-time extrapolation and training-time reweighting. The implementation details specify the evaluated models, sampling protocol, and extrapolation settings.

6 Conclusion

  • Δlog p identifies sparse, reasoning-critical RLVR updates better than magnitude-based metrics, and extrapolation and reweighting improve reasoning performance across settings.The authors frame update direction as a principle for diagnosing and improving RLVR.Source: §6 · p. 10
  • The extrapolation method requires two models and adds hyperparameters; proposed future work includes parameter-efficient finetuning and more adaptive threshold-strength selection.The authors also suggest evaluating reweighting at different model scales or combining it with adaptive training techniques.Source: §6 · p. 11
  • The reweighting approach remains to be evaluated across different model scales and in combination with other adaptive training techniques.These are identified as directions for future work.Source: §6 · p. 11

A.1 Implementation Details

  • The experiments use Qwen-series RLVR models: ORZ and DAPO finetuned from Qwen2.5-32B, and UniReason finetuned from Qwen3-14B-Base.ORZ uses PPO, DAPO uses DAPO, and UniReason uses GRPO (Hu et al., 2025a; Yu et al., 2025; Huan et al., 2025).Source: §A.1 · p. 15
  • Replacement performance is evaluated on AIME-24 with each model’s default chat template and a prompt requesting step-by-step reasoning and a boxed final answer.Sampling uses top-p=0.7, temperature=1.0, max-length=20k, and 32 responses per question; correctness is verified with Math-Verify and averaged as Avg@32.Source: §A.1 · p. 15
  • Selective replacement is controlled by threshold τ, while extrapolation strength is controlled by γ; the default ϵ_high=0.28 caused unstable training.The passage says the chosen value follows the recommendation in Yang et al. (2025b).Source: §6 · p. 10, §A.1 · p. 15

A Selective Token Replacement & Extraploation

  • For Fig. 4, Selective Extrapolate and Selective Replace share the same hyperparameters for each model, summarized in Table 4.The supplied table passage gives its scope but not hyperparameter values.Source: §A.1 · p. 15, Table 4 · p. 15

A.2 Additional Experiments

Additional experiments show that Δlog p selects reasoning-critical tokens with fewer replacements than divergence or entropy, while token-replacement effects vary across problems and can initially disrupt generation before stronger RL-token guidance helps.

Additional Experiments

  • On AIME-24, Δlog p reaches RLVR-level performance with fewer replacements than divergence or entropy, ranking highest for identifying reasoning-critical tokens.The ablation also evaluates RLVR-model entropy and two KL-divergence variants alongside base-model entropy, KL divergence, and log-probability difference.Source: §A.2 · p. 15, Figure 6 · p. 16
  • At an approximately 8% replacement rate, entropy favors logical transition words, whereas divergence and Δlog p select more specific mathematical tokens, including more math symbols.The authors suggest these mathematical tokens may improve reasoning more efficiently, consistent with entropy’s inferior performance.Source: §A.2 · pp. 15–16, Figure 7 · p. 16
  • Across problems, accuracy generally rises with increasing replacement before fluctuating, while inherently difficult problems remain at zero and replacement-only performance is capped by the RLVR model.A few problems initially lose accuracy at low replacement ratios and improve as replacement increases.Source: §A.2 · p. 16, Figure 8 · p. 17
  • For some initially harmed problems, unfamiliar RL-replaced tokens disrupt coherent generation; increasing replacement then guides generation more strongly with RL tokens, allowing performance to recover and improve.The passage gives DAPO problem 0 as an example of an initial accuracy drop followed by improvement.Source: §A.2 · pp. 16–17

A.3 Hyperparameter Sensitivity Analysis

Hyperparameter sensitivity tests show that test-time extrapolation robustly improves on replace-only token replacement and can exceed RLVR performance without further training.

A.3 Hyperparameter Sensitivity Analysis

  • The grid search tests γ values of 0.05 and 0.1 and varies τ by model, using a denser τ grid for UniReason because fewer replacements are needed to reach RLVR performance.DAPO and ORZ use τ ∈ {−0.5, −0.4, −0.3, −0.2, −0.1}; UniReason uses τ ∈ {−0.5, −0.45, −0.4, −0.35, −0.3}.Source: §A.3 · pp. 17–18
  • Once replacement matches RLVR performance, further increases in replacement provide little to no added benefit because performance is bounded by the RLVR model.In contrast, proper test-time extrapolation can exceed RLVR by 1–3 points.Source: §A.3 · p. 18
  • 1–3 points: Across nearly all models and hyperparameter settings, extrapolation outperforms replace-only replacement and can exceed RLVR performance without additional training.The comparison is reported for the selective extrapolation experiment in Table 5.Source: Table 5 · p. 18, §A.3 · p. 18

B RLVR Training Details

RLVR training uses the open-sourced DAPO recipe with model-specific generation limits and reweighting settings. Four Qwen2.5-Math-7B reweighting runs show consistent convergence, with all trials reaching or surpassing the reported Tab. 3 performance.

Hyperparameters Setting

  • The DAPO-based setup uses clip ratios ϵlow = 0.2 and ϵhigh = 0.28, learning rate 1e-6, 10-step warmup, and 16 gradient updates per RLVR step.Each step uses 512 prompts with 16 responses each, in mini-batches of 32 prompts; maximum generation lengths and overlong thresholds are 8k and 4k for Qwen2.5-Math-7B, and 20k and 16k for Qwen3-8b-base.Source: B RLVR Training Details · p. 18
  • Reweighting α is 0.2 for Qwen2.5 and 0.1 for Qwen3; α is 0.1 for ˜Adom_i,t and 0.01 for ˜APPL_i,t, with ϵhigh adjusted to 0.24.The latter settings follow recommended values by Deng et al. (2025) and Yang et al. (2025b).Source: B RLVR Training Details · p. 18

Reproducibility Analysis

  • Across four Qwen2.5-Math-7B reweighting runs, learning curves converged consistently and every trial reached or surpassed the performance reported in Tab. 3.Run 1 is the run reported in Fig. 5.Source: Figure 9 · p. 19, B RLVR Training Details · p. 19

C Performance beyond Pure-Math Reasoning Tasks

On Minerva, a 272-problem undergraduate STEM benchmark, both reweighting and test-time extrapolation show reasoning gains beyond pure-math tasks. Reweighting preserves exploration, while extrapolation improves accuracy over the RLVR model and πRL-only replacement.

C Performance beyond Pure-Math Reasoning Tasks

  • To assess reasoning generalization on non-math tasks beyond math-focused datasets, the authors use Minerva, which contains 272 undergraduate-level STEM problems spanning subjects such as Chemistry and Astronomy6.Source: C Performance beyond Pure-Math… · p. 19
  • Reweighting-trained models outperform baselines on Minerva reasoning accuracy while maintaining comparable or improved Pass@k scores.The evaluation uses the same sampling parameters as other evaluations; Table 6 reports RLVR-trained model performance on Minerva.Source: C Performance beyond Pure-Math… · p. 19, Table 6 · p. 19
  • Test-time extrapolation consistently improves Avg@8 accuracy over the RLVR model on Minerva for evaluated 14B–32B models.Avg@8 is reported because Minerva has around seven times as many questions as AIME24; hyperparameter-grid results also consistently outperform replacing with πRL only.Source: C Performance beyond Pure-Math… · p. 19, Figure 10 · p. 20, Table 7 · p. 20

D Proofs

The proofs derive a softmax-policy gradient expression for DAPO and establish that, under the bandit setting, extrapolating along the learned direction increases expected reward unless rewards are constant.

D Proofs

  • The appendix derives the gradient of DAPO for an unclipped token under a softmax-parameterized policy and expresses its gradient’s ℓ1-norm.The derivation omits context notation and treats the token as vocabulary index k.Source: D Proofs · pp. 20–21
  • Theorem 4.1’s proof shows that the extrapolation direction has a nonnegative expected-reward directional derivative, with equality only when rewards are constant.The proof uses the zero expected advantage and the Chebyshev sum inequality.Source: D Proofs · pp. 21–22
  • Under the bandit assumption, advantage and extrapolation-direction values are ordered like rewards; induction establishes this ordering for the direction at every iteration.The proof begins from the NPG update and extends the ordering from iteration t to t + 1.Source: D Proofs · p. 21

E Statistical Comparison of Different Metrics · F The Use of Large Language Models

Across three RLVR model pairs, signed Δlog p distinguishes RLVR-generated from base-generated text more clearly than the examined magnitude-based metrics, while high-Δlog p tokens are associated with problem-solving language. The paper also states that LLMs were used only to polish pre-existing text, with author review of suggested changes.

E Statistical Comparison of Different Metrics

F The Use of Large Language Models

E Statistical Comparison of Different Metrics

Loading 2603.22117v1…