English

Structure-enhanced Contrastive Learning for Graph Clustering

Machine Learning 2024-08-20 v1

Abstract

Graph clustering is a crucial task in network analysis with widespread applications, focusing on partitioning nodes into distinct groups with stronger intra-group connections than inter-group ones. Recently, contrastive learning has achieved significant progress in graph clustering. However, most methods suffer from the following issues: 1) an over-reliance on meticulously designed data augmentation strategies, which can undermine the potential of contrastive learning. 2) overlooking cluster-oriented structural information, particularly the higher-order cluster(community) structure information, which could unveil the mesoscopic cluster structure information of the network. In this study, Structure-enhanced Contrastive Learning (SECL) is introduced to addresses these issues by leveraging inherent network structures. SECL utilizes a cross-view contrastive learning mechanism to enhance node embeddings without elaborate data augmentations, a structural contrastive learning module for ensuring structural consistency, and a modularity maximization strategy for harnessing clustering-oriented information. This comprehensive approach results in robust node representations that greatly enhance clustering performance. Extensive experiments on six datasets confirm SECL's superiority over current state-of-the-art methods, indicating a substantial improvement in the domain of graph clustering.

Keywords

Cite

@article{arxiv.2408.09790,
  title  = {Structure-enhanced Contrastive Learning for Graph Clustering},
  author = {Xunlian Wu and Jingqi Hu and Anqi Zhang and Yining Quan and Qiguang Miao and Peng Gang Sun},
  journal= {arXiv preprint arXiv:2408.09790},
  year   = {2024}
}
R2 v1 2026-06-28T18:16:26.676Z