arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.06560cs.LGstat.ML
收藏

评价图神经网络的逻辑泛化能力

Evaluating Logical Generalization in Graph Neural Networks

Facebook AI Research(Facebook AI研究);McGill University(麦吉尔大学);Montreal Institute of Learning Algorithms (Mila)(蒙特利尔学习算法研究所(Mila));School of Computer Science(计算机科学学院)

Koustuv Sinha, Shagun Sodhani, Joelle Pineau, William L. Hamilton

摘要:

最近的研究强调了关系归纳偏差在构建学习代理中的作用,这些学习代理可以以组合方式进行概括和推理。然而,尽管诸如图神经网络(GNN)等关系学习算法显示出了希望,但我们不了解这些方法如何有效地适应新任务。在这项工作中,我们通过设计一个基于一阶逻辑的基准套件来研究使用GNNs进行逻辑泛化的任务。我们的基准测试套件GraphLog要求学习算法在不同的合成逻辑中执行规则归纳,表示为知识图谱。GraphLog由57个不同逻辑域上的关系预测任务组成。我们使用GraphLog在三种不同的设置中评估GNN:单任务监督学习,多任务预训练和持续学习。与以前的基准测试不同,我们的方法允许我们精确地控制不同任务之间的逻辑关系。我们发现,模型的泛化和适应能力在很大程度上取决于它们在训练过程中遇到的逻辑规则的多样性,我们的研究结果突出了GNN模型设计的新挑战。我们在https://www.cs.mcgill.ca/~ksinha4/graphlog公开发布数据集和用于生成数据集并与数据集交互的代码。

英文摘要:

Recent research has highlighted the role of relational inductive biases in building learning agents that can generalize and reason in a compositional manner. However, while relational learning algorithms such as graph neural networks (GNNs) show promise, we do not understand how effectively these approaches can adapt to new tasks. In this work, we study the task of logical generalization using GNNs by designing a benchmark suite grounded in first-order logic. Our benchmark suite, GraphLog, requires that learning algorithms perform rule induction in different synthetic logics, represented as knowledge graphs. GraphLog consists of relation prediction tasks on 57 distinct logical domains. We use GraphLog to evaluate GNNs in three different setups: single-task supervised learning, multi-task pretraining, and continual learning. Unlike previous benchmarks, our approach allows us to precisely control the logical relationship between the different tasks. We find that the ability for models to generalize and adapt is strongly determined by the diversity of the logical rules they encounter during training, and our results highlight new challenges for the design of GNN models. We publicly release the dataset and code used to generate and interact with the dataset at https://www.cs.mcgill.ca/~ksinha4/graphlog.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑