Topological Graph Neural Networks
Machine Learning
2022-03-18 v4 Algebraic Topology
Machine Learning
Abstract
Graph neural networks (GNNs) are a powerful architecture for tackling graph learning tasks, yet have been shown to be oblivious to eminent substructures such as cycles. We present TOGL, a novel layer that incorporates global topological information of a graph using persistent homology. TOGL can be easily integrated into any type of GNN and is strictly more expressive (in terms the Weisfeiler--Lehman graph isomorphism test) than message-passing GNNs. Augmenting GNNs with TOGL leads to improved predictive performance for graph and node classification tasks, both on synthetic data sets, which can be classified by humans using their topology but not by ordinary GNNs, and on real-world data.
Cite
@article{arxiv.2102.07835,
title = {Topological Graph Neural Networks},
author = {Max Horn and Edward De Brouwer and Michael Moor and Yves Moreau and Bastian Rieck and Karsten Borgwardt},
journal= {arXiv preprint arXiv:2102.07835},
year = {2022}
}