edGNN: a Simple and Powerful GNN for Directed Labeled Graphs
About
The ability of a graph neural network (GNN) to leverage both the graph topology and graph labels is fundamental to building discriminative node and graph embeddings. Building on previous work, we theoretically show that edGNN, our model for directed labeled graphs, is as powerful as the Weisfeiler-Lehman algorithm for graph isomorphism. Our experiments support our theoretical findings, confirming that graph neural networks can be used effectively for inference problems on directed graphs with both node and edge labels. Code available at https://github.com/guillaumejaume/edGNN.
Guillaume Jaume, An-phi Nguyen, Mar\'ia Rodr\'iguez Mart\'inez, Jean-Philippe Thiran, Maria Gabrani• 2019
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Graph Classification | MUTAG | Accuracy88.8 | 697 | |
| Graph Classification | MUTAG (10-fold cross-validation) | Accuracy88.8 | 206 | |
| Graph Classification | PTC-MR | Accuracy59.4 | 153 | |
| Graph Classification | PTC FM | Accuracy62.2 | 59 | |
| Graph Classification | PTC FR | Accuracy68 | 26 | |
| Graph Classification | PTC MM | Accuracy66.1 | 26 | |
| Graph Classification | PTC MR (10-fold cross val) | Accuracy59.4 | 21 | |
| Node Classification | AIFB | Accuracy97.2 | 7 | |
| Node Classification | MUTAG | Accuracy85.3 | 7 | |
| Graph Classification | PTC FR (10-fold cross val) | Accuracy68 | 5 |
Showing 10 of 14 rows