Adaptive Test-Time Reasoning via Reward-Guided Dual-Phase Search
2509.25420v1
cs.AI, cs.CL, cs.LG
2025-10-02
Авторы:
Yingqian Cui, Zhenwei Dai, Pengfei He, Bing He, Hui Liu, Xianfeng Tang, Jingying Zeng, Suhang Wang, Yue Xing, Jiliang Tang, Benoit Dumoulin
Abstract
Large Language Models (LLMs) have achieved significant advances in reasoning
tasks. A key approach is tree-based search with verifiers, which expand
candidate reasoning paths and use reward models to guide pruning and selection.
Although effective in improving accuracy, these methods are not optimal in
terms of efficiency: they perform simple decomposition on the reasoning
process, but ignore the planning-execution nature of tasks such as math
reasoning or code generation. This results in inefficient exploration of
reasoning process. To address this, we propose a dual-phase test-time scaling
framework that explicitly separates reasoning into planning and execution, and
performs search over the two phases individually. Specifically, we decompose
reasoning trajectories and develop reward models for each phase, enabling the
search to explore and prune plans and executions separately. We further
introduce a dynamic budget allocation mechanism that adaptively redistributes
sampling effort based on reward feedback, allowing early stopping on confident
steps and reallocation of computation to more challenging parts of the
reasoning process. Experiments on both mathematical reasoning and code
generation benchmarks demonstrate that our approach consistently improves
accuracy while reducing redundant computation.
Ссылки и действия
Дополнительные ресурсы: