HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

DeeperGCN: All You Need to Train Deeper GCNs

Guohao Li; Chenxin Xiong; Ali Thabet; Bernard Ghanem

DeeperGCN: All You Need to Train Deeper GCNs

Abstract

Graph Convolutional Networks (GCNs) have been drawing significant attention with the power of representation learning on graphs. Unlike Convolutional Neural Networks (CNNs), which are able to take advantage of stacking very deep layers, GCNs suffer from vanishing gradient, over-smoothing and over-fitting issues when going deeper. These challenges limit the representation power of GCNs on large-scale graphs. This paper proposes DeeperGCN that is capable of successfully and reliably training very deep GCNs. We define differentiable generalized aggregation functions to unify different message aggregation operations (e.g. mean, max). We also propose a novel normalization layer namely MsgNorm and a pre-activation version of residual connections for GCNs. Extensive experiments on Open Graph Benchmark (OGB) show DeeperGCN significantly boosts performance over the state-of-the-art on the large scale graph learning tasks of node property prediction and graph property prediction. Please visit https://www.deepgcns.org for more information.

Benchmarks

BenchmarkMethodologyMetrics
graph-property-prediction-on-ogbg-molhivDeeperGCN
Ext. data: No
Number of params: 531976
Test ROC-AUC: 0.7858 ± 0.0117
Validation ROC-AUC: 0.8427 ± 0.0063
graph-property-prediction-on-ogbg-molpcbaDeeperGCN+virtual node
Ext. data: No
Number of params: 5550208
Test AP: 0.2781 ± 0.0038
Validation AP: 0.2920 ± 0.0025
graph-property-prediction-on-ogbg-ppaDeeperGCN
Ext. data: No
Number of params: 2336421
Test Accuracy: 0.7712 ± 0.0071
Validation Accuracy: 0.7313 ± 0.0078
link-property-prediction-on-ogbl-collabDeeperGCN
Ext. data: No
Number of params: 117383
Test Hits@50: 0.5273 ± 0.0047
Validation Hits@50: 0.6187 ± 0.0045

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp