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

Strengthening structural baselines for graph classification using Local Topological Profile

About

We present the analysis of the topological graph descriptor Local Degree Profile (LDP), which forms a widely used structural baseline for graph classification. Our study focuses on model evaluation in the context of the recently developed fair evaluation framework, which defines rigorous routines for model selection and evaluation for graph classification, ensuring reproducibility and comparability of the results. Based on the obtained insights, we propose a new baseline algorithm called Local Topological Profile (LTP), which extends LDP by using additional centrality measures and local vertex descriptors. The new approach provides the results outperforming or very close to the latest GNNs for all datasets used. Specifically, state-of-the-art results were obtained for 4 out of 9 benchmark datasets. We also consider computational aspects of LDP-based feature extraction and model construction to propose practical improvements affecting execution speed and scalability. This allows for handling modern, large datasets and extends the portfolio of benchmarks used in graph representation learning. As the outcome of our work, we obtained LTP as a simple to understand, fast and scalable, still robust baseline, capable of outcompeting modern graph classification models such as Graph Isomorphism Network (GIN). We provide open-source implementation at \href{https://github.com/j-adamczyk/LTP}{GitHub}.

Jakub Adamczyk, Wojciech Czech• 2023

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy72.7
742
Graph ClassificationNCI1
Accuracy77
460
Graph ClassificationCOLLAB
Accuracy79.4
329
Graph ClassificationIMDB-B
Accuracy74.5
322
Graph ClassificationENZYMES
Accuracy42.5
305
Graph ClassificationIMDB-M
Accuracy50
218
Graph ClassificationDD
Accuracy77.1
175
Graph ClassificationREDDIT-B
Accuracy91.1
71
Graph ClassificationREDDIT-5K
Accuracy53.3
26
Graph ClassificationGraph classification benchmark suite (DD, NCI1, PROTEINS, ENZYMES, IMDB-B, IMDB-M, REDDIT-B, REDDIT-5K, COLLAB)
Average Rank2.6
8
Showing 10 of 10 rows

Other info

Follow for update