NeurIPS 2020

OGB

Weihua Hu, Matthias Fey, M. Zitnik, Yuxiao Dong, Hongyu Ren, Bowen Liu, Michele Catasta, J. Leskovec

Main figure for OGB

Abstract

We present the Open Graph Benchmark (OGB), a diverse set of challenging and realistic benchmark datasets to facilitate scalable, robust, and reproducible graph machine learning (ML) research. OGB datasets are large-scale (up to 100+ million nodes and 1+ billion edges), encompass multiple important graph ML tasks, and cover a diverse range of domains, ranging from social and information networks to biological networks, molecular graphs, source code ASTs, and knowledge graphs. For each dataset, we provide a unified evaluation protocol using meaningful application-specific data splits and evaluation metrics. In addition to building the datasets, we also perform extensive benchmark experiments for each dataset. Our experiments suggest that OGB datasets present significant challenges of scalability to large-scale graphs and out-of-distribution generalization under realistic data splits, indicating fruitful opportunities for future research. Finally, OGB provides an automated end-to-end graph ML pipeline that simplifies and standardizes the process of graph data loading, experimental setup, and model evaluation. OGB will be regularly updated and welcomes inputs from the community. OGB datasets as well as data loaders, evaluation scripts, baseline code, and leaderboards are publicly available at this https URL.

Benchmark Results

Tested on 6 datasets

Tables show presented results in global context (best 3 papers + GCN + MLP).

OGB

#ModelHits@50 ↑Paper
1
PROXI
0.7650 ±0.0027
2
OGB Leader
0.7129 ±0.0018
3
GIDN
0.7096 ±0.0055
9
GCN
actual #9 vs paper #85 76 places worse
0.6621 ±0.0033
best reported
actual #9 vs paper #85 76 places worse
paper 0.4714 ±0.0145
paper 0.4714 ±0.0145 Δ today 0.1907
28
SAGE
0.5944 ±0.0137
41
GraphSAGE
0.5463 ±0.0112
This paper
68
Node2Vec
actual #68 vs paper #76 8 places worse
0.4929 ±0.0064
best reported
actual #68 vs paper #76 8 places worse
paper 0.4888 ±0.0054

Difference within reported uncertainty; not counted as underperformance.

paper 0.4888 ±0.0054 Δ today 0.0041
94
MLP
actual #94 vs paper #110 16 places worse
0.4438 ±0.0347
best reported
actual #94 vs paper #110 16 places worse
paper 0.1927 ±0.0129
paper 0.1927 ±0.0129 Δ today 0.2511
102
MatrixFactorization
0.3886 ±0.0029
This paper
#ModelHits@20 ↑Paper
1
ELGNN
0.9777 ±0.0037
2
GIDN
0.9542
3
AGDN
0.9538 ±0.0094
20
SAGE
0.6719 ±0.0118
24
GCN
actual #24 vs paper #53 29 places worse
0.6476 ±0.0145
best reported
actual #24 vs paper #53 29 places worse
paper 0.3707 ±0.0507
paper 0.3707 ±0.0507 Δ today 0.2769
25
Node2Vec
actual #25 vs paper #68 43 places worse
0.6363 ±0.0205
best reported
actual #25 vs paper #68 43 places worse
paper 0.2326 ±0.0209
paper 0.2326 ±0.0209 Δ today 0.4037
38
GraphSAGE
0.5390 ±0.0474
This paper
82
MatrixFactorization
0.1368 ±0.0475
This paper
#ModelHits@100 ↑Paper
1
OGB Leader
0.6524 ±0.0150
2
PRB
0.6347 ±0.0175
3
LPFormer
0.6332 ±0.0063
14
SAGE
0.5013 ±0.0055
24
ML
0.4600
This paper
23
MLP
0.4600
This paper
44
MatrixFactorization
0.3229 ±0.0094
This paper
45
GCN
actual #45 vs paper #59 14 places worse
0.3084 ±0.0178
best reported
actual #45 vs paper #59 14 places worse
paper 0.1867 ±0.0132
paper 0.1867 ±0.0132 Δ today 0.1217
57
Node2Vec
actual #57 vs paper #57
0.2230 ±0.0080
best reported
actual #57 vs paper #57
paper 0.2226 ±0.0083

Difference within reported uncertainty; not counted as underperformance.

paper 0.2226 ±0.0083 Δ today 0.0004
63
GraphSAGE
0.1655 ±0.0240
This paper
#ModelMRR ↑Paper
1
CFG
0.8997 ±0.0015
2
SIEG
0.8987 ±0.0018
3
LPFormer
0.8981 ±0.0013
12
GCN
actual #12 vs paper #21 9 places worse
0.8705 ±0.0004
best reported
actual #12 vs paper #21 9 places worse
paper 0.8474 ±0.0021
paper 0.8474 ±0.0021 Δ today 0.0231
30
SAGE
0.8292 ±0.0022
32
GraphSAGE
actual #32 vs paper #32
0.8264 ±0.0001
best reported
actual #32 vs paper #32
paper 0.8260 ±0.0036

Difference within reported uncertainty; not counted as underperformance.

paper 0.8260 ±0.0036 Δ today 0.0004
40
NeighborSampling
0.8044 ±0.0010
This paper
42
ClusterGCN
0.8004 ±0.0025
This paper
43
GraphSAINT
0.7985 ±0.0040
This paper
59
Node2Vec
0.6141 ±0.0011
This paper
63
MatrixFactorization
0.5186 ±0.0443
This paper
71
MLP
actual #71 vs paper #71
0.3917 ±0.0044
best reported
actual #71 vs paper #71
paper 0.2895 ±0.0014
paper 0.2895 ±0.0014 Δ today 0.1022
#ModelMRR ↑Paper
1
InterHT + DigPiece
0.7293 ±0.0018
2
StarGraph + TripleRE'
0.7290
3
TripleRE + StarGraph
0.7201 ±0.0011
32
RotatE (5×dim)
0.4332 ±0.0025
This paper
34
TransE (5×dim)
0.4256 ±0.0030
This paper
37
ComplEx (5×dim)
0.4027 ±0.0027
This paper
40
DistMult (5×dim)
0.3729 ±0.0045
This paper
#ModelMRR ↑Paper
1
PairRE+LRE
0.8360
2
NBFnet
0.8317
3
AutoSF
0.8309 ±0.0008
7
ComplEx
0.8095 ±0.0007
This paper
8
DistMult
0.8043 ±0.0003
This paper
9
RotatE
0.7989 ±0.0004
This paper
11
TransE
0.7452 ±0.0004
This paper
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers) Best today — OK at publication; today's leaderboard best is elsewhere