HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

A Persistent Weisfeiler–Lehman Procedure for Graph Classification

{Karsten Borgwardt Bastian Rieck Christian Bock}

A Persistent Weisfeiler–Lehman Procedure for Graph Classification

Abstract

The Weisfeiler–Lehman graph kernel exhibits competitive performance in many graph classification tasks. However, its subtree features are not able to capture connected components and cycles, topological features known for characterising graphs. To extract such features, we leverage propagated node label information and transform unweighted graphs into metric ones. This permits us to augment the subtree features with topological information obtained using persistent homology, a concept from topological data analysis. Our method, which we formalise as a generalisation of Weisfeiler–Lehman subtree features, exhibits favourable classification accuracy and its improvements in predictive performance are mainly driven by including cycle information.

Benchmarks

BenchmarkMethodologyMetrics
graph-classification-on-mutagP-WL-C
Mean Accuracy: 90.51
graph-classification-on-proteinsP-WL-UC
Accuracy: 75.36%
graph-property-prediction-on-ogbg-molhivP-WL
Ext. data: No
Number of params: 4600000
Test ROC-AUC: 0.8039 ± 0.0040
Validation ROC-AUC: 0.8279 ± 0.0059

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 Persistent Weisfeiler–Lehman Procedure for Graph Classification | Papers | HyperAI