arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1807.03469stat.MEmath.STstat.APstat.MLstat.TH

用于社区检测的成对协变量调整块模型

Pairwise Covariates-adjusted Block Model for Community Detection

Sihan Huang, Jiajin Sun, Yang Feng

AI总结:

针对传统随机块模型(SBM)同社区节点随机等价假设的局限性,提出纳入成对协变量的PCABM模型,设计高效求解的SCWA算法并证明其一致性,还提出模型与特征选择算法,在多类网络中表现优于SBM和DCBM。

摘要:

网络研究中最基础的问题之一是社区检测。随机块模型(stochastic block model, SBM)是一种被广泛使用的模型,目前已有多种针对该模型的估计方法,且其社区检测一致性结果已被揭示。然而,SBM受限于同一社区内所有节点随机等价的强假设,这一假设可能不适用于实际应用。我们提出了成对协变量调整随机块模型(pairwise covariates-adjusted stochastic block model, PCABM),它是SBM的一种推广形式,纳入了成对协变量信息。我们研究了协变量系数以及社区分配的最大似然估计,结果表明,在合适的稀疏性条件下,协变量系数估计和社区分配均具有一致性。我们提出了带调整的谱聚类(Spectral Clustering with Adjustment, SCWA)来高效求解PCABM,在特定条件下,我们推导了SCWA下社区检测的误差界,并证明其具有社区检测一致性。此外,我们还研究了社区数量层面的模型选择以及成对协变量的特征选择问题,并提出了两种对应的算法。在可获取协变量信息的情况下,PCABM在广泛的模拟网络和真实网络中,表现均优于SBM或度校正随机块模型(degree-corrected stochastic block model, DCBM)。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑