One of the most fundamental problems in network study is community detection. The stochastic block model (SBM) is a widely used model, for which various estimation methods have been developed with their community detection consistency results unveiled. However, the SBM is restricted by the strong assumption that all nodes in the same community are stochastically equivalent, which may not be suitable for practical applications. We introduce a pairwise covariates-adjusted stochastic block model (PCABM), a generalization of SBM that incorporates pairwise covariate information. We study the maximum likelihood estimates of the coefficients for the covariates as well as the community assignments. It is shown that both the coefficient estimates of the covariates and the community assignments are consistent under suitable sparsity conditions. Spectral clustering with adjustment (SCWA) is introduced to efficiently solve PCABM. Under certain conditions, we derive the error bound of community detection under SCWA and show that it is community detection consistent. In addition, we investigate model selection in terms of the number of communities and feature selection for the pairwise covariates, and propose two corresponding algorithms. PCABM compares favorably with the SBM or degree-corrected stochastic block model (DCBM) under a wide range of simulated and real networks when covariate information is accessible.
翻译:网络研究中最基本的问题之一是社区检测。随机块模型(SBM)是一种广泛使用的模型,针对它已经开发了各种估计方法并揭示了它们的社区检测一致性结果。然而,SBM受到强烈的假设约束——同一社区内的所有节点是随机等价的,这可能不适用于实际应用。我们引入了一种成对协变量校正的随机块模型(PCABM),它是SBM的一种推广,可纳入成对协变量信息。我们研究了协变量系数和社区赋值的最大似然估计。在适当的稀疏条件下,它被证明是一致的。引入了具有调整的谱聚类(SCWA)以有效地解决PCABM。在某些条件下,我们推导了在SCWA下社区检测的误差界,并表明它是社区检测一致的。此外,我们研究了模型选择方面的工作,包括确定社区数和选择成对协变量,以及提出了两种相应的算法。当协变量信息可获得时,PCABM在各种模拟和实际网络中的表现要优于SBM或度校正随机块模型(DCBM)。