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

图神经网络的基准测试

Benchmarking Graph Neural Networks

Nanyang Technological University(南洋理工大学);University of Cambridge(剑桥大学);Loyola Marymount University(洛约拉马利蒙特大学);Mila(米拉人工智能研究所);University of Montréal(蒙特利尔大学);National University of Singapore(新加坡国立大学)

Vijay Prakash Dwivedi, Chaitanya K. Joshi, Anh Tuan Luu, Thomas Laurent, Yoshua Bengio, Xavier Bresson

摘要:

在过去的几年里,图神经网络(GNN)已经成为分析和学习图上数据的标准工具包。这一新兴领域已经见证了广泛的增长,有前途的技术已成功地应用于计算机科学,数学,生物学,物理学和化学。但是,任何成功的领域要成为主流和可靠的,必须制定基准来量化进展。这促使我们在2020年3月发布了一个基准框架,该框架i)由各种数学和现实世界的图形组成,ii)能够以相同的参数预算进行公平的模型比较,以确定关键架构,iii)具有开源,易于使用和可复制的代码基础设施,iv)研究人员可以灵活地实验新的理论思想。截至2022年12月,GitHub存储库已达到2,000个star和380个fork,通过GNN社区的广泛使用,证明了拟议开源框架的实用性。在本文中,我们提出了我们的基准的更新版本,简要介绍了上述框架特征,一个额外的中型分子数据集AQSOL,类似于流行的ZINC,但具有真实世界的测量化学目标,并讨论了如何利用这个框架来探索新的GNN设计和见解。作为我们的基准的价值的证明,我们研究了GNN中的图位置编码(PE)的情况,这是与此基准一起引入的,并且已经激发了在强大的实验环境中探索更强大的PE用于Transformers和GNN的兴趣。

英文摘要:

In the last few years, graph neural networks (GNNs) have become the standard toolkit for analyzing and learning from data on graphs. This emerging field has witnessed an extensive growth of promising techniques that have been applied with success to computer science, mathematics, biology, physics and chemistry. But for any successful field to become mainstream and reliable, benchmarks must be developed to quantify progress. This led us in March 2020 to release a benchmark framework that i) comprises of a diverse collection of mathematical and real-world graphs, ii) enables fair model comparison with the same parameter budget to identify key architectures, iii) has an open-source, easy-to-use and reproducible code infrastructure, and iv) is flexible for researchers to experiment with new theoretical ideas. As of December 2022, the GitHub repository has reached 2,000 stars and 380 forks, which demonstrates the utility of the proposed open-source framework through the wide usage by the GNN community. In this paper, we present an updated version of our benchmark with a concise presentation of the aforementioned framework characteristics, an additional medium-sized molecular dataset AQSOL, similar to the popular ZINC, but with a real-world measured chemical target, and discuss how this framework can be leveraged to explore new GNN designs and insights. As a proof of value of our benchmark, we study the case of graph positional encoding (PE) in GNNs, which was introduced with this benchmark and has since spurred interest of exploring more powerful PE for Transformers and GNNs in a robust experimental setting.

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

↑