Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Predict then Propagate: Graph Neural Networks meet Personalized PageRank

About

Neural message passing algorithms for semi-supervised classification on graphs have recently achieved great success. However, for classifying a node these methods only consider nodes that are a few propagation steps away and the size of this utilized neighborhood is hard to extend. In this paper, we use the relationship between graph convolutional networks (GCN) and PageRank to derive an improved propagation scheme based on personalized PageRank. We utilize this propagation procedure to construct a simple model, personalized propagation of neural predictions (PPNP), and its fast approximation, APPNP. Our model's training time is on par or faster and its number of parameters on par or lower than previous models. It leverages a large, adjustable neighborhood for classification and can be easily combined with any neural network. We show that this model outperforms several recently proposed methods for semi-supervised classification in the most thorough study done so far for GCN-like models. Our implementation is available online.

Johannes Gasteiger, Aleksandar Bojchevski, Stephan G\"unnemann• 2018

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy87.71
885
Node ClassificationCiteseer
Accuracy80.47
804
Node ClassificationPubmed
Accuracy88.43
742
Node ClassificationCiteseer (test)
Accuracy0.7547
729
Node ClassificationCora (test)
Mean Accuracy88.14
687
Node ClassificationChameleon
Accuracy52.15
549
Node ClassificationPubMed (test)
Accuracy85.02
500
Node ClassificationSquirrel
Accuracy36.88
500
Node ClassificationCornell
Accuracy63.78
426
Node ClassificationTexas
Accuracy0.9098
410
Showing 10 of 250 rows
...

Other info

Code

Follow for update