DeepSeek-Prover-V1.5:利用证明助手反馈进行强化学习与蒙特卡洛树搜索

DeepSeek-Prover-V1.5: Harnessing Proof Assistant Feedback for Reinforcement Learning and Monte-Carlo Tree Search

深度求索 DeepSeek-AI · DeepSeek · 2024-08-15 · arXiv:2408.08152 ↗ · 被引 188

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们推出了 DeepSeek-Prover-V1.5,这是一个专为 Lean 4 定理证明设计的开源语言模型,它通过优化训练和推理过程增强了 DeepSeek-Prover-V1。该模型在 DeepSeek-Math-Base 上预训练,专注于形式化数学语言,然后使用来自 DeepSeek-Prover-V1 的增强形式化定理证明数据集进行监督微调。通过证明助手反馈的强化学习(RLPAF)进一步优化。除了 DeepSeek-Prover-V1 的单次完整证明生成方法外,我们提出了 RMaxTS,这是一种蒙特卡洛树搜索的变体,采用内在奖励驱动的探索策略来生成多样化的证明路径。DeepSeek-Prover-V1.5 在高中水平的 miniF2F 基准测试集(63.5%)和本科水平的 ProofNet 基准测试集(25.3%)上取得了新的最先进结果,相比 DeepSeek-Prover-V1 有显著提升。

We introduce DeepSeek-Prover-V1.5, an open-source language model designed for theorem proving in Lean 4, which enhances DeepSeek-Prover-V1 by optimizing both training and inference processes. Pre-trained on DeepSeekMath-Base with specialization in formal mathematical languages, the model undergoes supervised fine-tuning using an enhanced formal theorem proving dataset derived from DeepSeek-Prover-V1. Further refinement is achieved through reinforcement learning from proof assistant feedback (RLPAF). Beyond the single-pass whole-proof generation approach of DeepSeek-Prover-V1, we propose RMaxTS, a variant of Monte-Carlo tree search that employs an intrinsic-reward-driven exploration strategy to generate diverse proof paths. DeepSeek-Prover-V1.5 demonstrates significant improvements over DeepSeek-Prover-V1, achieving new state-of-the-art results on the test set of the high school level miniF2F benchmark ($63.5\%$) and the undergraduate level ProofNet benchmark ($25.3\%$).

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 18)

阅读逐段中英对照全文 →