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

Pitfalls of Graph Neural Network Evaluation

About

Semi-supervised node classification in graphs is a fundamental problem in graph mining, and the recently proposed graph neural networks (GNNs) have achieved unparalleled results on this task. Due to their massive success, GNNs have attracted a lot of attention, and many novel architectures have been put forward. In this paper we show that existing evaluation strategies for GNN models have serious shortcomings. We show that using the same train/validation/test splits of the same datasets, as well as making significant changes to the training procedure (e.g. early stopping criteria) precludes a fair comparison of different architectures. We perform a thorough empirical evaluation of four prominent GNN models and show that considering different splits of the data leads to dramatically different rankings of models. Even more importantly, our findings suggest that simpler GNN architectures are able to outperform the more sophisticated ones if the hyperparameters and the training procedure are tuned fairly for all models.

Oleksandr Shchur, Maximilian Mumme, Aleksandar Bojchevski, Stephan G\"unnemann• 2018

Related benchmarks

TaskDatasetResultRank
Node ClassificationCoauthor-CS (test)
Accuracy91.3
120
Node ClassificationCora Full
Accuracy58.6
88
Node ClassificationCoauthor Phy (test)
Accuracy93
41
Showing 3 of 3 rows

Other info

Follow for update