Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Long Range Graph Benchmark

About

Graph Neural Networks (GNNs) that are based on the message passing (MP) paradigm generally exchange information between 1-hop neighbors to build node representations at each layer. In principle, such networks are not able to capture long-range interactions (LRI) that may be desired or necessary for learning a given task on graphs. Recently, there has been an increasing interest in development of Transformer-based methods for graphs that can consider full node connectivity beyond the original sparse structure, thus enabling the modeling of LRI. However, MP-GNNs that simply rely on 1-hop message passing often fare better in several existing graph benchmarks when combined with positional feature representations, among other innovations, hence limiting the perceived utility and ranking of Transformer-like architectures. Here, we present the Long Range Graph Benchmark (LRGB) with 5 graph learning datasets: PascalVOC-SP, COCO-SP, PCQM-Contact, Peptides-func and Peptides-struct that arguably require LRI reasoning to achieve strong performance in a given task. We benchmark both baseline GNNs and Graph Transformer networks to verify that the models which capture long-range dependencies perform significantly better on these tasks. Therefore, these datasets are suitable for benchmarking and exploration of MP-GNNs and Graph Transformer architectures that are intended to capture LRI.

Vijay Prakash Dwivedi, Ladislav Ramp\'a\v{s}ek, Mikhail Galkin, Ali Parviz, Guy Wolf, Anh Tuan Luu, Dominique Beaini• 2022

Related benchmarks

TaskDatasetResultRank
Node ClassificationWisconsin
Accuracy79.8
864
Node ClassificationCornell
Accuracy71.8
851
Node ClassificationTexas
Accuracy0.773
801
Graph RegressionPeptides struct LRGB (test)
MAE0.2509
238
Graph Classificationogbg-molpcba (test)
AP27.65
212
Graph ClassificationPeptides-func LRGB (test)
AP0.6535
196
Graph RegressionZINC 12K (test)
MAE0.139
173
Graph RegressionPeptides-struct
MAE0.2529
134
Graph ClassificationPeptides func
AP60.69
110
Node ClassificationTexas (48/32/20)
Mean Accuracy57.36
78
Showing 10 of 47 rows

Other info

Code

Follow for update