Graph contrastive learning with cross-encoder for community discovery

Zhenpei Shan,Defu Zhang,Yunqi Lei
DOI: https://doi.org/10.1007/s10489-024-05287-3
IF: 5.3
2024-02-02
Applied Intelligence
Abstract:Community detection is a crucial task that enables the extraction of valuable knowledge and patterns from complex networks. However, the node labels are often unavailable in most real-world applications, which poses limitations on practical implementations. It is common to deal with graphs that have side information. How to effectively fuse the topology information of a graph and the attribute information of nodes also brings great challenges to community discovery task. In this paper, a graph contrastive learning algorithm with cross-encoder is proposed for community discovery. The proposed algorithm introduces graph convolution neural network and graph attention network to fuse the topology information and the attribute information of nodes. Then it employs cross-encoder to obtain the embedding vectors of nodes and the neural networks are trained by contrastive learning which can learn the embedding vectors of nodes from different views. The proposed algorithm and the comparison algorithms are conducted on multiple real data sets. The experimental results show that the proposed algorithm outperforms the comparison algorithms on the most experimental data sets. The experimental results prove that the proposed algorithm is effective for community discovery.
computer science, artificial intelligence
What problem does this paper attempt to address?