中文

Autobahn:基于自同构的图神经网络

机器学习 2022-02-04 v3

摘要

我们提出基于自同构的图神经网络(Autobahn),一类新的图神经网络。在 Autobahn 中,我们将图分解为子图的集合,并应用对每个子图自同构群等变的局部卷积。对局部邻域与子图的特定选择可恢复现有架构,如消息传递神经网络。我们的形式体系也包含新颖架构:作为一个例子,我们引入一种将图分解为路径与圈的图神经网络。所得卷积反映了图各部分可变换的自然方式,在保持卷积直观含义的同时不牺牲全局置换等变性。我们通过将 Autobahn 应用于分子图来验证我们的方法,其取得了与最先进消息传递算法相竞争的结果。

关键词

引用

@article{arxiv.2103.01710,
  title  = {Autobahn: Automorphism-based Graph Neural Nets},
  author = {Erik Henning Thiede and Wenda Zhou and Risi Kondor},
  journal= {arXiv preprint arXiv:2103.01710},
  year   = {2022}
}

备注

10 pages, 5 figures