Graph Neural Networks in TensorFlow and Keras with Spektral

被引:107
作者
Grattarola, Daniele [1 ]
Alippi, Cesare [1 ,2 ]
机构
[1] Univ Svizzera Italiana, Lugano, Switzerland
[2] Politecn Milan, Milan, Italy
关键词
Graph neural networks;
D O I
10.1109/MCI.2020.3039072
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Graph neural networks have enabled the application of deep learning to problems that can be described by graphs, which are found throughout the different fields of science, from physics to biology, natural language processing, telecommunications or medicine. In this paper we present Spektral, an open-source Python library for building graph neural networks with TensorFlow and the Keras application programming interface. Spektral implements a large set of methods for deep learning on graphs, including message-passing and pooling operators, as well as utilities for processing graphs and loading popular benchmark datasets. The purpose of this library is to provide the essential building blocks for creating graph neural networks, focusing on the guiding principles of user-fr iendliness and quick prototyping on which Keras is based. Spektral is, therefore, suitable for absolute beginners and expert deep learning practitioners alike. In this work, we present an overview of Spektral's features and report the performance of the methods implemented by the library in scenarios of node classification, graph classification, and graph regression.
引用
收藏
页码:99 / 106
页数:8
相关论文
共 53 条
  • [1] Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
  • [2] Allamanis M., 2017, ARXIV171100740
  • [3] [Anonymous], 2019, PROC ICLR WORKSHOP R
  • [4] [Anonymous], 2018, ARXIV180809920
  • [5] [Anonymous], 2018, P INT C MACH LEARN
  • [6] Battaglia P. W., 2018, ARXIV 180601261
  • [7] Battaglia PW, 2016, ADV NEUR IN, V29
  • [8] Bianchi F. M., 2019, ARXIV190101343
  • [9] Brockschmidt M, 2018, ARXIV181101824
  • [10] Chollet F., 2015, KERAS 20 COMPUTER SO