Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networks
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Graph Classification | PROTEINS | Accuracy75.5 | 742 | |
| Graph Classification | MUTAG | Accuracy86.1 | 697 | |
| Graph Classification | NCI1 | Accuracy76.2 | 460 | |
| Graph Classification | IMDB-B | Accuracy74.2 | 322 | |
| Graph Classification | Mutag (test) | Accuracy86.1 | 217 | |
| Graph Classification | MUTAG (10-fold cross-validation) | Accuracy86.1 | 206 | |
| Graph Classification | PROTEINS (10-fold cross-validation) | Accuracy75.5 | 197 | |
| Graph Classification | PROTEINS (test) | Accuracy75.5 | 180 | |
| Molecular property prediction | QM9 (test) | mu0.493 | 174 | |
| Graph Regression | ZINC 12K (test) | MAE0.069 | 164 |