Variational Structural Deep Clustering Network

Haomin Wu,Bin Tang,Feiyu Chen,Wei Wang
DOI: https://doi.org/10.1007/978-981-97-5678-0_31
2024-01-01
Abstract:Inferring the latent embedding and clustering labels from both attribute and structural graph data is a challenging problem in machine learning. In this paper, we introduce an efficient stochastic variational inference framework, called Variational Structural Deep Clustering Network (VSDC), to sequentially predict the distribution of embedding and clustering variables. Firstly, we derive the Evidence lower bound (ELBO) of the log-likelihood function about attribute and graph variables based on some mild assumptions of independence. Secondly, we design an end-to-end neural network to approximate the intractable posterior distributions which involved in ELBO. Finally, we assign the joint distribution of embedding and clustering variables such that the parameters can be iteratively updated by minimizing the graph reconstruction error and the cross entropy between posterior distributions of latent embedding variables. The extensive experiments on several benchmark datasets empirically demonstrate the superiority of the proposed VSDC over the state-of-the-art attribute graph clustering and fusion clustering networks.
What problem does this paper attempt to address?