Command Palette
Search for a command to run...
Clemens Damke Vitalik Melnikov Eyke Hüllermeier

Abstract
Current GNN architectures use a vertex neighborhood aggregation scheme, which limits their discriminative power to that of the 1-dimensional Weisfeiler-Lehman (WL) graph isomorphism test. Here, we propose a novel graph convolution operator that is based on the 2-dimensional WL test. We formally show that the resulting 2-WL-GNN architecture is more discriminative than existing GNN approaches. This theoretical result is complemented by experimental studies using synthetic and real data. On multiple common graph classification benchmarks, we demonstrate that the proposed model is competitive with state-of-the-art graph kernels and GNNs.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| graph-classification-on-dd | 2-WL-GNN | Accuracy: 75.4 |
| graph-classification-on-imdb-b | 2-WL-GNN | Accuracy: 72.2 |
| graph-classification-on-nci1 | 2-WL-GNN | Accuracy: 73.5 |
| graph-classification-on-proteins | 2-WL-GNN | Accuracy: 76.5 |
| graph-classification-on-reddit-b | 2-WL-GNN | Accuracy: 89.4 |
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.