Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networks

In recent years, graph neural networks (GNNs) have emerged as a powerful neural architecture to learn vector representations of nodes and graphs in a supervised, end-to-end fashion. Up to now, GNNs have only been evaluated empirically -- showing promising results. The following work investigates GNNs from a theoretical point of view and relates them to the $1$-dimensional Weisfeiler-Leman graph isomorphism heuristic ($1$-WL). We show that GNNs have the same expressiveness as the $1$-WL in terms of distinguishing non-isomorphic (sub-)graphs. Hence, both algorithms also have the same shortcomings. Based on this, we propose a generalization of GNNs, so-called $k$-dimensional GNNs ($k$-GNNs), which can take higher-order graph structures at multiple scales into account. These higher-order structures play an essential role in the characterization of social networks and molecule graphs. Our experimental evaluation confirms our theoretical findings as well as confirms that higher-order information is useful in the task of graph classification and regression.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Graph Classification IMDb-B k-GNN Accuracy 74.2% # 18
Graph Classification IMDb-B 3-WL Kernel Accuracy 73.5% # 20
Graph Classification IMDb-M k-GNN Accuracy 49.5% # 26
Graph Classification IMDb-M 1-WL Kernel Accuracy 51.5% # 14
Graph Classification MUTAG k-GNN Accuracy 86.1% # 55
Graph Classification MUTAG Graphlet Kernel Accuracy 87.7% # 41
Graph Classification NCI1 k-GNN Accuracy 76.2% # 37
Graph Classification NCI1 WL-OA Kernel Accuracy 86.1% # 4
Graph Classification PROTEINS k-GNN Accuracy 75.9% # 49
Graph Classification PROTEINS Shortest-Path Kernel Accuracy 76.4% # 39

Methods


No methods listed for this paper. Add relevant methods here