用蒙特卡洛树搜索提升强化学习的探索效率,突破推理模型训练瓶颈。
DeepSearch: Overcome the Bottleneck of Reinforcement Learning with Verifiable Rewards via Monte Carlo Tree Search
- 将树搜索嵌入训练过程,实现系统性路径探索。
- 在数学推理任务上达62.95%准确率,节省5.7倍显卡时间。
- 适合追求高效推理训练的研究者与工程师。
尽管基于验证奖励的强化学习(RLVR)已成为语言模型发展高级推理能力的关键组件,但现有研究发现,在数千次优化步骤后会出现训练停滞现象,即性能提升显著放缓,尽管计算投入持续增加。这一限制源于当前RLVR实践中探索稀疏的问题:模型依赖有限的回溯推演,常遗漏关键推理路径,无法系统覆盖解空间。我们提出DeepSearch框架,将蒙特卡洛树搜索(MCTS)直接融入RLVR训练流程。与仅在推理阶段使用树搜索的方法不同,DeepSearch在训练中嵌入结构化搜索,实现对推理步骤的系统性探索与细粒度信用分配。通过训练时探索,解决了因探索不足导致的性能增益衰减问题。主要贡献包括:(1) 全局前沿节点选择策略,优先关注搜索树中高潜力节点;(2) 基于熵的引导选择,识别高置信度推理路径用于监督;(3) 自适应重放缓冲与解缓存机制以提升效率。在数学推理基准测试中,DeepSearch实现平均准确率62.95%,达到新的最优水平,且相比延长训练方法减少5.7倍GPU小时。结果表明,战略性探索优于盲目扩展算力,凸显算法创新对推进RLVR方法的重要性。DeepSearch为通过系统性搜索而非冗长计算来扩展推理能力开辟了新方向。
原文摘要 · Abstract (English)
Although RLVR has become an essential component for developing advanced reasoning skills in language models, contemporary studies have documented training plateaus after thousands of optimization steps, i.e., notable decreases in performance gains despite increased computational investment. This limitation stems from the sparse exploration patterns inherent in current RLVR practices, where models rely on limited rollouts that often miss critical reasoning paths and fail to provide systematic coverage of the solution space. We present DeepSearch, a framework that integrates Monte Carlo Tree Search (MCTS) directly into RLVR training. In contrast to existing methods that rely on tree search only at inference, DeepSearch embeds structured search into the training loop, enabling systematic exploration and fine-grained credit assignment across reasoning steps. Through training-time exploration, DeepSearch addresses the fundamental bottleneck of insufficient exploration, which leads to diminishing performance gains over prolonged training. Our contributions include: (1) a global frontier selection strategy that prioritizes promising nodes across the search tree, (2) selection with entropy-based guidance that identifies confident paths for supervision, and (3) adaptive replay buffer training with solution caching for efficiency. Experiments on mathematical reasoning benchmarks show that DeepSearch achieves an average accuracy of 62.95\% and establishes a new state-of-the-art reasoning model, while using 5.7x fewer GPU hours than extended training approaches. These results highlight the importance of strategic exploration over brute-force scaling and demonstrate the promise of algorithmic innovation for advancing RLVR methodologies. DeepSearch establishes a new direction for scaling reasoning capabilities through systematic search rather than prolonged computation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。