Graph Persistence goes Spectral
Mattie Ji, Amauri H. Souza, Vikas Garg
TL;DR
This work introduces SpectRe, a spectrum-informed persistence descriptor for graphs that augments color-based persistent homology (via RePHINE) with Laplacian spectral information. SpectRe is proven to be strictly more expressive than both RePHINE and the Laplacian spectrum, while exhibiting local stability under a curated bottleneck-like metric. The authors detail integration of SpectRe with graph neural networks using per-layer vectorization through DeepSets, and validate the approach with extensive synthetic and real-world experiments, showing robust performance gains. The method broadens the toolkit for graph representation learning by fusing topological signals with spectral information, at the cost of increased computational complexity, which can be mitigated with approximations and scheduling.
Abstract
Including intricate topological information (e.g., cycles) provably enhances the expressivity of message-passing graph neural networks (GNNs) beyond the Weisfeiler-Leman (WL) hierarchy. Consequently, Persistent Homology (PH) methods are increasingly employed for graph representation learning. In this context, recent works have proposed decorating classical PH diagrams with vertex and edge features for improved expressivity. However, these methods still fail to capture basic graph structural information. In this paper, we propose SpectRe -- a new topological descriptor for graphs that integrates spectral information into PH diagrams. Notably, SpectRe is strictly more expressive than existing descriptors on graphs. We also introduce notions of global and local stability to analyze existing descriptors and establish that SpectRe is locally stable. Finally, experiments on synthetic and real-world datasets demonstrate the effectiveness of SpectRe and its potential to enhance the capabilities of graph models in relevant learning tasks. Code is available at https://github.com/Aalto-QuML/SpectRe/.
