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
627
Node ClassificationTexas
Accuracy0.773
616
Node ClassificationCornell
Accuracy71.8
582
Graph Classificationogbg-molpcba (test)
AP27.65
206
Graph RegressionPeptides struct LRGB (test)
MAE0.2509
187
Graph RegressionZINC 12K (test)
MAE0.139
164
Graph ClassificationPeptides-func LRGB (test)
AP0.6535
145
Node ClassificationTexas (48/32/20)
Mean Accuracy57.36
78
Graph RegressionPeptides-struct
MAE0.3357
76
Node ClassificationPascalVOC-SP LRGB (test)
F1 Score32.3
60
Showing 10 of 38 rows

Other info

Code

Follow for update