Sean Lamont
PhD Candidate, The Australian National University
Research Scientist, Defence Science and Technology Group
My current research focus is on AI/ML applications to Interactive Theorem Proving (ITP).
Selected Publications
- 3D-Prover: Diversity Driven Theorem Proving With Determinantal Point ProcessesSean Lamont, Christian Walder, Amir Dezfouli, Paul Montague, and Michael NorrishNeurIPS, 2025
A key challenge in automated formal reasoning is the intractable search space, which grows exponentially with the depth of the proof. This branching is caused by the large number of candidate proof tactics which can be applied to a given goal. Nonetheless, many of these tactics are semantically similar or lead to an execution error, wasting valuable resources in both cases. We address the problem of effectively pruning this search, using only synthetic data generated from previous proof attempts. We first demonstrate that it is possible to generate semantically aware tactic representations which capture the effect on the proving environment, likelihood of success, and execution time. We then propose a novel filtering mechanism which leverages these representations to select semantically diverse and high quality tactics, using Determinantal Point Processes. Our approach, 3D- Prover, is designed to be general, and to augment any underlying tactic generator. We demonstrate the effectiveness of 3D-Prover on the miniF2F and LeanDojo benchmarks by augmenting popular open source proving LLMs. We show that our approach leads to an increase in the overall proof rate, as well as a significant improvement in the tactic success rate, execution time and diversity. We make our code available at https://github.com/sean-lamont/3D-Prover.
@article{lamont20243d, title = {3D-Prover: Diversity Driven Theorem Proving With Determinantal Point Processes}, author = {Lamont, Sean and Walder, Christian and Dezfouli, Amir and Montague, Paul and Norrish, Michael}, journal = {NeurIPS}, year = {2025}, url = {https://github.com/sean-lamont/3D-Prover}, } - BAIT: Benchmarking (Embedding) Architectures for Interactive Theorem-ProvingSean Lamont, Michael Norrish, Amir Dezfouli, Christian Walder, and Paul MontagueIn Proceedings of the AAAI Conference on Artificial Intelligence, 2024
Artificial Intelligence for Theorem Proving has given rise to a plethora of benchmarks and methodologies, particularly in Interactive Theorem Proving (ITP). Research in the area is fragmented, with a diverse set of approaches being spread across several ITP systems. This presents a significant chal- lenge to the comparison of methods, which are often complex and difficult to replicate. Addressing this, we present BAIT, a framework for fair and streamlined comparison of learning approaches in ITP. We demonstrate BAIT’s capabilities with an in-depth comparison, across several ITP benchmarks, of state-of-the-art architec- tures applied to the problem of formula embedding. We find that Structure Aware Transformers perform particularly well, improving on techniques associated with the original prob- lem sets. BAIT also allows us to assess the end-to-end prov- ing performance of systems built on interactive environments. This unified perspective reveals a novel end-to-end system that improves on prior work. We also provide a qualitative analysis, illustrating that improved performance is associated with more semantically-aware embeddings. By streamlining the implementation and comparison of Machine Learning al- gorithms in the ITP context, we anticipate BAIT will be a springboard for future research.
@inproceedings{lamont2024bait, title = {BAIT: Benchmarking (Embedding) Architectures for Interactive Theorem-Proving}, author = {Lamont, Sean and Norrish, Michael and Dezfouli, Amir and Walder, Christian and Montague, Paul}, booktitle = {Proceedings of the AAAI Conference on Artificial Intelligence}, volume = {38}, number = {9}, pages = {10607--10615}, year = {2024}, } - Generalised Discount Functions applied to a Monte-Carlo AImu ImplementationSean Lamont, John Aslanides, Jan Leike, and Marcus HutterAutonomous Agents and Multiagent Systems, 2017
In recent years, work has been done to develop the theory of General Reinforcement Learning (GRL). However, there are few examples demonstrating the known results regarding generalised discounting. We have added to the GRL simulation platform AIXIjs the functionality to assign an agent arbitrary discount functions, and an environment which can be used to determine the effect of discounting on an agent’s policy. Using this, we investigate how geometric, hyperbolic and power discounting affect an informed agent in a simple MDP. We experimentally reproduce a number of theoretical results, and discuss some related subtleties. It was found that the agent’s behaviour followed what is expected theoretically, assuming appropriate parameters were chosen for the Monte-Carlo Tree Search (MCTS) planning algorithm.
@article{lamont2017generalised, title = {Generalised Discount Functions applied to a Monte-Carlo AImu Implementation}, author = {Lamont, Sean and Aslanides, John and Leike, Jan and Hutter, Marcus}, journal = {Autonomous Agents and Multiagent Systems}, pages = {1589-1591}, year = {2017}, url = {https://www.ifaamas.org/Proceedings/aamas2017/pdfs/p1589.pdf}, publisher = {International Foundation for Autonomous Agents and Multiagent Systems}, }