Community detection based on BernNet graph convolutional neural network

Hui Xie,Yixin Ning,Ning, Yixin
DOI: https://doi.org/10.1007/s40042-023-00823-9
2023-08-16
Journal of the Korean Physical Society
Abstract:As an important part of network analysis, community detection has attracted extensive attention of researchers in various fields. Using Graph Neural Network (GNN) model to solve the problem of community detection is a new direction of community detection research. BernNet is a graph convolutional neural network based on Bernstein polynomial approximation of order , which can be used to design arbitrary filters, but it is slow because of its quadratic dependence on . In this paper, a novel algorithm using first-order Bernstein polynomial is proposed: (i) approximate the definition of filters, takes the original labels of data as the basis for community division; (ii) defines the loss function between the labels predicted by the algorithm and the original data labels, (iii) acquires the probability model of node labels predicted by the node characteristic matrix with the smallest loss function through backpropagation. The results of community segmentation are evaluated using Normalized Mutual Information (NMI). Compared with the state-of-the-art methods, the experimental results show that: the algorithm can achieve better community detection results on real data sets and reduce the complexity of the algorithm while obtaining better results.
physics, multidisciplinary
What problem does this paper attempt to address?