Bag of Tricks for Node Classification with Graph Neural Networks

24 Mar 2021  ·  Yangkun Wang, Jiarui Jin, Weinan Zhang, Yong Yu, Zheng Zhang, David Wipf ·

Over the past few years, graph neural networks (GNN) and label propagation-based methods have made significant progress in addressing node classification tasks on graphs. However, in addition to their reliance on elaborate architectures and algorithms, there are several key technical details that are frequently overlooked, and yet nonetheless can play a vital role in achieving satisfactory performance. In this paper, we first summarize a series of existing tricks-of-the-trade, and then propose several new ones related to label usage, loss function formulation, and model design that can significantly improve various GNN architectures. We empirically evaluate their impact on final node classification accuracy by conducting ablation studies and demonstrate consistently-improved performance, often to an extent that outweighs the gains from more dramatic changes in the underlying GNN architecture. Notably, many of the top-ranked models on the Open Graph Benchmark (OGB) leaderboard and KDDCUP 2021 Large-Scale Challenge MAG240M-LSC benefit from these techniques we initiated.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Node Property Prediction ogbn-arxiv GAT+norm. adj.+label reuse Test Accuracy 0.7391 ± 0.0012 # 33
Validation Accuracy 0.7516 ± 0.0008 # 27
Number of params 1441580 # 32
Ext. data No # 1
Node Property Prediction ogbn-arxiv GAT+norm.adj.+labels Test Accuracy 0.7365 ± 0.0011 # 40
Validation Accuracy 0.7504 ± 0.0006 # 32
Number of params 1628440 # 22
Ext. data No # 1
Node Property Prediction ogbn-arxiv GAT+norm. adj.+labels Test Accuracy 0.7366 ± 0.0011 # 39
Validation Accuracy 0.7508 ± 0.0009 # 30
Number of params 1441580 # 32
Ext. data No # 1
Node Property Prediction ogbn-arxiv GCN+linear+labels Test Accuracy 0.7306 ± 0.0024 # 47
Validation Accuracy 0.7442 ± 0.0012 # 47
Number of params 238632 # 53
Ext. data No # 1
Node Property Prediction ogbn-proteins GAT+BoT Test ROC-AUC 0.8765 ± 0.0008 # 7
Validation ROC-AUC 0.9280 ± 0.0008 # 7
Number of params 2484192 # 10
Ext. data No # 1
Node Property Prediction ogbn-proteins GAT+EdgeFeatureAtt Test ROC-AUC 0.8682 ± 0.0021 # 11
Validation ROC-AUC 0.9194 ± 0.0003 # 10
Number of params 2475232 # 11
Ext. data No # 1

Methods


No methods listed for this paper. Add relevant methods here