Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering

In this work, we are interested in generalizing convolutional neural networks (CNNs) from low-dimensional regular grids, where image, video and speech are represented, to high-dimensional irregular domains, such as social networks, brain connectomes or words' embedding, represented by graphs. We present a formulation of CNNs in the context of spectral graph theory, which provides the necessary mathematical background and efficient numerical schemes to design fast localized convolutional filters on graphs. Importantly, the proposed technique offers the same linear computational complexity and constant learning complexity as classical CNNs, while being universal to any graph structure. Experiments on MNIST and 20NEWS demonstrate the ability of this novel deep learning system to learn local, stationary, and compositional features on graphs.

PDF Abstract NeurIPS 2016 PDF NeurIPS 2016 Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Node Classification CiteSeer (0.5%) ChebyNet Accuracy 45.3% # 10
Node Classification CiteSeer (1%) ChebyNet Accuracy 59.4% # 10
Node Classification CiteSeer with Public Split: fixed 20 nodes per class ChebyNet Accuracy 70.1% # 32
Node Classification Cora (0.5%) ChebyNet Accuracy 33.9% # 15
Node Classification Cora (1%) ChebyNet Accuracy 44.2% # 15
Node Classification Cora (3%) ChebyNet Accuracy 62.1% # 14
Node Classification Cora with Public Split: fixed 20 nodes per class ChebyNet Accuracy 78.0% # 31
Graph Property Prediction ogbg-molpcba ChebNet Test AP 0.2306 ± 0.0016 # 30
Validation AP 0.2372 ± 0.0018 # 28
Number of params 1475003 # 29
Ext. data No # 1
Node Classification PubMed (0.03%) ChebyNet Accuracy 45.3% # 14
Node Classification PubMed (0.05%) ChebyNet Accuracy 48.2% # 14
Node Classification PubMed (0.1%) ChebyNet Accuracy 55.2% # 14
Node Classification PubMed with Public Split: fixed 20 nodes per class ChebyNet Accuracy 69.8% # 36
Skeleton Based Action Recognition SBU ChebyNet Accuracy 96.00% # 4

Results from Other Papers


Task Dataset Model Metric Name Metric Value Rank Source Paper Compare
Node Classification Citeseer ChebNet Accuracy 69.8% # 62
Node Classification Cora ChebNet Accuracy 81.2% # 62
Node Classification Pubmed ChebNet Accuracy 74.4% # 63

Methods