AAAI 2022

KerGNNs: Interpretable Graph Neural Networks with Graph Kernels

Aosong Feng, Chenyu You, Shiqiang Wang, L. Tassiulas

Main figure for KerGNNs: Interpretable Graph Neural Networks with Graph Kernels

Abstract

Graph kernels are historically the most widely-used technique for graph classification tasks. However, these methods suffer from limited performance because of the hand-crafted combinatorial features of graphs. In recent years, graph neural networks (GNNs) have become the state-of-the-art method in downstream graph-related tasks due to their superior performance. Most GNNs are based on Message Passing Neural Network (MPNN) frameworks. However, recent studies show that MPNNs can not exceed the power of the Weisfeiler-Lehman (WL) algorithm in graph isomorphism test. To address the limitations of existing graph kernel and GNN methods, in this paper, we propose a novel GNN framework, termed Kernel Graph Neural Networks (KerGNNs), which integrates graph kernels into the message passing process of GNNs. Inspired by convolution filters in convolutional neural networks (CNNs), KerGNNs adopt trainable hidden graphs as graph filters which are combined with subgraphs to update node embeddings using graph kernels. In addition, we show that MPNNs can be viewed as special cases of KerGNNs. We apply KerGNNs to multiple graph-related tasks and use cross-validation to make fair comparisons with benchmarks. We show that our method achieves competitive performance compared with existing state-of-the-art methods, demonstrating the potential to increase the representation ability of GNNs. We also show that the trained graph filters in KerGNNs can reveal the local graph structures of the dataset, which significantly improves the model interpretability compared with conventional GNN models.

Benchmark Results

Tested on 3 datasets 4 baseline values reported too low

Tables show presented results in global context (best 3 papers + GCN + MLP).

TU Dortmund

#ModelAccuracy ↑Paper
1
WL-MLP
0.9954
2
GIN
0.9934
3
GCN
0.9727
28
PK
actual #28 vs paper #459 431 places worse
0.8450 ±0.0010
best reported
actual #28 vs paper #459 431 places worse
paper 0.7230 ±0.0280
best at pub: Propagation Kernels (2014)
paper 0.7230 ±0.0280 Δ at pub 0.1220
36
GNTK
actual #36 vs paper #49 13 places worse
0.8420 ±0.0150
best reported
actual #36 vs paper #49 13 places worse
paper 0.8350 ±0.0120

Difference within reported uncertainty; not counted as underperformance.

paper 0.8350 ±0.0120 Δ at pub 0.0070 Δ today 0.0070
55
Powerful GNN
0.8340 ±0.0180
This paper
83
KerGNN-2-DRW
0.8280 ±0.0180
This paper
131
GraphSAGE
0.8200 ±0.0100
193
KerGNN-3
0.8050 ±0.0190
This paper
248
WL-sub
0.7950 ±0.0330
This paper
257
GCKN
actual #257 vs paper #257
0.7920 ±0.0120
actual #257 vs paper #257
paper 0.7920 ±0.0120
paper 0.7920 ±0.0120 Δ at pub 0.0860
383
KerGNN-2
0.7630 ±0.0260
This paper
389
1-2-3 GNN
actual #389 vs paper #455 66 places worse
0.7620
best reported
actual #389 vs paper #455 66 places worse
paper 0.7270 ±0.0290
best at pub: Provably Powerful Graph Networks (NeurIPS 2019)
paper 0.7270 ±0.0290 Δ at pub 0.0350
421
SP
actual #421 vs paper #515 94 places worse
0.7450 ±0.0030
best reported
actual #421 vs paper #515 94 places worse
paper 0.6630 ±0.0260
paper 0.6630 ±0.0260 Δ at pub 0.0820
431
KerGNN-1
0.7430 ±0.0220
This paper
#ModelAccuracy ↑Paper
1
WL-MLP
0.9792
2
MSH-GNN
0.9640 ±0.0070
3
CoCN
0.8722 ±0.0013
23
GCN
0.8260 ±0.0220
151
MLP
0.7920 ±0.0240
193
WL-sub
0.7750 ±0.0240
This paper
197
PK
0.7730 ±0.0240
This paper
245
Powerful GNN
0.7540 ±0.0140
This paper
251
KerGNN-2-DRW
0.7510 ±0.0230
This paper
260
GCKN
actual #260 vs paper #260
0.7430 ±0.0280
actual #260 vs paper #260
paper 0.7430 ±0.0280
paper 0.7430 ±0.0280 Δ at pub 0.2000
279
KerGNN-2
0.7270 ±0.0210
This paper
291
KerGNN-3
0.7110 ±0.0200
This paper
297
KerGNN-1
0.7050 ±0.0160
This paper
332
SP
actual #332 vs paper #332
0.5880 ±0.0020
best reported
actual #332 vs paper #332
paper 0.5840 ±0.0130

Difference within reported uncertainty; not counted as underperformance.

paper 0.5840 ±0.0130 Δ at pub 0.0040
#ModelAccuracy ↑Paper
1
GMN
0.7866
2
MemGNN
0.7550
3
TFGW SP
0.7510 ±0.0500
17
GatedGCN
0.7000 ±0.0494
26
GraphSAGE
0.6817 ±0.0545
74
KerGNN-1
0.6210 ±0.0550
This paper
84
PK
0.6100 ±0.0670
This paper
103
KerGNN-2-DRW
0.5950 ±0.0450
This paper
106
GCKN
0.5930 ±0.0560
This paper
124
MLP
0.5583 ±0.0352
133
KerGNN-2
0.5500 ±0.0500
This paper
136
Powerful GNN
0.5480 ±0.0550
This paper
139
KerGNN-3
0.5410 ±0.0430
This paper
154
WL-sub
0.5120 ±0.0530
This paper
163
GNTK
0.4820 ±0.0240
This paper
179
SP
actual #179 vs paper #255 76 places worse
0.4260 ±0.0160
best reported
actual #179 vs paper #255 76 places worse
paper 0.2500 ±0.0560
paper 0.2500 ±0.0560 Δ at pub 0.1760
SOTA performance (not significantly different from the best) Baseline / comparison (this paper) Global context (other papers) Best at pub — reported result worse than the best known when published (best reported in green) Best at pub — reported value improved over the best known when published