Search papers, labs, and topics across Lattice.
This paper investigates the role of high-entropy tokens in Reinforcement Learning with Verifiable Rewards (RLVR) for enhancing the reasoning capabilities of large language models (LLMs). The authors introduce the Gradient Magnitude-based Token Selection (GMTS) method, which improves upon traditional entropy-based selection by better quantifying token importance through the connection between entropy and gradient magnitude. Results show that training with the top 20% of tokens selected via GMTS yields superior performance across multiple reasoning tasks compared to entropy-based methods, highlighting the method's effectiveness in optimizing RLVR training.
Training LLMs with tokens selected by gradient magnitude rather than just entropy boosts reasoning performance across diverse tasks.
Reinforcement learning (RL), particularly RL with Verifiable Rewards (RLVR), has recently emerged as a central paradigm for enhancing large language models' (LLMs) reasoning abilities, demonstrating remarkable effectiveness across reasoning tasks. Recent studies suggest that high-entropy tokens play an exceptionally important role in model training, since training with only the highest 20% entropy tokens yields significant performance gains. However, why such high-entropy tokens are beneficial remains insufficiently understood. In this work, we find that although high-entropy tokens within one answer tend to correlate with large gradient magnitude, entropy alone fails to consistently reflect token importance across different answers, considering the variations in the answer-level reward signals. Based on this observation, we introduce the Gradient Magnitude-based Token Selection (GMTS) method to quantify token importance, which leverages the entropy-gradient connection to approximate gradient-magnitude rankings for token selection. We find that training on the top 20% tokens ranked by GMTS consistently outperforms entropy-based token selection across three reasoning domains and various model sizes, suggesting that GMTS provides a more fine-grained estimate of token contribution for RLVR training.
