학술논문

Graph Neural Networks in TensorFlow and Keras with Spektral [Application Notes]
Document Type
Periodical
Source
IEEE Computational Intelligence Magazine IEEE Comput. Intell. Mag. Computational Intelligence Magazine, IEEE. 16(1):99-106 Feb, 2021
Subject
Computing and Processing
Signal Processing and Analysis
Communication, Networking and Broadcast Technologies
Graph neural networks
Predictive models
Biological system modeling
Computational modeling
Deep learning
Benchmark testing
Language
ISSN
1556-603X
1556-6048
Abstract
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-friendliness 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.