Scalable and Adaptive Graph Neural Networks with Self-Label-Enhanced training

19 Apr 2021  ·  Chuxiong Sun, Hongming Gu, Jie Hu ·

It is hard to directly implement Graph Neural Networks (GNNs) on large scaled graphs. Besides of existed neighbor sampling techniques, scalable methods decoupling graph convolutions and other learnable transformations into preprocessing and post classifier allow normal minibatch training. By replacing redundant concatenation operation with attention mechanism in SIGN, we propose Scalable and Adaptive Graph Neural Networks (SAGN). SAGN can adaptively gather neighborhood information among different hops. To further improve scalable models on semi-supervised learning tasks, we propose Self-Label-Enhance (SLE) framework combining self-training approach and label propagation in depth. We add base model with a scalable node label module. Then we iteratively train models and enhance train set in several stages. To generate input of node label module, we directly apply label propagation based on one-hot encoded label vectors without inner random masking. We find out that empirically the label leakage has been effectively alleviated after graph convolutions. The hard pseudo labels in enhanced train set participate in label propagation with true labels. Experiments on both inductive and transductive datasets demonstrate that, compared with other sampling-based and sampling-free methods, SAGN achieves better or comparable results and SLE can further improve performance.

PDF Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Node Property Prediction ogbn-mag NARS_SAGN+SLE Test Accuracy 0.5440 ± 0.0015 # 14
Validation Accuracy 0.5591 ± 0.0017 # 14
Number of params 3846330 # 33
Ext. data No # 1
Node Property Prediction ogbn-papers100M SAGN+SLE (4 stages) Test Accuracy 0.6830 ± 0.0008 # 4
Validation Accuracy 0.7163 ± 0.0007 # 8
Number of params 8556888 # 10
Ext. data No # 1
Node Property Prediction ogbn-papers100M SAGN Test Accuracy 0.6675 ± 0.0084 # 14
Validation Accuracy 0.7034 ± 0.0099 # 13
Number of params 6098092 # 13
Ext. data No # 1
Node Property Prediction ogbn-papers100M SAGN+SLE Test Accuracy 0.6800 ± 0.0015 # 9
Validation Accuracy 0.7131 ± 0.0010 # 10
Number of params 8556888 # 10
Ext. data No # 1
Node Property Prediction ogbn-products SAGN+SLE (4 stages) Test Accuracy 0.8468 ± 0.0012 # 18
Validation Accuracy 0.9309 ± 0.0007 # 19
Number of params 2179678 # 17
Ext. data No # 1
Node Property Prediction ogbn-products SAGN Test Accuracy 0.8120 ± 0.0007 # 36
Validation Accuracy 0.9309 ± 0.0004 # 19
Number of params 2233391 # 16
Ext. data No # 1
Node Property Prediction ogbn-products SAGN+SLE Test Accuracy 0.8428 ± 0.0014 # 23
Validation Accuracy 0.9287 ± 0.0003 # 27
Number of params 2179678 # 17
Ext. data No # 1
Node Property Prediction ogbn-products SAGN+SLE (4 stages)+C&S Test Accuracy 0.8485 ± 0.0010 # 17
Validation Accuracy 0.9302 ± 0.0003 # 23
Number of params 2179678 # 17
Ext. data No # 1

Methods


No methods listed for this paper. Add relevant methods here