Directed Graph Auto-Encoders
Machine Learning
2022-02-28 v1 Artificial Intelligence
Abstract
We introduce a new class of auto-encoders for directed graphs, motivated by a direct extension of the Weisfeiler-Leman algorithm to pairs of node labels. The proposed model learns pairs of interpretable latent representations for the nodes of directed graphs, and uses parameterized graph convolutional network (GCN) layers for its encoder and an asymmetric inner product decoder. Parameters in the encoder control the weighting of representations exchanged between neighboring nodes. We demonstrate the ability of the proposed model to learn meaningful latent embeddings and achieve superior performance on the directed link prediction task on several popular network datasets.
Cite
@article{arxiv.2202.12449,
title = {Directed Graph Auto-Encoders},
author = {Georgios Kollias and Vasileios Kalantzis and Tsuyoshi Idé and Aurélie Lozano and Naoki Abe},
journal= {arXiv preprint arXiv:2202.12449},
year = {2022}
}
Comments
AAAI 2022