Learning Lean Feedback for Efficient Proof-Candidate Evaluation on GPUs
Developed neural evaluators trained on 4.84M labelled Lean programs to predict proof validity and compiler feedback on GPUs. A two-stage evaluator achieved 87.8% accuracy on 68,932 proof candidates and processed 3.5 candidates per second on a single H100 GPU. Hybrid rewards combining model predictions with Lean checks shortened GRPO training steps by approximately 31%, with comparable miniF2F pass@1 to Lean-only rewards (62.2% vs. 62.4%).