Cluster-Guided Contrastive Graph Clustering Network

Authors

  • Xihong Yang National University of Defense Technology
  • Yue Liu National University of Defense Technology
  • Sihang Zhou National University of Defense Technology
  • Siwei Wang National University of Defense Technology
  • Wenxuan Tu National University of Defense Technology
  • Qun Zheng University of Science and Technology of China
  • Xinwang Liu National University of Defense Technology
  • Liming Fang Nanjing University of Aeronautics and Astronautics
  • En Zhu National University of Defense Technology

DOI:

https://doi.org/10.1609/aaai.v37i9.26285

Keywords:

ML: Clustering, DMKM: Graph Mining, Social Network Analysis & Community Mining, ML: Graph-based Machine Learning, ML: Multi-Instance/Multi-View Learning

Abstract

Benefiting from the intrinsic supervision information exploitation capability, contrastive learning has achieved promising performance in the field of deep graph clustering recently. However, we observe that two drawbacks of the positive and negative sample construction mechanisms limit the performance of existing algorithms from further improvement. 1) The quality of positive samples heavily depends on the carefully designed data augmentations, while inappropriate data augmentations would easily lead to the semantic drift and indiscriminative positive samples. 2) The constructed negative samples are not reliable for ignoring important clustering information. To solve these problems, we propose a Cluster-guided Contrastive deep Graph Clustering network (CCGC) by mining the intrinsic supervision information in the high-confidence clustering results. Specifically, instead of conducting complex node or edge perturbation, we construct two views of the graph by designing special Siamese encoders whose weights are not shared between the sibling sub-networks. Then, guided by the high-confidence clustering information, we carefully select and construct the positive samples from the same high-confidence cluster in two views. Moreover, to construct semantic meaningful negative sample pairs, we regard the centers of different high-confidence clusters as negative samples, thus improving the discriminative capability and reliability of the constructed sample pairs. Lastly, we design an objective function to pull close the samples from the same cluster while pushing away those from other clusters by maximizing and minimizing the cross-view cosine similarity between positive and negative samples. Extensive experimental results on six datasets demonstrate the effectiveness of CCGC compared with the existing state-of-the-art algorithms. The code of CCGC is available at https://github.com/xihongyang1999/CCGC on Github.

Downloads

Published

2023-06-26

How to Cite

Yang, X., Liu, Y., Zhou, S., Wang, S., Tu, W., Zheng, Q., Liu, X., Fang, L., & Zhu, E. (2023). Cluster-Guided Contrastive Graph Clustering Network. Proceedings of the AAAI Conference on Artificial Intelligence, 37(9), 10834-10842. https://doi.org/10.1609/aaai.v37i9.26285

Issue

Section

AAAI Technical Track on Machine Learning IV