Token Hidden Reward: Steering Exploration-Exploitation in Group Relative Deep Reinforcement Learning
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
- 2025Pass@k Training for Adaptively Balancing Exploration and Exploitation of Large Reasoning Models14 Aug
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning2 Jun
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models28 May
- 2025On the Effect of Negative Gradient in Group Relative Deep Reinforcement Optimization24 May
- 2025SimpleRL-Zoo: Investigating and Taming Zero Reinforcement Learning for Open Base Models in the Wild24 Mar
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRs3 Mar
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMs22 Jan
- 2024Rewarding Progress: Scaling Automated Process Verifiers for LLM Reasoning10 Oct
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 25
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
- 2025Group Sequence Policy Optimizationcited by 788
- 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 147
13 more
- 2025Med-R1: Reinforcement Learning for Generalizable Medical Reasoning in Vision-Language Modelscited by 144
- 2020Neural collapse with unconstrained featurescited by 140
- 2025MedReason: Eliciting Factual Medical Reasoning Steps in LLMs via Knowledge Graphscited by 105
- 2017A Tutorial on Kernel Density Estimation and Recent Advancescited by 100
- 2024Learning Dynamics of LLM Finetuningcited by 93
- 2025Is Best-of-N the Best of Them? Coverage, Scaling, and Optimality in Inference-Time Alignmentcited by 91
- 2024Inference-Aware Fine-Tuning for Best-of-N Sampling in Large Language Modelscited by 86
- 2025Pass@K Policy Optimization: Solving Harder Reinforcement Learning Problemscited by 83
- 2024Unintentional Unalignment: Likelihood Displacement in Direct Preference Optimizationcited by 66
- 2024LLM Discussion: Enhancing the Creativity of Large Language Models via Discussion Framework and Role-Playcited by 65
- 2024Code Repair with LLMs gives an Exploration-Exploitation Tradeoffcited by 50
- 2018Exploration versus exploitation in reinforcement learning: a stochastic control approachcited by 17
- 2025Should You Use Your Large Language Model to Explore or Exploit?cited by 8
Built on by
Citing papers per year: 2025: 4; 2026: 5.
9 papers.
- 2025On Group Relative Policy Optimization Collapse in Agent Search: The Lazy Likelihood-Displacementcited by 15
- 2025Advantage Shaping as Surrogate Reward Maximization: Unifying Pass@K Policy Gradientscited by 10
- 2025Sharpness-Guided Group Relative Policy Optimization via Probability Shapingcited by 4
- 2026QUATRO: Query-Adaptive Trust Region Policy Optimization for LLM Fine-tuningcited by 2
- 2026The Cancellation Hypothesis in Critic-Free RL: From Outcome Rewards to Token Creditscited by 2
- 2025CoRPO: Adding a Correctness Bias to GRPO Improves Generalizationcited by 1
- 2026Tool-Aware Optimization with Entropy Guidance for Efficient Agentic Reinforcement Learningcited by 1
- 2026Holder Policy Optimisationcited by 0
- 2026Learning Dynamics of Continual Learning: A Unified View of Data Attribution, Forgetting, and Plasticity Losscited by 0
- 2026Learning Dynamics of Continual Learning: A Unified View of Data Attribution, Forgetting, and Plasticity Loss27 Sep 2026
- 2026Tool-Aware Optimization with Entropy Guidance for Efficient Agentic Reinforcement Learning2 Jun 2026
- 2026Holder Policy Optimisation12 May 2026
- 2026The Cancellation Hypothesis in Critic-Free RL: From Outcome Rewards to Token Credits9 May 2026
- 2026QUATRO: Query-Adaptive Trust Region Policy Optimization for LLM Fine-tuning4 Feb 2026
- 2025On Group Relative Policy Optimization Collapse in Agent Search: The Lazy Likelihood-Displacement3 Dec 2025
- 2025CoRPO: Adding a Correctness Bias to GRPO Improves Generalization6 Nov 2025
- 2025Sharpness-Guided Group Relative Policy Optimization via Probability Shaping29 Oct 2025
- 2025Advantage Shaping as Surrogate Reward Maximization: Unifying Pass@K Policy Gradients27 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2025Rewarding the Unlikely: Lifting GRPO Beyond Distribution Sharpening
- 2025λ-GRPO: Unifying the GRPO Frameworks with Learnable Token Preferences
- 2025Low-probability Tokens Sustain Exploration in Reinforcement Learning with Verifiable Reward
- 2026GRAIL: Gradient-Reweighted Advantages for Reinforcement Learning with Verifiable Rewards
- 2025Diversity or Precision? A Deep Dive into Next Token Prediction
- 2026From Absolute to Relative: Rethinking Reward Shaping in Group-Based Reinforcement Learning
- 2025Noise-corrected GRPO: From Noisy Rewards to Unbiased Gradients
- 2025Exploration vs Exploitation: Rethinking RLVR through Clipping, Entropy, and Spurious Reward