Share your thoughts, 1 month free Claude Pro on usSee more
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
994
Node ClassificationChameleon
Accuracy62.92
640
Node ClassificationWisconsin
Accuracy86.86
627
Node ClassificationTexas
Accuracy0.846
616
Node ClassificationSquirrel
Accuracy46.37
591
Node ClassificationCornell
Accuracy85.13
582
Graph ClassificationNCI1
Accuracy77.5
501
Graph ClassificationIMDB-M
Accuracy52.1
275
Node ClassificationCornell (test)
Mean Accuracy85.13
274
Node ClassificationTexas (test)
Mean Accuracy84.33
269
Showing 10 of 15 rows

Other info

Code

Follow for update