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

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

TaskDatasetResultRank
Graph ClassificationMUTAG
Accuracy88.8
1229
Graph ClassificationPTC-MR
Accuracy59.4
271
Graph ClassificationMUTAG (10-fold cross-validation)
Accuracy88.8
236
Graph ClassificationPTC FM
Accuracy62.2
70
Graph ClassificationPTC MR (10-fold cross val)
Accuracy59.4
30
Graph ClassificationPTC FR
Accuracy68
26
Graph ClassificationPTC MM
Accuracy66.1
26
Node ClassificationAIFB
Accuracy97.2
7
Node ClassificationMUTAG
Accuracy85.3
7
Graph ClassificationPTC FR (10-fold cross val)
Accuracy68
5
Showing 10 of 14 rows

Other info

Code

Follow for update