mlsanityData to 14 Oct 2026

Which Tokens Matter? Adaptive Token Selection for RLVR with the Relative Surprisal Index

arXiv 2606.31575 ↗First posted 30 Jun 2026cs.AIcited by 3

On the map: LLM reasoning → Reinforcement learning for reasoning

Abstract (arXiv)

Reinforcement learning (RL) has become a powerful tool for propelling Large Language Models (LLMs) beyond imitation-based training towards more robust reasoning capabilities. Among existing approaches, RL with Verifiable Rewards (RLVR) has emerged as a pivotal paradigm for advancing LLM reasoning. Despite its empirical success, recent studies have offered different insights. One line of inquiry advocates prioritizing high-entropy token positions during training, while another perspective cautions against allowing low-probability tokens to dominate gradient updates. Notably, although high-entropy tokens are usually correlated with low probability, both paradigms empirically yield substantial performance gains. In this work, we argue that evaluating sampled-token probability or entropy in isolation is insufficient to capture the policy optimization dynamics. To resolve this tension, we introduce the Relative Surprisal Index (RSI), a principled, information-theoretic metric that naturally couples the token's entropy with the probability of the selected token. We show that, under mild conditions, RSI is related to the local ratio between the first-order variations of the logit-gradient norm and predictive entropy under a selected-logit perturbation. Building on RSI, we propose RSI Selection (RSI-S), an entropy-adaptive token filtering method that retains tokens within a stable RSI interval. RSI-S successfully reconciles previous contradictory paradigms and filters out both redundant low-surprisal tokens and unstable high-surprisal tail tokens. Empirical evaluations show that RSI-S achieves higher avg@32 accuracy across different model scales (Qwen2.5-1.5B, 3B, and 7B) on AIME and AMC benchmarks: RSI-S improves avg@32 accuracy by 2--3 percentage points over GRPO. Overall, RSI offers a promising perspective for RLVR improvement.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 32 of its 34 references that are in the map. First posted on arXiv 30 Jun 2026.

Builds on

32 of 34 references.

Closest ancestors · 19

Same direction, newest first

  1. 2026Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMs23 Mar
  2. 2026FIPO: Eliciting Deep Reasoning with Future-KL Influenced Policy Optimization20 Mar
  3. 2026Future-KL Regularized GRPO: Process-Level Credit Assignment from f-Divergence Regularization15 Jan
  4. 2026GDPO: Group reward-Decoupled Normalization Policy Optimization for Multi-reward RL Optimization8 Jan
  5. 2025Rethinking Sample Polarity in Reinforcement Learning with Verifiable Rewards25 Dec
  6. 2025Generalization of RLVR Using Causal Reasoning as a Testbed23 Dec
  7. 2025Rethinking Entropy Interventions in RLVR: An Entropy Change Perspective11 Oct
  8. 2025EDGE-GRPO: Entropy-Driven GRPO with Guided Error Correction for Advantage Diversity29 Jul
  9. 2025Reasoning or Memorization? Unreliable Results of Reinforcement Learning Due to Data Contamination14 Jul
  10. 2025Reshaping Reasoning in LLMs: A Theoretical Analysis of RL Training Dynamics through Pattern Selection5 Jun
  11. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning2 Jun
  12. 2025The Surprising Effectiveness of Negative Reinforcement in LLM Reasoning2 Jun
7 more
  1. 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models28 May
  2. 2025Do Not Let Low-Probability Tokens Over-Dominate in RL for LLMs19 May
  3. 2025SEED-GRPO: Semantic Entropy Enhanced GRPO for Uncertainty-Aware Policy Optimization18 May
  4. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?18 Apr
  5. 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
  6. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
  7. 2017Proximal Policy Optimization Algorithms20 Jul

Other references · 13

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  3. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
  4. 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,090
  5. 2025Group Sequence Policy Optimizationcited by 835
  6. 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 672
  7. 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
  8. 2026Qwen3.5-Omni Technical Reportcited by 174
  9. 2025Does Math Reasoning Improve General LLM Capabilities? Understanding Transferability of LLM Reasoningcited by 106
  10. 2025DeepSeekMath-V2: Towards Self-Verifiable Mathematical Reasoningcited by 94
  11. 2026LLaDA2.1: Speeding Up Text Diffusion via Token Editingcited by 54
  12. 2025BAPO: Stabilizing Off-Policy Reinforcement Learning for LLMs via Balanced Policy Optimization with Adaptive Clippingcited by 47
1 more
  1. 2026On the Entropy Dynamics in Reinforcement Fine-Tuning of Large Language Modelscited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Stabilizing Knowledge, Promoting Reasoning: Dual-Token Constraints for RLVR
  2. 2025Revisiting Entropy in Reinforcement Learning for Large Reasoning Models
  3. 2025Diversity or Precision? A Deep Dive into Next Token Prediction
  4. 2026Rethinking RL for LLM Reasoning: It's Sparse Policy Selection, Not Capability Learning
  5. 2026Not All Tokens Learn Alike: Attention Entropy Reveals Heterogeneous Signals in RL Reasoning
  6. 2025Beyond High-Entropy Exploration: Correctness-Aware Low-Entropy Segment-Based Advantage Shaping for Reasoning LLMs
  7. 2025Low-probability Tokens Sustain Exploration in Reinforcement Learning with Verifiable Reward
  8. 2026On the Direction of RLVR Updates for LLM Reasoning: Identification and Exploitation