mlsanityData to 7 Oct 2026

Token Hidden Reward: Steering Exploration-Exploitation in Group Relative Deep Reinforcement Learning

arXiv 2510.03669 ↗First posted 4 Oct 2025cs.LG cs.CLcited by 9

On the map: LLM reasoning → Reinforcement learning for reasoning

Abstract (arXiv)

Reinforcement learning with verifiable rewards has significantly advanced the reasoning capabilities of large language models, yet how to explicitly steer training toward exploration or exploitation remains an open problem. We introduce Token Hidden Reward (THR), a token-level metric that quantifies each token's influence on the likelihood of correct responses under Group Relative Policy Optimization (GRPO). We find that training dynamics are dominated by a small subset of tokens with high absolute THR values. Most interestingly, tokens with positive THR strengthen confidence in correct outputs, thus favoring exploitation, while tokens with negative THR preserve probability mass for alternative outputs, enabling exploration. This insight suggests a natural intervention: a THR-guided reweighting algorithm that modulates GRPO's learning signals to explicitly bias training toward exploitation or exploration. We validate the efficacy of this algorithm on diverse math reasoning benchmarks. By amplifying tokens with positive THR value and weakening negative ones, our algorithm improves greedy-decoding accuracy, favoring exploitation. The reverse strategy yields consistent gains in Pass@K accuracy, favoring exploration. We further demonstrate that our algorithm integrates seamlessly with other RL objectives such as GSPO and generalizes across architectures including Llama. These findings establish THR as a principled and fine-grained mechanism for dynamically controlling exploration and exploitation in RL-tuned LLMs, providing new tools for targeted fine-tuning in reasoning-intensive applications.

Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 35 of its 40 references that are in the map. First posted on arXiv 4 Oct 2025.

Builds on

35 of 40 references.

Closest ancestors · 10

Same direction, newest first

  1. 2025Pass@k Training for Adaptively Balancing Exploration and Exploitation of Large Reasoning Models14 Aug
  2. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning2 Jun
  3. 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models28 May
  4. 2025On the Effect of Negative Gradient in Group Relative Deep Reinforcement Optimization24 May
  5. 2025SimpleRL-Zoo: Investigating and Taming Zero Reinforcement Learning for Open Base Models in the Wild24 Mar
  6. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
  7. 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRs3 Mar
  8. 2025Kimi k1.5: Scaling Reinforcement Learning with LLMs22 Jan
  9. 2024Rewarding Progress: Scaling Automated Process Verifiers for LLM Reasoning10 Oct
  10. 2017Proximal Policy Optimization Algorithms20 Jul

Other references · 25

  1. 2022Training language models to follow instructions with human feedbackcited by 15,712
  2. 2024The Llama 3 Herd of Modelscited by 14,192
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  4. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  5. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  6. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
  7. 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
  8. 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
  9. 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
  10. 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
  11. 2025Group Sequence Policy Optimizationcited by 788
  12. 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 147
13 more
  1. 2025Med-R1: Reinforcement Learning for Generalizable Medical Reasoning in Vision-Language Modelscited by 144
  2. 2020Neural collapse with unconstrained featurescited by 140
  3. 2025MedReason: Eliciting Factual Medical Reasoning Steps in LLMs via Knowledge Graphscited by 105
  4. 2017A Tutorial on Kernel Density Estimation and Recent Advancescited by 100
  5. 2024Learning Dynamics of LLM Finetuningcited by 93
  6. 2025Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time Alignmentcited by 91
  7. 2024Inference-Aware Fine-Tuning for Best-of-N Sampling in Large Language Modelscited by 86
  8. 2025Pass@K Policy Optimization: Solving Harder Reinforcement Learning Problemscited by 83
  9. 2024Unintentional Unalignment: Likelihood Displacement in Direct Preference Optimizationcited by 66
  10. 2024LLM Discussion: Enhancing the Creativity of Large Language Models via Discussion Framework and Role-Playcited by 65
  11. 2024Code Repair with LLMs gives an Exploration-Exploitation Tradeoffcited by 50
  12. 2018Exploration versus exploitation in reinforcement learning: a stochastic control approachcited by 17
  13. 2025Should You Use Your Large Language Model to Explore or Exploit?cited by 8

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 4; 2026: 5.

9 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Rewarding the Unlikely: Lifting GRPO Beyond Distribution Sharpening
  2. 2025λ-GRPO: Unifying the GRPO Frameworks with Learnable Token Preferences
  3. 2025Low-probability Tokens Sustain Exploration in Reinforcement Learning with Verifiable Reward
  4. 2026GRAIL: Gradient-Reweighted Advantages for Reinforcement Learning with Verifiable Rewards
  5. 2025Diversity or Precision? A Deep Dive into Next Token Prediction
  6. 2026From Absolute to Relative: Rethinking Reward Shaping in Group-Based Reinforcement Learning
  7. 2025Noise-corrected GRPO: From Noisy Rewards to Unbiased Gradients
  8. 2025Exploration vs Exploitation: Rethinking RLVR through Clipping, Entropy, and Spurious Reward