TacticZero: Learning to Prove Theorems from Scratch with Deep Reinforcement Learning
Minchao Wu, Michael Norrish, Christian Walder, Amir Dezfouli
Abstract
We propose a novel approach to interactive theorem-proving (ITP) using deep reinforcement learning. The proposed framework is able to learn proof search strategies as well as tactic and arguments prediction in an end-to-end manner. We formulate the process of ITP as a Markov decision process (MDP) in which each state represents a set of potential derivation paths. This structure allows us to introduce a novel backtracking mechanism which enables the agent to efficiently discard (predicted) dead-end derivations and restart from promising alternatives. We implement the framework in the HOL4 theorem prover. Experimental results show that the framework outperforms existing automated theorem provers (i.e. hammers) available in HOL4 when evaluated on unseen problems. We further elaborate the role of key components of the framework using ablation studies. Preprint. Under review.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers10
- Autoformalization with Large Language ModelsYuhuai Wu, Albert Qiaochu Jiang, Wenda Li, Markus N. Rabe et al.NeurIPS 2022 · 364 citations
- Baldur: Whole-Proof Generation and Repair with Large Language ModelsEmily First, Markus N. Rabe, Talia Ringer, Yuriy BrunFSE 2023 · 89 citations
- Diversity-Driven Automated Formal VerificationEmily First, Yuriy BrunICSE 2022 · 28 citations
- Rango: Adaptive Retrieval-Augmented Proving for Automated Software VerificationKyle Thompson, Nuno Saavedra, Pedro Carrott, Kevin Fisher et al.ICSE 2025 · 4 citations
- QEDCartographer: Automating Formal Verification Using Reward-Free Reinforcement LearningAlex Sanchez-Stern, Abhishek Varghese, Zhanna Kaufman, Shizhuo Dylan Zhang et al.ICSE 2025 · 2 citations
Builds on3
- Proof Artifact Co-Training for Theorem Proving with Language ModelsJesse Michael Han, Jason Rute, Yuhuai Wu, Edward W. Ayers et al.ICLR 2022 · 149 citations
- TacTok: semantics-aware proof synthesisEmily First, Yuriy Brun, Arjun GuhaOOPSLA 2020 · 39 citations
- Learning Heuristics for Quantified Boolean Formulas through Reinforcement LearningGil Lederman, Markus N. Rabe, Sanjit A. Seshia, Edward A. LeeICLR 2020 · 3 citations
Related papers
- A Deep Reinforcement Learning Approach to First-Order Logic Theorem ProvingMaxwell Crouse, Ibrahim Abdelaziz, Bassem Makni, Spencer Whitehead et al.AAAI 2021 · 41 citations
- BAIT: Benchmarking (Embedding) Architectures for Interactive Theorem-ProvingSean Lamont, Michael Norrish, Amir Dezfouli, Christian Walder et al.AAAI 2024 · 3 citations
- Lean-Auto: An Interface Between Lean 4 and Automated Theorem ProversYicheng Qian, Joshua Clune, Clark W. Barrett, Jeremy AvigadCAV 2025 · 6 citations
- Thor: Wielding Hammers to Integrate Language Models and Automated Theorem ProversAlbert Qiaochu Jiang, Wenda Li, Szymon Tworkowski, Konrad Czechowski et al.NeurIPS 2022 · 154 citations
- Graph Representations for Higher-Order Logic and Theorem ProvingAditya Paliwal, Sarah M. Loos, Markus N. Rabe, Kshitij Bansal et al.AAAI 2020 · 110 citations
