Lean Proof Assistant Enhances Reinforcement Learning for Theorem Proving

Minsu Kim, Se-Young Yun· June 19, 2026 View original

Summary

This paper introduces a method for process-verified reinforcement learning (RLVR) in theorem proving, using the Lean proof assistant to provide dense, fine-grained, and sound feedback. By parsing proof attempts into tactic sequences, Lean offers both outcome-level and tactic-level verified signals, significantly improving performance over outcome-only baselines.

Reinforcement learning from verifiable rewards (RLVR) typically relies on a simple binary signal to indicate success or failure. However, symbolic proof assistants like Lean offer a much richer, structured form of feedback. This research bridges the gap between structured processes and unstructured rewards by demonstrating how the Lean proof assistant can serve as a symbolic process oracle, providing both overall outcome and fine-grained tactic-level verified feedback during the training of theorem-proving agents. Proof attempts are analyzed by parsing them into sequences of tactics. Lean's elaboration process then marks each step as locally sound or identifies the earliest failing step. This provides dense, verifier-grounded credit signals rooted in type theory, offering a more detailed understanding of the proof process than a simple pass/fail. These structured rewards are integrated into a GRPO-style reinforcement learning objective, incorporating first-error propagation and first-token credit methods. Experiments with existing theorem provers like STP-Lean and DeepSeek-Prover-V1.5 show that this tactic-level supervision consistently outperforms baselines that only use outcome-level feedback, leading to improvements on benchmarks such as MiniF2F and ProofNet. The study highlights that symbolic proof assistants can function not just as evaluators but as crucial process-level reward oracles during training, paving the way for more reliable and scalable RL frameworks in formal reasoning.

Why it matters

For professionals in AI research, formal verification, and software engineering, this work offers a powerful new paradigm for training AI systems to perform complex symbolic reasoning tasks. Leveraging structured feedback from proof assistants can lead to more robust and trustworthy automated theorem provers and code verification tools.

How to implement this in your domain

  1. 1Explore integrating symbolic proof assistants like Lean as process-level reward oracles in your reinforcement learning pipelines for formal reasoning tasks.
  2. 2Design reward functions that leverage fine-grained, tactic-level feedback from verification tools, beyond simple binary success signals.
  3. 3Apply first-error propagation and first-token credit methods to balance outcome and process-level advantages in your RL objectives.
  4. 4Evaluate the performance of your RL agents on formal reasoning benchmarks using this enhanced feedback mechanism.

Who benefits

Software EngineeringAI ResearchCybersecurityAcademiaLegalTech

Key takeaways

  • Symbolic proof assistants can provide dense, fine-grained feedback for reinforcement learning in theorem proving.
  • Tactic-level supervision significantly outperforms outcome-only baselines in formal reasoning tasks.
  • Lean can act as a process-level reward oracle, not just an evaluation verifier.
  • This approach combines language model scalability with symbolic verification reliability.

Original post by Minsu Kim, Se-Young Yun

"arXiv:2606.20068v1 Announce Type: new Abstract: While reinforcement learning from verifiable rewards (RLVR) typically has relied on a single binary verification signal, symbolic proof assistants in formal reasoning offer rich, fine-grained structured feedback. This gap between st…"

View on X

Originally posted by Minsu Kim, Se-Young Yun on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses