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

EIGNN: Efficient Infinite-Depth Graph Neural Networks

About

Graph neural networks (GNNs) are widely used for modelling graph-structured data in numerous applications. However, with their inherently finite aggregation layers, existing GNN models may not be able to effectively capture long-range dependencies in the underlying graphs. Motivated by this limitation, we propose a GNN model with infinite depth, which we call Efficient Infinite-Depth Graph Neural Networks (EIGNN), to efficiently capture very long-range dependencies. We theoretically derive a closed-form solution of EIGNN which makes training an infinite-depth GNN model tractable. We then further show that we can achieve more efficient computation for training EIGNN by using eigendecomposition. The empirical results of comprehensive experiments on synthetic and real-world datasets show that EIGNN has a better ability to capture long-range dependencies than recent baselines, and consistently achieves state-of-the-art performance. Furthermore, we show that our model is also more robust against both noise and adversarial perturbations on node features.

Juncheng Liu, Kenji Kawaguchi, Bryan Hooi, Yiwei Wang, Xiaokui Xiao• 2022

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy75.9
742
Node ClassificationChameleon
Accuracy62.92
549
Node ClassificationSquirrel
Accuracy46.37
500
Graph ClassificationNCI1
Accuracy77.5
460
Node ClassificationCornell
Accuracy85.13
426
Node ClassificationWisconsin
Accuracy86.86
410
Node ClassificationTexas
Accuracy0.846
410
Node ClassificationTexas (test)
Mean Accuracy84.33
228
Graph ClassificationIMDB-M
Accuracy52.1
218
Graph ClassificationMUTAG (10-fold cross-validation)
Accuracy88.9
206
Showing 10 of 15 rows

Other info

Code

Follow for update