Process-Verified Reinforcement Learning for Theorem Proving via Lean
Summary (Overview)
-
Core Contribution: This paper introduces a novel reinforcement learning (RL) framework that leverages the Lean proof assistant as a process-level reward oracle, providing both global outcome signals and fine-grained tactic-level rewards during training—moving beyond the conventional use of Lean solely as a binary verifier.
-
Key Innovation: The authors formalize Lean's symbolic feedback into a tactic-level MDP, using a first-error propagation rule and a first-token credit assignment strategy to transform structured proof feedback into dense, sound training signals within a GRPO-style objective.
-
Empirical Results: The method consistently outperforms outcome-only GRPO baselines across MiniF2F and ProofNet benchmarks. For STP-Lean, it achieves up to +2.5%p improvement on MiniF2F (pass@64) and +1.4%p on ProofNet (pass@32), while demonstrating more stable training dynamics.
-
Key Insight: The paper demonstrates that proof assistants can serve not only as verifiers at inference inference time but also as structured## Introduction and Theoretical Foundation
Background## Background and Motivation
The paper addresses a fundamental challenge in reinforcement learning for formal theorem proving: traditional RLVR (Reinforcement Learning from Verifiable Rewards) relies on sparse binary outcome signals, while formal proof assistants like Lean offer rich, structured feedback at each step of proof## 3.1 Lean4 Formalization
The paper formalizes Lean's feedback mechanism through three key functions:
- Parsing function : extracts the sequence of tactics from a proof, sorted by starting position
- Global scoring function ### g: \mathcal{Y}# 3.2 Tactic-Level MDP
The framework models the proof generation process as a tactic-level MDP $\mathcal{M} = (\mathcal{S# 3.3 Credit Assignment in RL
The paper contrasts two RL approaches:
- PPO: Uses Generalized AdvantageSkip to content Advantage Estimation with# Methodology
4.1import numpy as np 1 Defining Tconst T# Tactic-Level Rewards
The## Empirical Validation / Results
The paper# Methodology
The paper defines two complementary reward signals:
- Outcome reward: where if the proof passes Lean verification, 0 otherwise
- Process reward: assigning values:
- for successful tactics in complete# Empirical Validation / Results
The paper provides a detailed analysis of experimental results across multiple dimensions:
- Main Results (Table 1): STP-Lean + Ours achieves 59.2% on MiniF2F## Theoretical and Practical Implications
The paper# Conclusion
The paper concludes with several key takeaways:
Related papers
- Prove2Me: An Open Collaborative Platform for Scaling Math Formalization
Prove2Me enables large-scale formalization by having AI agents prove audited statements on consumer subscriptions, cutting costs three orders of magnitude versus centralized swarms.
- Beyond Solvability: Task Learnability as a Static Prior for LLM RL Post-Training
Learnability, distinct from solvability, is estimated from endpoint reward deltas via TRAJVAL and triples training efficiency in LLM RL post-training.
- Explore More, Drift Less: Outcome-Only Reinforcement Learning Can Suffice for Long-Horizon Interactive Agents
Outcome-only reinforcement learning with CANOPY, a protocol fixing signal starvation and policy drift, lets a single open 14B model top the AppWorld leaderboard.