Angles Don't Lie: Unlocking Training-Efficient RL Through the Model's Own Signals
Not on the map. Why?
Abstract (arXiv)
Current Reinforcement Fine-tuning (RFT) paradigms for Large Language Models (LLMs) suffer from sample inefficiency due to the redundant exposure of identical queries under uniform data sampling. While previous work has explored curriculum learning via heuristic difficulty metrics, these strategies exhibit limitations by neglecting the intrinsic learning signals generated by the model itself, thus leading to suboptimal training regimes. In this paper, we identify a model-inherent signal termed angle concentration that effectively reflects an LLM's capacity to learn from specific data. We theoretically and empirically demonstrate a correlation between the angular distribution of token hidden state vectors and the resulting gradient, revealing a learning preference for data exhibiting higher angle concentration. Inspired by this finding, we propose GAIN-RL, a Gradient-driven Angle-Informed Navigated RL framework. By leveraging the model's intrinsic angle concentration signal, GAIN-RL dynamically selects training data in each epoch, ensuring consistently impactful gradient updates and thus significantly enhancing overall training efficiency. Empirical evaluations show that GAIN-RL (GRPO) achieves over a 2.5x acceleration in training efficiency across diverse mathematical and coding tasks and varying model scales. Furthermore, GAIN-RL (GRPO)'s efficient sampling yields data-efficient training, achieving better performance with half the original data compared to vanilla GRPO with full training data. Code is realsed at https://github.com/wangqinsi1/GAINRL/tree/main.
Key facts. Cited by 13 papers in the map (data through 14 Oct 2026). Builds on 31 of its 48 references that are in the map. First posted on arXiv 2 Jun 2025.
Builds on
31 of 48 references.
References · 31
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
- 2025s1: Simple test-time scalingcited by 1,341
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
- 2025SimpleRL-Zoo: Investigating and Taming Zero Reinforcement Learning for Open Base Models in the Wildcited by 511
- 2025LIMO: Less is More for Reasoningcited by 498
- 2023Symbolic Discovery of Optimization Algorithmscited by 434
19 more
- 2019Selection via Proxy: Efficient Data Selection for Deep Learningcited by 339
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 187
- 2024DsDm: Model-Aware Dataset Selection with Datamodelscited by 105
- 2025H-CoT: Hijacking the Chain-of-Thought Safety Reasoning Mechanism to Jailbreak Large Reasoning Models, Including OpenAI o1/o3, DeepSeek-R1, and Gemini 2.0 Flash Thinkingcited by 99
- 2025Efficient Reinforcement Finetuning via Adaptive Curriculum Learningcited by 93
- 2019Orthogonal Deep Neural Networkscited by 72
- 2024Active Preference Optimization for Sample Efficient RLHFcited by 62
- 2023No Train No Gain: Revisiting Efficient Training Algorithms For Transformer-based Language Modelscited by 41
- 2023Neuron-Level Knowledge Attribution in Large Language Modelscited by 37
- 2024Configurable Foundation Models: Building LLMs from a Modular Perspectivecited by 22
- 2024Towards Data-Centric RLHF: Simple Metrics for Preference Dataset Comparisoncited by 21
- 2025HippoMM: Hippocampal-inspired Multimodal Memory for Long Audiovisual Event Understandingcited by 19
- 2025Keyframe-oriented Vision Token Pruning: Enhancing Efficiency of Large Vision Language Models on Long-Form Video Processingcited by 18
- 2024Training-Free Adaptive Diffusion with Bounded Difference Approximation Strategycited by 14
- 2024CoreInfer: Accelerating Large Language Model Inference with Semantics-Inspired Adaptive Sparse Activationcited by 14
- 2023GIO: Gradient Information Optimization for Training Dataset Selectioncited by 13
- 2024Once for Both: Single Stage of Importance and Sparsity Search for Vision Transformer Compressioncited by 8
- 2025CoreMatching: A Co-adaptive Sparse Inference Framework with Token and Neuron Pruning for Comprehensive Acceleration of Vision-Language Modelscited by 8
- 2023MathNAS: If Blocks Have a Role in Mathematical Architecture Designcited by 4
Built on by
Citing papers per year: 2025: 6; 2026: 7.
13 papers.
Sort:
By area:
- 2025KVCOMM: Online Cross-context KV-cache Communication for Efficient LLM-based Multi-agent Systemscited by 40
- 2025Vision-Zero: Scalable VLM Self-Improvement via Strategic Gamified Self-Playcited by 37
- 2025AMAS: Adaptively Determining Communication Topology for LLM-based Multi-Agent Systemcited by 15
- 2026CoBA-RL: Capability-Oriented Budget Allocation for Reinforcement Learning in LLMscited by 12
- 2025FT-MDT: Extracting Decision Trees from Medical Texts via a Novel Low-rank Adaptation Methodcited by 7
- 2025PuzzleCraft: Exploration-Aware Curriculum Learning for Puzzle-Based RLVR in VLMscited by 4
- 2026When RL Meets Adaptive Speculative Training: A Unified Training-Serving Systemcited by 3
- 2026EVA: Accelerating LLM Decoding via an Efficient Vector Quantization Architecturecited by 2
- 2026Guiding LLM Post-training Data Engineering with Model Internals from Sparse Autoencoderscited by 2
- 2025Coarse-to-Fine Personalized LLM Impressions for Streamlined Radiology Reportscited by 1
- 2026Curriculum Learning-Guided Progressive Distillation in Large Language Modelscited by 1
- 2026A Survey of Reinforcement Learning for Large Language Models under Data Scarcity: Challenges and Solutionscited by 0
1 more
- 2026Vortex: Bridging Extreme Compression and Efficient LLM Inference10 Sep 2026
- 2026Guiding LLM Post-training Data Engineering with Model Internals from Sparse Autoencoders26 May 2026
- 2026EVA: Accelerating LLM Decoding via an Efficient Vector Quantization Architecture22 May 2026
- 2026Curriculum Learning-Guided Progressive Distillation in Large Language Models11 May 2026
- 2026A Survey of Reinforcement Learning for Large Language Models under Data Scarcity: Challenges and Solutions19 Apr 2026
- 2026When RL Meets Adaptive Speculative Training: A Unified Training-Serving System6 Feb 2026
- 2026CoBA-RL: Capability-Oriented Budget Allocation for Reinforcement Learning in LLMs3 Feb 2026
- 2025PuzzleCraft: Exploration-Aware Curriculum Learning for Puzzle-Based RLVR in VLMs16 Dec 2025
- 2025KVCOMM: Online Cross-context KV-cache Communication for Efficient LLM-based Multi-agent Systems14 Oct 2025
- 2025FT-MDT: Extracting Decision Trees from Medical Texts via a Novel Low-rank Adaptation Method6 Oct 2025
- 2025AMAS: Adaptively Determining Communication Topology for LLM-based Multi-Agent System2 Oct 2025
- 2025Vision-Zero: Scalable VLM Self-Improvement via Strategic Gamified Self-Play29 Sep 2025
1 more
- 2026CoBA-RL: Capability-Oriented Budget Allocation for Reinforcement Learning in LLMscited by 12
- 2025PuzzleCraft: Exploration-Aware Curriculum Learning for Puzzle-Based RLVR in VLMscited by 4
- 2026A Survey of Reinforcement Learning for Large Language Models under Data Scarcity: Challenges and Solutionscited by 0
- 2026A Survey of Reinforcement Learning for Large Language Models under Data Scarcity: Challenges and Solutions19 Apr 2026
- 2026CoBA-RL: Capability-Oriented Budget Allocation for Reinforcement Learning in LLMs3 Feb 2026
- 2025PuzzleCraft: Exploration-Aware Curriculum Learning for Puzzle-Based RLVR in VLMs16 Dec 2025
- 2025FT-MDT: Extracting Decision Trees from Medical Texts via a Novel Low-rank Adaptation Methodcited by 7
- 2026EVA: Accelerating LLM Decoding via an Efficient Vector Quantization Architecturecited by 2
- 2025Coarse-to-Fine Personalized LLM Impressions for Streamlined Radiology Reportscited by 1
- 2026Curriculum Learning-Guided Progressive Distillation in Large Language Modelscited by 1
- 2026EVA: Accelerating LLM Decoding via an Efficient Vector Quantization Architecture22 May 2026
- 2026Curriculum Learning-Guided Progressive Distillation in Large Language Models11 May 2026
- 2025FT-MDT: Extracting Decision Trees from Medical Texts via a Novel Low-rank Adaptation Method6 Oct 2025
- 2025Coarse-to-Fine Personalized LLM Impressions for Streamlined Radiology Reports19 Aug 2025
Similar, not cited
Close in wording, no citation either way
- 2025Reinforcement Learning Fine-Tuning Enhances Activation Intensity and Diversity in the Internal Circuitry of LLMs
- 2025Ignore the KL Penalty! Boosting Exploration on Critical Tokens to Enhance RL Fine-Tuning
- 2026Application-Driven Pedagogical Knowledge Optimization of Open-Source LLMs via Reinforcement Learning and Supervised Fine-Tuning
- 2026Supervised Fine-Tuning versus Reinforcement Learning: A Study of Post-Training Methods for Large Language Models
- 2026GradAlign: Gradient-Aligned Data Selection for LLM Reinforcement Learning
- 2026ReMiT: RL-Guided Mid-Training for Iterative LLM Evolution
- 2025On Predictability of Reinforcement Learning Dynamics for Large Language Models
- 2025Reinforcement Learning Fine-Tunes a Sparse Subnetwork in Large Language Models