Deep Graph Clustering via Dual Correlation Reduction

Authors

  • Yue Liu National University of Defense Technology
  • Wenxuan Tu National University of Defense Technology
  • Sihang Zhou National University of Defense Technology
  • Xinwang Liu National University of Defense Technology
  • Linxuan Song National University of Defense Technology
  • Xihong Yang National University of Defense Technology
  • En Zhu National University of Defense Technology

DOI:

https://doi.org/10.1609/aaai.v36i7.20726

Keywords:

Machine Learning (ML)

Abstract

Deep graph clustering, which aims to reveal the underlying graph structure and divide the nodes into different groups, has attracted intensive attention in recent years. However, we observe that, in the process of node encoding, existing methods suffer from representation collapse which tends to map all data into the same representation. Consequently, the discriminative capability of the node representation is limited, leading to unsatisfied clustering performance. To address this issue, we propose a novel self-supervised deep graph clustering method termed Dual Correlation Reduction Network (DCRN) by reducing information correlation in a dual manner. Specifically, in our method, we first design a siamese network to encode samples. Then by forcing the cross-view sample correlation matrix and cross-view feature correlation matrix to approximate two identity matrices, respectively, we reduce the information correlation in the dual-level, thus improving the discriminative capability of the resulting features. Moreover, in order to alleviate representation collapse caused by over-smoothing in GCN, we introduce a propagation regularization term to enable the network to gain long-distance information with the shallow network structure. Extensive experimental results on six benchmark datasets demonstrate the effectiveness of the proposed DCRN against the existing state-of-the-art methods. The code of DCRN is available at https://github.com/yueliu1999/DCRN and a collection (papers, codes and, datasets) of deep graph clustering is shared at https://github.com/yueliu1999/Awesome-Deep-Graph-Clustering on Github.

Downloads

Published

2022-06-28

How to Cite

Liu, Y., Tu, W., Zhou, S., Liu, X., Song, L., Yang, X., & Zhu, E. (2022). Deep Graph Clustering via Dual Correlation Reduction. Proceedings of the AAAI Conference on Artificial Intelligence, 36(7), 7603-7611. https://doi.org/10.1609/aaai.v36i7.20726

Issue

Section

AAAI Technical Track on Machine Learning II