用强化学习和多智能体搜索提升大模型自动证明能力。
Scaling up Multi-Turn Off-Policy RL and Multi-Agent Tree Search for LLM Step-Provers
- 训练时用多轮离策略强化学习持续优化模型
- 推理时通过分层规划缩小搜索空间,实现高效并行求证
- 在数学证明任务上达到95.08%准确率,适合复杂推理场景
将大语言模型(LLM)融入自动定理证明展现出巨大潜力,但受限于训练与推理阶段的可扩展性问题。本文提出 exttt{BFS-Prover-V2} 系统,应对这一双重挑战。首先,设计了一种受 AlphaZero 启发的多轮离策略强化学习框架,通过分阶段专家迭代、自适应策略级数据过滤与周期性重训练,突破长期训练中的性能瓶颈。其次,提出一种增强型多智能体搜索架构,在推理阶段利用通用推理模型作为高层规划器,将复杂定理逐步分解为简单子目标,大幅压缩搜索空间,并通过共享证明缓存实现多个证明代理的高效协作。该方法在经典形式数学基准测试中表现优异:在 MiniF2F 上达到 95.08\\% 准确率,在 ProofNet 上达到 41.4\\"。尽管应用于形式数学领域,其强化学习与推理技术亦适用于其他需要长程多轮推理与复杂搜索的任务。
原文摘要 · Abstract (English)
The integration of Large Language Models (LLMs) into automated theorem proving has shown immense promise, yet is fundamentally constrained by challenges in scaling up both training-time reinforcement learning (RL) and inference-time compute. This paper introduces \texttt{BFS-Prover-V2}, a system designed to address this dual scaling problem. We present two primary innovations. The first is a novel multi-turn off-policy RL framework for continually improving the performance of LLM step-prover at training time. This framework, inspired by the principles of AlphaZero, utilizes a multi-stage expert iteration pipeline featuring adaptive tactic-level data filtering and periodic retraining to surmount the performance plateaus that typically curtail long-term RL in LLM-based agents. The second innovation is a planner-enhanced multi-agent search architecture that scales reasoning capabilities at inference time. This architecture employs a general reasoning model as a high-level planner to iteratively decompose complex theorems into a sequence of simpler subgoals. This hierarchical approach substantially reduces the search space, enabling a team of parallel prover agents to collaborate efficiently by leveraging a shared proof cache. We demonstrate that this dual approach to scaling yields state-of-the-art results on established formal mathematics benchmarks. \texttt{BFS-Prover-V2} achieves 95.08\% and 41.4\% on the MiniF2F and ProofNet test sets respectively. While demonstrated in the domain of formal mathematics, the RL and inference techniques presented in this work are of broader interest and may be applied to other domains requiring long-horizon multi-turn reasoning and complex search.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。