Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

KerGNNs: Interpretable Graph Neural Networks with Graph Kernels

About

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 \textit{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.

Aosong Feng, Chenyu You, Shiqiang Wang, Leandros Tassiulas• 2022

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS (test)
Accuracy70.6
213
Graph ClassificationNCI1 (test)
Accuracy67.4
177
Graph ClassificationBaMultiShapes (test)
Accuracy82.7
17
Graph ClassificationMutagenicity (test)
Accuracy72.2
14
Graph ClassificationBa2Motifs (test)
Accuracy97.9
12
High-throughput screeningPubChem AID 435008
logAUC14.7
8
High-throughput screeningPubChem AID 1798
logAUC7.8
8
High-throughput screeningPubChem AID 435034
logAUC [0.001, 0.1]0.179
8
High-throughput screeningPubChem AID 2258
logAUC [0.001, 0.1]19.5
8
High-throughput screeningPubChem AID 463087
logAUC (0.001, 0.1)15
8
Showing 10 of 13 rows

Other info

Follow for update