Graph2Tac: Online Representation Learning of Formal Math Concepts

📄 arXiv: 2401.02949v3 📥 PDF

作者: Lasse Blaauwbroek, Miroslav Olšák, Jason Rute, Fidel Ivan Schaposnik Massolo, Jelle Piepenbrock, Vasily Pestun

分类: cs.LG, cs.AI

发布日期: 2024-01-05 (更新: 2024-06-23)

备注: 31 pages


💡 一句话要点

提出Graph2Tac以提升形式数学概念的在线学习能力

🎯 匹配领域: 支柱二:RL算法与架构 (RL & Architecture)

关键词: 在线学习 图神经网络 形式数学 定理证明 层次表示 Coq证明助手 机器学习

📋 核心要点

  1. 现有的离线学习方法在处理未见数学环境时表现不佳,无法充分利用最近的证明信息。
  2. 本文提出的Graph2Tac通过在线学习技术,利用数学概念的局部性特征,构建层次化表示以提升求解能力。
  3. 实验结果表明,Graph2Tac在定理证明上实现了1.5倍的提升,结合k近邻求解器后,整体性能提升达到1.27倍。

📝 摘要(中文)

在证明助手中,两个形式数学概念的物理接近性是其相互相关性的强预测因子。本文展示了如何利用这种局部性特征,通过在线学习技术,开发出在未见数学环境中证明定理的求解代理,显著超越离线学习者。我们在Coq证明助手的Tactician平台上广泛基准测试了两种在线求解器:首先,Tactician的在线k近邻求解器在证明定理方面表现出1.72倍的提升;其次,我们引入了Graph2Tac图神经网络,采用新颖的方法构建新定义的层次表示,其在线定义任务实现了1.5倍的提升。两者结合后,性能提升达到1.27倍,均优于其他通用证明器。

🔬 方法详解

问题定义:本文旨在解决现有离线学习方法在未见数学环境中证明定理的不足,尤其是无法有效利用最近证明信息的问题。

核心思路:通过在线学习技术,利用数学概念之间的局部性特征,构建层次化表示,从而提升求解器的性能。设计上强调实时学习和适应性,以便快速响应新定义和定理。

技术框架:整体架构包括两个主要模块:在线k近邻求解器和Graph2Tac图神经网络。前者通过学习最近的证明来增强求解能力,后者则专注于新定义的层次表示构建。

关键创新:Graph2Tac的创新在于其独特的层次化表示构建方法,能够有效捕捉数学概念之间的关系,与传统的离线学习方法相比,显著提升了求解效率。

关键设计:在网络结构上,Graph2Tac采用图神经网络架构,结合特定的损失函数来优化层次表示的学习过程。关键参数设置经过实验验证,以确保模型的稳定性和高效性。

🖼️ 关键图片

img_0

📊 实验亮点

实验结果显示,Graph2Tac在定理证明方面实现了1.5倍的提升,结合k近邻求解器后,整体性能提升达到1.27倍。两者均优于CoqHammer、Proverbot9001和基于变换器的基线,提升幅度至少为1.48倍,展现了其强大的实用性。

🎯 应用场景

该研究的潜在应用领域包括自动定理证明、数学教育和智能辅导系统等。通过提升形式数学概念的在线学习能力,能够为用户提供更高效的证明工具,促进数学研究和学习的进步。未来,Graph2Tac的技术可能在其他领域的知识表示和推理中发挥重要作用。

📄 摘要(原文)

In proof assistants, the physical proximity between two formal mathematical concepts is a strong predictor of their mutual relevance. Furthermore, lemmas with close proximity regularly exhibit similar proof structures. We show that this locality property can be exploited through online learning techniques to obtain solving agents that far surpass offline learners when asked to prove theorems in an unseen mathematical setting. We extensively benchmark two such online solvers implemented in the Tactician platform for the Coq proof assistant: First, Tactician's online $k$-nearest neighbor solver, which can learn from recent proofs, shows a $1.72\times$ improvement in theorems proved over an offline equivalent. Second, we introduce a graph neural network, Graph2Tac, with a novel approach to build hierarchical representations for new definitions. Graph2Tac's online definition task realizes a $1.5\times$ improvement in theorems solved over an offline baseline. The $k$-NN and Graph2Tac solvers rely on orthogonal online data, making them highly complementary. Their combination improves $1.27\times$ over their individual performances. Both solvers outperform all other general-purpose provers for Coq, including CoqHammer, Proverbot9001, and a transformer baseline by at least $1.48\times$ and are available for practical use by end-users.