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 条
  • [51] Zambaldi V., 2018, Relational deep reinforcement learning
  • [52] Zhang M., 2018, P AAAI C ART INT
  • [53] Predicting multicellular function through multi-layer tissue networks
    Zitnik, Marinka
    Leskovec, Jure
    [J]. BIOINFORMATICS, 2017, 33 (14) : I190 - I198