arXiv:2509.24351cs.AI2025-09

动态自适应搜索提升数学推理数据质量,显著增强大模型能力

From Static to Dynamic: Adaptive Monte Carlo Search for Mathematical Process Supervision

  • 根据推理步骤不确定性动态分配采样资源,优化节点评估效率
  • 采用时序自适应策略从广度探索逐步转向高潜力路径挖掘,提升搜索效果
  • 在20万条数据上训练的奖励模型,使7B模型超越72B模型性能

过程数据的质量对训练过程奖励模型(PRM)至关重要,可提升大语言模型的复杂数学推理能力。现有方法依赖固定预算采样策略,在自动化数据生成中导航庞大搜索空间,导致效率低且缺乏灵活性。为此,我们提出自适应蒙特卡洛搜索(AMCS),将数据生成从静态固定搜索转变为节点评估与路径扩展层面的自适应动态搜索。一方面,AMCS通过向不确定的推理步骤分配更多样本、对易估计步骤减少采样来动态优化评估精度;另一方面,利用具有时序自适应策略的蒙特卡洛算法,初期进行广泛探索,后期聚焦最有前景的方向。基于AMCS,我们构建了约20万条过程监督样本的数据集MathSearch-200K用于训练PRMs。在四个数学推理基准上的实验表明,使用GLM-4-9B时,Qwen2.5-Math-7B-PRM-AMCS在MATH500上达到最高76.2%准确率,优于所有基线模型。值得注意的是,一个由该模型监督的7B模型性能超过未受优监督的72B模型。此外,该模型在分布外问题上仍保持显著优势,展现出强泛化能力。

原文摘要 · Abstract (English)

The quality of process data plays a key role in training a Process Reward Model (PRM), which can enhance the complex mathematical reasoning capability of large language models. Existing methods estimate the quality of reasoning steps based on a fixed-budget sampling strategy and navigate a vast search space to perform path expansion during the automated data generation process, resulting in their inefficiency and inflexibility. To address these issues, we propose Adaptive Monte Carlo Search (AMCS), a framework that transforms data generation from fixed, static to adaptive, dynamic search at the level of node value estimation and path expansion. On one hand, AMCS adaptively refines estimation by allocating more samples to uncertain reasoning steps while using fewer samples for those that are easier to estimate. On the other hand, it enhances the path expansion through a Monte Carlo algorithm with a temporally adaptive policy that begins with broad exploration and gradually shifts toward exploiting the most promising directions. With AMCS, we construct a large-scale dataset MathSearch-200K of about 200K process supervision examples for training PRMs. To verify the effectiveness of our method, we conduct extensive experiments on four mathematical reasoning benchmarks. Experimental results show that Qwen2.5-Math-7B-PRM-AMCS achieves up to 76.2% accuracy on MATH500 with GLM-4-9B, outperforming all baseline PRMs. Notably, a 7B model supervised by Qwen2.5-Math-7B-PRM-AMCS surpasses a 72B model with weaker supervision. Moreover, Qwen2.5-Math-7B-PRM-AMCS maintains consistent advantages on out-of-distribution problems, demonstrating strong generalization capability. Our code is available at https://github.com/reml-group/AMCS.

数学推理强化学习数据生成动态搜索

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。