Learning to Explore the Crux in Formal Theorem Proving

Linhan Li, Wenyao Cui, Gong-Duo Zhang, Lihong Gu, Lin Zhou, Jianyun Shang, Huaping Zhang


Abstract
Recent advancements in the application of Large Language Models (LLMs) to formal theorem proving in Lean have unveiled new prospects in complex logical reasoning scenarios. Acknowledging that the difficulty across reasoning steps in a formal theorem proof is not uniform, prior proof search algorithms have not systematically investigated which proof steps are more crucial to explore. This paper introduces Entropy-Based Best-First Search (EBFS), an entropy-inspired heuristic tree search algorithm that enhances the efficiency of proof generation models to explore proof trees, thereby increasing the probability of finding the correct proof. We also introduce a reinforcement fine-tuning approach specifically for proof tree search. Using EBFS during the generation stage further boosts the model’s performance in proof generation. Evaluations demonstrate that our fine-tuned EBFS model achieves consistent improvements over the baseline models. Under the same sample budget, it outperforms whole proof generation by 3.2% on the ProofNet testset, with further gains of 1.9% on MiniF2F and 1.5% on ProverBench.
Anthology ID:
2026.emnlp-main.1932
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
43958–43979
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1932/
DOI:
Bibkey:
Cite (ACL):
Linhan Li, Wenyao Cui, Gong-Duo Zhang, Lihong Gu, Lin Zhou, Jianyun Shang, and Huaping Zhang. 2026. Learning to Explore the Crux in Formal Theorem Proving. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 43958–43979, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Learning to Explore the Crux in Formal Theorem Proving (Li et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1932.pdf