MUR: Momentum Uncertainty guided Reasoning for Large Language Models
Large Language Models (LLMs) have achieved impressive performance on reasoning-intensive tasks, yet optimizing their reasoning efficiency remains an open challenge. While Test-Time Scaling (TTS) improves reasoning quality, it often leads to overthinking, wasting tokens on redundant computations. This work investigates how to efficiently and adaptively guide LLM test-time scaling without additional training. Inspired by the concept of momentum in physics, we propose Momentum Uncertainty-guided Reasoning (MUR), which dynamically allocates thinking budgets to critical reasoning steps by tracking and aggregating stepwise uncertainty over time. To support flexible inference-time control, we introduce gamma-control, a simple mechanism that tunes the reasoning budget via a single hyperparameter. We provide in-depth theoretical proof to support the superiority of MUR in terms of stability and biases. MUR is comprehensively evaluated against various TTS methods across four challenging benchmarks (MATH-500, AIME24, AIME25, and GPQA-diamond) using different sizes of recent Qwen3 models (1.7B, 4B, and 8B). Results demonstrate that MUR reduces computation by over 50% on average while improving accuracy by 0.62-3.37%.
Authors
- Luu Anh Tuan (ORCID: https://orcid.org/0000-0001-6062-207X)
- Qisheng Lin (ORCID: https://orcid.org/0000-0002-5149-2717)
- Jian Zhang (ORCID: https://orcid.org/0000-0001-8870-515X)
- Xiaofei Wu (ORCID: https://orcid.org/0000-0002-4797-1256)
- Haiteng Zhao
- Yifei Li (ORCID: https://orcid.org/0009-0003-3343-2870)
- Jun Liu (ORCID: https://orcid.org/0000-0002-2504-3651)
- Hang Yan (ORCID: https://orcid.org/0009-0004-5675-0406)
- He-San Luo (ORCID: https://orcid.org/0000-0003-4153-7683)
- Rongman Xu
- Fangzhi Xu
Publication Details
- Journal
- ArXiv.org
- Published
- 2025-07-20
- Primary Topic
- Topic Modeling
- Type
- article
- Field-Weighted Citation Impact
- 0.00