Learning Convolutional Neural Networks for Graphs

17 May 2016  ·  Mathias Niepert, Mohamed Ahmed, Konstantin Kutzkov ·

Numerous important problems can be framed as learning from graph data. We propose a framework for learning convolutional neural networks for arbitrary graphs. These graphs may be undirected, directed, and with both discrete and continuous node and edge attributes. Analogous to image-based convolutional networks that operate on locally connected regions of the input, we present a general approach to extracting locally connected regions from graphs. Using established benchmark data sets, we demonstrate that the learned feature representations are competitive with state of the art graph kernels and that their computation is highly efficient.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Graph Classification COX2 PSCN Accuracy(10-fold) 75.21 # 3
Graph Classification D&D PSCN Accuracy 76.27% # 32
Graph Classification IMDb-B PSCN Accuracy 71.00% # 34
Graph Classification MUTAG PATCHY-SAN Accuracy 92.63% # 12
Graph Classification MUTAG PSCN Accuracy 88.95% # 30
Graph Classification NCI1 PSCN Accuracy 76.34% # 36
Graph Classification PTC PATCHY-SAN Accuracy 60.00% # 34

Methods


No methods listed for this paper. Add relevant methods here