Unified Data Selection for LLM Reasoning
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Effectively training Large Language Models (LLMs) for complex, long-CoT reasoning is often bottlenecked by the need for massive high-quality reasoning data. Existing methods are either computationally expensive or fail to reliably distinguish high- from low-quality reasoning samples. To address this, we propose High-Entropy Sum (HES), a training-free metric that quantifies reasoning quality by summing only the entropy of the top (e.g., 0.5%) highest-entropy tokens in each reasoning sample. We validate HES across three mainstream training paradigms: Supervised Fine-tuning (SFT), Rejection Fine-tuning (RFT), and Reinforcement Learning (RL), with extensive results demonstrating its consistent effectiveness and significantly reduced computational overhead. In SFT, training on the top 20% HES-ranked data matches full-dataset performance, while using the lowest-HES data degrades it. In RFT, our HES-based training approach significantly outperforms baseline methods. In RL, HES-selected successful trajectories enable the model to learn strong reasoning patterns, significantly surpassing other compared methods. Our findings establish HES as a robust, training-free metric that enables a unified, effective, and efficient method for developing advanced reasoning in LLMs.
Key facts. Cited by 2 papers in the map (data through 13 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 31 of its 60 references that are in the map. First posted on arXiv 21 May 2026.
Builds on
31 of 60 references.
Closest ancestors · 4
Same direction, newest first
- 2025Deep Think with Confidence21 Aug
- 2025LightThinker: Thinking Step-by-Step Compression21 Feb
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters6 Aug
- 2023Algorithm of Thoughts: Enhancing Exploration of Ideas in Large Language Models20 Aug
Other references · 27
- 2017Proximal Policy Optimization Algorithmscited by 14,592
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,091
- 2025Qwen3 Technical Reportcited by 8,510
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,247
- 2024OpenAI o1 System Cardcited by 1,233
- 2021Scaling Language Models: Methods, Analysis & Insights from Training Gophercited by 1,201
- 2022Program of Thoughts Prompting: Disentangling Computation from Reasoning for Numerical Reasoning Taskscited by 1,066
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 548
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 518
- 2025Process Reinforcement through Implicit Rewardscited by 484
- 2023Scaling Relationship on Learning Mathematical Reasoning with Large Language Modelscited by 421
- 2025Towards Reasoning Era: A Survey of Long Chain-of-Thought for Reasoning Large Language Modelscited by 383
15 more
- 2025Chain of Draft: Thinking Faster by Writing Lesscited by 292
- 2023ToRA: A Tool-Integrated Reasoning Agent for Mathematical Problem Solvingcited by 278
- 2024A Survey on Data Selection for Language Modelscited by 232
- 2025AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning datasetcited by 171
- 2024QuRating: Selecting High-Quality Data for Training Language Modelscited by 153
- 2025Sketch-of-Thought: Efficient LLM Reasoning with Adaptive Cognitive-Inspired Sketchingcited by 116
- 2024BOND: Aligning LLMs with Best-of-N Distillationcited by 74
- 2025Challenging the Boundaries of Reasoning: An Olympiad-Level Math Benchmark for Large Language Modelscited by 67
- 2025Unlocking Multimodal Mathematical Reasoning via Process Reward Modelcited by 64
- 2024Self-Reflection in LLM Agents: Effects on Problem-Solving Performancecited by 39
- 2025Chain-of-Reasoning: Towards Unified Mathematical Reasoning in Large Language Models via a Multi-Paradigm Perspectivecited by 29
- 2022Data-Efficient Finetuning Using Cross-Task Nearest Neighborscited by 26
- 2023In-Context Alignment: Chat with Vanilla Language Models Before Fine-Tuningcited by 19
- 2023When Less is More: Investigating Data Pruning for Pretraining LLMs at Scalecited by 10
- 2023Comprehensive Benchmarking of Entropy and Margin Based Scoring Metrics for Data Selectioncited by 1
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Think Just Enough: Sequence-Level Entropy as a Confidence Signal for LLM Reasoning
- 2026InstructDiff: Domain-Adaptive Data Selection via Differential Entropy for Efficient LLM Fine-Tuning
- 2026GMTS: Gradient Magnitude-based Token Selection Improves RLVR Training for LLM Reasoning
- 2025Right Question is Already Half the Answer: Fully Unsupervised LLM Reasoning Incentivization
- 2025SRFT: A Single-Stage Method with Supervised and Reinforcement Fine-Tuning for Reasoning
- 2025DeepDistill: Enhancing LLM Reasoning Capabilities via Large-Scale Difficulty-Graded Data Training
- 2026On the Step Length Confounding in LLM Reasoning Data Selection
- 2025Efficient Reinforcement Learning with Semantic and Token Entropy for LLM Reasoning