Boosting Graph Pooling with Persistent Homology

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ying, Chaolong, Zhao, Xinjian, Yu, Tianshu
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910655107301376
author Ying, Chaolong
Zhao, Xinjian
Yu, Tianshu
author_facet Ying, Chaolong
Zhao, Xinjian
Yu, Tianshu
contents Recently, there has been an emerging trend to integrate persistent homology (PH) into graph neural networks (GNNs) to enrich expressive power. However, naively plugging PH features into GNN layers always results in marginal improvement with low interpretability. In this paper, we investigate a novel mechanism for injecting global topological invariance into pooling layers using PH, motivated by the observation that filtration operation in PH naturally aligns graph pooling in a cut-off manner. In this fashion, message passing in the coarsened graph acts along persistent pooled topology, leading to improved performance. Experimentally, we apply our mechanism to a collection of graph pooling methods and observe consistent and substantial performance gain over several popular datasets, demonstrating its wide applicability and flexibility.
format Preprint
id arxiv_https___arxiv_org_abs_2402_16346
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Boosting Graph Pooling with Persistent Homology
Ying, Chaolong
Zhao, Xinjian
Yu, Tianshu
Machine Learning
Algebraic Topology
Recently, there has been an emerging trend to integrate persistent homology (PH) into graph neural networks (GNNs) to enrich expressive power. However, naively plugging PH features into GNN layers always results in marginal improvement with low interpretability. In this paper, we investigate a novel mechanism for injecting global topological invariance into pooling layers using PH, motivated by the observation that filtration operation in PH naturally aligns graph pooling in a cut-off manner. In this fashion, message passing in the coarsened graph acts along persistent pooled topology, leading to improved performance. Experimentally, we apply our mechanism to a collection of graph pooling methods and observe consistent and substantial performance gain over several popular datasets, demonstrating its wide applicability and flexibility.
title Boosting Graph Pooling with Persistent Homology
topic Machine Learning
Algebraic Topology
url https://arxiv.org/abs/2402.16346