Deep Graph Clustering with Disentangled Representation Learning

Yifan Wang, Yuntai Ding, Yiyang Gu, Ziyue Qiao, Chong Chen, Xian-Sheng Hua, Ming Zhang, Wei Ju

Abstract: Deep graph clustering, which aims to uncover the underlying structure within graphs and partition nodes into distinct groups, is a challenging research spot. However, the formation of the cluster in real-world graphs typically governed by the highly complex interaction of many underlying latent factors. Existing methods typically rely on the features and structure associated with the graph, and neglect the entanglement of these factors, resulting in sub-optimal clustering performance. In this paper, we propose a novel deep graph clustering framework named DisenCluster, which learns disentangled representations to simultaneously consider node separation results from diverse perspectives. Specifically, we introduce a disentangled graph encoder that iteratively identifies the latent factors of the input graph by modeling the distribution over different factors for each edge. Subsequently, we utilize a factor-wise contrastive loss to encourage clustering-friendly disentangled representations, allowing us to derive different clustering results based on the corresponding factor. These results are then structured as anchor graphs and seamlessly integrated into a unified graph. Finally, we formulate the framework as a continuous relaxation of the high-order graph cut problem and optimize the objective to obtain effective cluster assignments. Results from experiments on a variety of publicly available datasets further reveal the effectiveness and superiority of our DisenCluster compared with baselines.

External IDs:doi:10.1145/3746027.3755826