HTG-GCL: Leveraging Hierarchical Topological Granularity from Cellular Complexes for Graph Contrastive Learning
Qirui Ji, Bin Qin, Yifan Jin, Yunze Zhao, Chuxiong Sun, Changwen Zheng, Jianwen Cao, Jiangmeng Li
TL;DR
HTG-GCL addresses the limitation of fixed topological granularity in graph contrastive learning by transforming graphs into a hierarchy of ring-based cellular complexes to capture multiple topology scales. It introduces multi-granularity decoupled contrastive learning with an uncertainty-based weighting mechanism to downweight uninformative views, and uses a common space contrast alongside granularity-specific spaces. Empirically, HTG-GCL achieves state-of-the-art performance on TU benchmarks in both unsupervised and semi-supervised settings, with ablations confirming the value of multi-granularity integration and MGDC. This work advances higher-order topology in GCL and enables adaptive, task-aware use of complex topological structures for improved graph representations.
Abstract
Graph contrastive learning (GCL) aims to learn discriminative semantic invariance by contrasting different views of the same graph that share critical topological patterns. However, existing GCL approaches with structural augmentations often struggle to identify task-relevant topological structures, let alone adapt to the varying coarse-to-fine topological granularities required across different downstream tasks. To remedy this issue, we introduce Hierarchical Topological Granularity Graph Contrastive Learning (HTG-GCL), a novel framework that leverages transformations of the same graph to generate multi-scale ring-based cellular complexes, embodying the concept of topological granularity, thereby generating diverse topological views. Recognizing that a certain granularity may contain misleading semantics, we propose a multi-granularity decoupled contrast and apply a granularity-specific weighting mechanism based on uncertainty estimation. Comprehensive experiments on various benchmarks demonstrate the effectiveness of HTG-GCL, highlighting its superior performance in capturing meaningful graph representations through hierarchical topological information.
