HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

A Fair Comparison of Graph Neural Networks for Graph Classification

Federico Errica Marco Podda Davide Bacciu Alessio Micheli

A Fair Comparison of Graph Neural Networks for Graph Classification

Abstract

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.

Code Repositories

diningphil/CGMM
pytorch
Mentioned in GitHub
diningphil/icgmm
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
graph-classification-on-collabGraphSAGE
Accuracy: 73.9%
graph-classification-on-ddDGCNN
Accuracy: 76.6%
graph-classification-on-enzymesGraphSAGE
Accuracy: 58.2%
graph-classification-on-enzymesGIN
Accuracy: 59.6%
graph-classification-on-imdb-bGraphSAGE
Accuracy: 68.8%
graph-classification-on-imdb-mGraphSAGE
Accuracy: 47.6%
graph-classification-on-nci1GIN
Accuracy: 80%
graph-classification-on-nci1DGCNN
Accuracy: 76.4%
graph-classification-on-proteinsGraphSAGE
Accuracy: 73%
graph-classification-on-proteinsDiffPool
Accuracy: 73.7%
graph-classification-on-reddit-bGraphSAGE
Accuracy: 84.3
graph-classification-on-reddit-multi-5kGraphSAGE
Accuracy: 50

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
A Fair Comparison of Graph Neural Networks for Graph Classification | Papers | HyperAI