mlsanityData to 7 Oct 2026

Entropy Centroids as Intrinsic Rewards for Test-Time Scaling

arXiv 2604.26173 ↗First posted 28 Apr 2026cs.LG cs.AI cs.CLcited by 0

On the map: LLM reasoning → Test-time compute and efficient thinking

also in 1 more
Abstract (arXiv)

An effective way to scale up test-time compute of large language models is to sample multiple responses and then select the best one, as in Grok Heavy and Gemini Deep Think. Existing selection methods often rely on external reward models, which requires training a strong reward model and introduces additional computation overhead. As an alternative, previous approaches have explored intrinsic signals, such as confidence and entropy, but these signals are noisy with naive aggregation. In this work, we observe that high-entropy tokens tend to cluster into consecutive groups during inference, providing a more stable notion of model uncertainty than individual tokens. Together, these clusters reveal temporal patterns of model uncertainty throughout the inference process. Motivated by this observation, we propose to use the temporal structure of uncertainty as an intrinsic reward. To this end, we first formalize the basic unit of segment-level uncertainty as the High Entropy Phase (HEP), a variable-length segment that begins at a high-entropy token and ends when consecutive low-entropy tokens appear. We then define the Entropy Centroid, inspired by the concept of the center of mass in physics, as the weighted average position of all HEPs along the trajectory. Intuitively, a lower centroid indicates early exploration followed by confident generation, which we find often corresponds to higher response quality. Based on this insight, we propose the Lowest Centroid method, which selects the response with the lowest entropy centroid among multiple candidates. Experiments on mathematics, code generation, logical reasoning, and agentic tasks, across model scales ranging from 14B to 480B, show that Lowest Centroid consistently outperforms existing baselines and delivers stable gains as model size increases. Code is available at https://github.com/hkust-nlp/entropy-centroid.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 40 of its 57 references that are in the map. First posted on arXiv 28 Apr 2026.

Builds on

40 of 57 references.

Closest ancestors · 8

Same direction, newest first

  1. 2025SolverLLM: Leveraging Test-Time Scaling for Optimization Problem via LLM-Guided Search19 Oct
  2. 2025TrajSelector: Harnessing Latent Representations for Efficient and Effective Best-of-N in Large Reasoning Model18 Oct
  3. 2025Think Just Enough: Sequence-Level Entropy as a Confidence Signal for LLM Reasoning9 Oct
  4. 2025Deep Think with Confidence21 Aug
  5. 2025gpt-oss-120b & gpt-oss-20b Model Card8 Aug
  6. 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attention16 Jun
  7. 2025Does Thinking More always Help? Mirage of Test-Time Scaling in Reasoning Models4 Jun
  8. 2025Scalable Best-of-N Selection for Large Language Models via Self-Certainty25 Feb

Other references · 32

  1. 2025Qwen3 Technical Reportcited by 8,195
  2. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  3. 2019The Curious Case of Neural Text Degenerationcited by 2,881
  4. 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 1,906
  5. 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
  6. 2023ChatEval: Towards Better LLM-based Evaluators through Multi-Agent Debatecited by 724
  7. 2025τ2-Bench: Evaluating Conversational Agents in a Dual-Control Environmentcited by 558
  8. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 540
  9. 2024BigCodeBench: Benchmarking Code Generation with Diverse Function Calls and Complex Instructionscited by 527
  10. 2025LIMO: Less is More for Reasoningcited by 488
  11. 2023DoLa: Decoding by Contrasting Layers Improves Factuality in Large Language Modelscited by 367
  12. 2024Quiet-STaR: Language Models Can Teach Themselves to Think Before Speakingcited by 292
20 more
  1. 2025TTRL: Test-Time Reinforcement Learningcited by 253
  2. 2025Learning to Reason without External Rewardscited by 199
  3. 2024Language Model Cascades: Token-level uncertainty and beyondcited by 115
  4. 2025Olmo 3cited by 95
  5. 2026Ministral 3cited by 88
  6. 2025SynLogic: Synthesizing Verifiable Reasoning Data at Scale for Learning Logical Reasoning and Beyondcited by 47
  7. 2025Evolving Language Models without Labels: Majority Drives Selection, Novelty Promotes Variationcited by 44
  8. 2024In-Context Sharpness as Alerts: An Inner Representation Perspective for Hallucination Mitigationcited by 43
  9. 2025MCTS-Judge: Test-Time Scaling in LLM-as-a-Judge for Code Correctness Evaluationcited by 32
  10. 2026Towards Autonomous Mathematics Researchcited by 31
  11. 2025Diverse Preference Learning for Capabilities and Alignmentcited by 25
  12. 2025MARS-SQL: A multi-agent reinforcement learning framework for Text-to-SQLcited by 14
  13. 2025More Bang for the Buck: Process Reward Modeling with Entropy-Driven Uncertaintycited by 10
  14. 2025Self-Guided Process Reward Optimization with Redefined Step-wise Advantage for Process Reinforcement Learningcited by 10
  15. 2025GroundedPRM: Tree-Guided and Fidelity-Aware Process Reward Modeling for Step-Level Reasoningcited by 10
  16. 2025SCAN: Self-Denoising Monte Carlo Annotation for Robust Process Reward Learningcited by 6
  17. 2025ClinTutor-R1: Advancing Scalable and Robust One-to-Many Alignment in Clinical Socratic Educationcited by 5
  18. 2026Entropy-Gated Selective Policy Optimization:Token-Level Gradient Allocation for Hybrid Training of Large Language Modelscited by 3
  19. 2026Towards Robust Process Reward Modeling via Noise-aware Learningcited by 2
  20. 2026Trust in One Round: Confidence Estimation for Large Language Models via Structural Signalscited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Towards Better Exploration in Sequential Test-Time Scaling
  2. 2026EDIS: Diagnosing LLM Reasoning via Entropy Dynamics
  3. 2026On the Power of (Approximate) Reward Models for Inference-Time Scaling
  4. 2026Towards Physical Intuitions for Alignment Dynamics: A Case Study With Randomness Crystallization
  5. 2025Rethinking Fine-Tuning when Scaling Test-Time Compute: Limiting Confidence Improves Mathematical Reasoning
  6. 2025The Sequential Edge: Inverse-Entropy Voting Beats Parallel Self-Consistency at Matched Compute
  7. 2025Parallel Test-Time Scaling for Latent Reasoning Models
  8. 2026From Curiosity to Caution: Mitigating Reward Hacking for Best-of-N with Pessimism