Semi-supervised community detection method based on generative adversarial networks
Xiaoyang Liu,Mengyao Zhang,Yanfei Liu,Chao Liu,Chaorong Li,Wei Wang,Xiaoqin Zhang,Asgarali Bouyer
DOI: https://doi.org/10.1016/j.jksuci.2024.102008
IF: 9.006
2024-03-01
Journal of King Saud University - Computer and Information Sciences
Abstract:Community detection in complex networks often suffers from insufficient data and limited utilization of prior knowledge. In this paper we propose “Semi-supervised Generative Adversarial Network” (GANSE), a novel algorithm that integrates Generative Adversarial Networks (GANs) and semi-supervised learning to address these challenges. This method addresses the issues above through a multi-step process. Initially, the network is rewired using vertex similarity metrics, thereby enhancing its structural integrity. Subsequently, a novel generative adversarial network model is designed, and our model facilitates the reconstruction of the network, thereby yielding partitions. Which form the basis for identifying core communities. Additionally, the local clustering coefficient is incorporated as a reward signal and injected into the node selection process. Moreover, isolated nodes are reallocated, ultimately culminating in the derivation of the final community structure. Experimental results on four large real-life datasets demonstrate the clear superiority of the proposed algorithm in terms of F1 and Jaccard metrics when compared to existing algorithms. Notably, our GANSE method outperforms the traditional algorithms in networks with “missing data”. Thus showing its robustness and effectiveness in real-world incomplete datasets. Our findings highlight the potential of GANs and semi-supervised learning for enhancing community detection accuracy in complex networks.
computer science, information systems