English

CliquePH: Higher-Order Information for Graph Neural Networks through Persistent Homology on Clique Graphs

Machine Learning 2024-11-27 v2 Artificial Intelligence

Abstract

Graph neural networks have become the default choice by practitioners for graph learning tasks such as graph classification and node classification. Nevertheless, popular graph neural network models still struggle to capture higher-order information, i.e., information that goes \emph{beyond} pairwise interactions. Recent work has shown that persistent homology, a tool from topological data analysis, can enrich graph neural networks with topological information that they otherwise could not capture. Calculating such features is efficient for dimension 0 (connected components) and dimension 1 (cycles). However, when it comes to higher-order structures, it does not scale well, with a complexity of O(nd)O(n^d), where nn is the number of nodes and dd is the order of the structures. In this work, we introduce a novel method that extracts information about higher-order structures in the graph while still using the efficient low-dimensional persistent homology algorithm. On standard benchmark datasets, we show that our method can lead to up to 31%31\% improvements in test accuracy.

Keywords

Cite

@article{arxiv.2409.08217,
  title  = {CliquePH: Higher-Order Information for Graph Neural Networks through Persistent Homology on Clique Graphs},
  author = {Davide Buffelli and Farzin Soleymani and Bastian Rieck},
  journal= {arXiv preprint arXiv:2409.08217},
  year   = {2024}
}

Comments

Published in Proceedings of the Third Learning on Graphs Conference (LoG 2024), PMLR 269