Semi-supervised community detection based on non-negative matrix factorization with node popularity

Abstract

A plethora of exhaustive studies have proved that the community detection merely based on topological information often leads to relatively low accuracy. Several approaches aim to achieve performance improvement by utilizing the background information. But they ignore the effect of node degrees on the availability of prior information. In this paper, by combining the idea of graph regularization with the pairwise constraints, we present a semi-supervised non-negative matrix factorization (SSNMF) model for community detection. And then, to alleviate the influence of the heterogeneity of node degrees and community sizes, we propose an improved SSNMF model by introducing the node popularity, namely PSSNMF, which helps to utilize the prior information more effectively. At last, the extensive experiments on both artificial and real-world networks show that the proposed method improves, as expected, the accuracy of community detection, especially on networks with large degree heterogeneity and unbalanced community structure.