Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A Fair Comparison of Graph Neural Networks for Graph Classification

About

Experimental reproducibility and replicability are critical topics in machine learning. Authors have often raised concerns about their lack in scientific publications to improve the quality of the field. Recently, the graph representation learning field has attracted the attention of a wide research community, which resulted in a large stream of works. As such, several Graph Neural Network models have been developed to effectively tackle graph classification. However, experimental procedures often lack rigorousness and are hardly reproducible. Motivated by this, we provide an overview of common practices that should be avoided to fairly compare with the state of the art. To counter this troubling trend, we ran more than 47000 experiments in a controlled and uniform framework to re-evaluate five popular models across nine common benchmarks. Moreover, by comparing GNNs with structure-agnostic baselines we provide convincing evidence that, on some datasets, structural information has not been exploited yet. We believe that this work can contribute to the development of the graph learning field, by providing a much needed grounding for rigorous evaluations of graph classification models.

Federico Errica, Marco Podda, Davide Bacciu, Alessio Micheli• 2019

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy75.8
742
Graph ClassificationNCI1
Accuracy80
460
Graph ClassificationCOLLAB
Accuracy75.6
329
Graph ClassificationIMDB-B
Accuracy71.2
322
Graph ClassificationENZYMES
Accuracy65.2
305
Graph ClassificationIMDB-M
Accuracy49.1
218
Graph ClassificationDD
Accuracy78.4
175
Graph ClassificationNCI1 (test)
Accuracy80
174
Graph ClassificationREDDIT-B
Accuracy89.9
71
Graph ClassificationREDDIT-5K
Accuracy56.1
26
Showing 10 of 11 rows

Other info

Code

Follow for update