mlsanityData to 7 Oct 2026

Reward-Oracle MCTS for Formal Theorem Proving: Sample-Efficient Search and the Need for Kernel-Level Proof Auditing

arXiv 2608.28639 ↗First posted 11 Aug 2026cs.AI cs.LG cs.LOcited by 0

Not on the map. Why?

Abstract (arXiv)

Formal theorem proving with large language models remains challenging due to the difficulty of navigating large proof search spaces efficiently. Existing tree search approaches either feed verbose compiler error messages directly into the generation context, increasing context usage during search, or employ non-standard evaluation protocols that prevent direct comparison with established baselines. We propose a three-role Monte Carlo Tree Search (MCTS) framework that treats the Lean 4 compiler purely as a reward oracle using compiler output as a scalar signal for UCB-guided tree updates without feeding error content into the generation context. Our framework decomposes proof search into three roles: a generator for proof attempts, a decomposer for subgoal decomposition, and a critic for subgoal quality evaluation. We evaluate across 4 benchmarks spanning competition mathematics and physics (MiniF2F, PutnamBench, LeanPhysBench, PhysLeandata) with three prover models at standard proof attempt budgets (PAB@16 to PAB@256). Our method achieves 87.1% on MiniF2F with Goedel-Prover-V2-8B at PAB@256 and solves 26/659 PutnamBench problems at PAB@32 surpassing base sampling 18/659 at same proof attempt budget. Through an exhaustive axiom-level audit of every compiled proof, we further identify reward hacking in search-based theorem proving: DeepSeek-Prover-V2-7B produces proofs on PutnamBench that pass compilation and the standard sorry-token scan while depending on sorryAx. The audit removes 4 and 8 such proofs from whole-proof sampling at PAB@32 and PAB@128, and 11 and 19 from MCTS. We do not attribute these counts to the search procedure; we report them to establish that kernel-level auditing is necessary for compiler-verified evaluation.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 16 of its 29 references that are in the map. First posted on arXiv 11 Aug 2026.

Builds on

16 of 29 references.

References · 16

  1. 2022Scaling Laws for Reward Model Overoptimizationcited by 1,056
  2. 2023Reasoning with Language Model is Planning with World Modelcited by 846
  3. 2021MiniF2F: a cross-system benchmark for formal Olympiad-level mathematicscited by 380
  4. 2020Generative Language Modeling for Automated Theorem Provingcited by 379
  5. 2023Alphazero-like Tree-Search can Guide Large Language Model Decoding and Trainingcited by 291
  6. 2022Draft, Sketch, and Prove: Guiding Formal Theorem Provers with Informal Proofscited by 278
  7. 2022HyperTree Proof Search for Neural Theorem Provingcited by 192
  8. 2024PutnamBench: Evaluating Neural Theorem-Provers on the Putnam Mathematical Competitioncited by 153
  9. 2025Goedel-Prover-V2: Scaling Formal Theorem Proving with Scaffolded Data Synthesis and Self-Correctioncited by 140
  10. 2025BFS-Prover: Scalable Best-First Tree Search for LLM-based Automatic Theorem Provingcited by 82
  11. 2023An In-Context Learning Agent for Formal Theorem-Provingcited by 78
  12. 2025STP: Self-play LLM Theorem Provers with Iterative Conjecturing and Provingcited by 77
4 more
  1. 2018Reinforcement Learning of Theorem Provingcited by 69
  2. 2024Lean-STaR: Learning to Interleave Thinking and Provingcited by 64
  3. 2024InternLM2.5-StepProver: Advancing Automated Theorem Proving via Critic-Guided Searchcited by 34
  4. 2025ProofOptimizer: Training Language Models to Simplify Proofs without Human Demonstrationscited by 10

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Direct Optimization of Generators for Search in Automated Theorem Proving
  2. 2026TheoremBench: Evaluating LLMs on Theorem Proving in Formal Mathematics
  3. 2026TCSAlgBench: Benchmarking Automated Proving for Research-Level Theoretical Computer Science
  4. 2025A Combinatorial Identities Benchmark for Theorem Proving via Automated Theorem Generation
  5. 2025MPS-Prover: Advancing Stepwise Theorem Proving by Multi-Perspective Search and Data Curation
  6. 2026Compile to Compress: Boosting Formal Theorem Provers by Compiler Outputs
  7. 2025Bourbaki: Self-Generated and Goal-Conditioned MDPs for Theorem Proving
  8. 2026Learning to Disprove: Formal Counterexample Generation with Large Language Models