Motif-Based Contrastive Learning for Community Detection

• Published in: IEEE transaction on neural networks and learning systems 2024-09, Vol.35 (9), p.11706-11719
• Main Authors: Wu, Xunxun, Wang, Chang-Dong, Lin, Jia-Qi, Xi, Wu-Dong, Yu, Philip S.
• Format: Article
• Language: English
• Subjects: Community detection; complex network; Complex networks; Computer science; contrastive learning; Deep learning; Image edge detection; Matrix decomposition; motif; Self-supervised learning; Tensors
• ISSN: 2162-237X; 2162-2388; 2162-2388
• DOI: 10.1109/TNNLS.2024.3367873

Community detection has become a prominent task in complex network analysis. However, most of the existing methods for community detection only focus on the lower order structure at the level of individual nodes and edges and ignore the higher order connectivity patterns that characterize the fundamental building blocks within the network. In recent years, researchers have shown interest in motifs and their role in network analysis. However, most of the existing higher order approaches are based on shallow methods, failing to capture the intricate nonlinear relationships between nodes. In order to better fuse higher order and lower order structural information, a novel deep learning framework called motif-based contrastive learning for community detection (MotifCC) is proposed. First, a higher order network is constructed based on motifs. Subnetworks are then obtained by removing isolated nodes, addressing the fragmentation issue in the higher order network. Next, the concept of contrastive learning is applied to effectively fuse various kinds of information from nodes, edges, and higher order and lower order structures. This aims to maximize the similarity of corresponding node information, while distinguishing different nodes and different communities. Finally, based on the community structure of subnetworks, the community labels of all nodes are obtained by using the idea of label propagation. Extensive experiments on real-world datasets validate the effectiveness of MotifCC.