Thomas N. Kipf, Max Welling
We present a scalable approach for semi-supervised learning on graph-structured data that is based on an efficient variant of convolutional neural networks which operate directly on graphs. We motivate the choice of our convolutional architecture via a localized first-order approximation of spectral graph convolutions. Our model scales linearly in the number of graph edges and learns hidden layer representations that encode both local graph structure and features of nodes. In a number of experiments on citation networks and on a knowledge graph dataset we demonstrate that our approach outperforms related methods by a significant margin.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Graph Regression | PCQM4Mv2-LSC | Test MAE | 0.1398 | GCN |
| Graph Regression | PCQM4Mv2-LSC | Validation MAE | 0.1379 | GCN |
| Node Classification | Accuracy | 64.6 | GCN_cheby (Kipf and Welling, 2017) | |
| Node Classification | Accuracy | 57.5 | GCN (Kipf and Welling, 2017) | |
| Node Classification | Brazil Air-Traffic | Accuracy | 0.516 | GCN_cheby (Kipf and Welling, 2017) |
| Node Classification | Citeseer | Accuracy | 70.3 | GCN |
| Node Classification | NELL | Accuracy | 66 | GCN |
| Node Classification | Wiki-Vote | Accuracy | 49.5 | GCN_cheby (Kipf and Welling, 2017) |
| Node Classification | Wiki-Vote | Accuracy | 32.9 | GCN (Kipf and Welling, 2017) |
| Node Classification | Pubmed | Accuracy | 79 | GCN |
| Node Classification | Europe Air-Traffic | Accuracy | 46 | GCN_cheby (Kipf and Welling, 2017) |
| Node Classification | Europe Air-Traffic | Accuracy | 37.1 | GCN (Kipf and Welling, 2017) |
| Node Classification | Flickr | Accuracy | 0.546 | GCN (Kipf and Welling, 2017) |
| Node Classification | Flickr | Accuracy | 0.479 | GCN_cheby (Kipf and Welling, 2017) |
| Node Classification | IMDB (Heterogeneous Node Classification) | Macro-F1 | 57.88 | GCN |
| Node Classification | IMDB (Heterogeneous Node Classification) | Micro-F1 | 64.82 | GCN |
| Node Classification | Freebase (Heterogeneous Node Classification) | Macro-F1 | 27.84 | GCN |
| Node Classification | Freebase (Heterogeneous Node Classification) | Micro-F1 | 60.23 | GCN |
| Node Classification | DBLP (Heterogeneous Node Classification) | Macro-F1 | 90.84 | GCN |
| Node Classification | DBLP (Heterogeneous Node Classification) | Micro-F1 | 91.47 | GCN |
| Node Classification | ACM (Heterogeneous Node Classification) | Macro-F1 | 92.17 | GCN |
| Node Classification | ACM (Heterogeneous Node Classification) | Micro-F1 | 92.12 | GCN |
| Link Property Prediction | ogbl-ddi | Number of params | 1421571 | GCN+JKNet |
| Link Property Prediction | ogbl-ddi | Number of params | 1289985 | GCN |
| Link Property Prediction | ogbl-citation2 | Number of params | 296449 | Full-batch GCN |
| Link Property Prediction | ogbl-collab | Number of params | 296449 | GCN (val as input) |
| Link Property Prediction | ogbl-collab | Number of params | 296449 | GCN |
| Link Property Prediction | ogbl-ppa | Number of params | 278529 | GCN |
| Graph Property Prediction | ogbg-molhiv | Number of params | 527701 | GCN |
| Graph Property Prediction | ogbg-molhiv | Number of params | 1978801 | GCN+virtual node |
| Graph Property Prediction | ogbg-molhiv | Number of params | 527701 | GCN (in Julia) |
| Graph Property Prediction | ogbg-code2 | Number of params | 12484310 | GCN+virtual node |
| Graph Property Prediction | ogbg-code2 | Number of params | 11033210 | GCN |
| Graph Property Prediction | ogbg-ppa | Number of params | 1930537 | GCN+virtual node |
| Graph Property Prediction | ogbg-ppa | Number of params | 479437 | GCN |
| Graph Property Prediction | ogbg-molpcba | Number of params | 2017028 | GCN+virtual node |
| Graph Property Prediction | ogbg-molpcba | Number of params | 565928 | GCN |
| Node Property Prediction | ogbn-arxiv | Number of params | 122542 | GCN+residual+6 layers |
| Node Property Prediction | ogbn-arxiv | Number of params | 21885098 | GCN+residual+node2vec |
| Node Property Prediction | ogbn-arxiv | Number of params | 155824 | GCN_res + 8 layers |
| Node Property Prediction | ogbn-arxiv | Number of params | 110120 | GCN |
| Node Property Prediction | ogbn-products | Number of params | 103727 | Full-batch GCN |
| Node Property Prediction | ogbn-proteins | Number of params | 96880 | GCN |