arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.02090stat.MEcs.SIstat.COstat.ML

带有协变量的网络的贝叶斯社区检测

Bayesian community detection for networks with covariates

Luyi Shen, Arash Amini, Nathaniel Josephs, Lizhen Lin

AI总结:

本文提出一种带有依赖协变量随机划分先验的贝叶斯随机块模型,用于密集和稀疏网络的社区检测,能通过后验推断自动学习社区数量并优于现有方法。

摘要:

网络数据在众多领域中日益普遍,以及从中提取有用信息的需求,推动了相关模型和算法的快速发展。在网络数据的各种学习任务中,社区检测,即发现节点簇或“社区”,可以说在科学界受到了最多的关注。在许多实际应用中,网络数据通常带有节点或边协变量形式的附加信息,理想情况下应利用这些信息进行推断。在本文中,我们通过提出一个带有依赖协变量的随机划分先验的贝叶斯随机块模型,为带有协变量的网络的社区检测这一有限文献增添了内容。在我们的先验下,协变量在指定簇成员资格的先验分布时被明确表达。我们的模型具有对所有参数估计(包括社区成员资格)的不确定性进行建模的灵活性。重要的是,与大多数现有方法不同,我们的模型能够通过后验推断学习社区的数量,而无需假设其为已知。我们的模型可应用于密集和稀疏网络的社区检测,同时支持分类和连续协变量,且我们的MCMC算法非常高效,具有良好的混合特性。我们在一项全面的模拟研究和两个真实数据集的应用中,展示了我们的模型相对于现有模型的优越性能。

英文摘要:

The increasing prevalence of network data in a vast variety of fields and the need to extract useful information out of them have spurred fast developments in related models and algorithms. Among the various learning tasks with network data, community detection, the discovery of node clusters or "communities," has arguably received the most attention in the scientific community. In many real-world applications, the network data often come with additional information in the form of node or edge covariates that should ideally be leveraged for inference. In this paper, we add to a limited literature on community detection for networks with covariates by proposing a Bayesian stochastic block model with a covariate-dependent random partition prior. Under our prior, the covariates are explicitly expressed in specifying the prior distribution on the cluster membership. Our model has the flexibility of modeling uncertainties of all the parameter estimates including the community membership. Importantly, and unlike the majority of existing methods, our model has the ability to learn the number of the communities via posterior inference without having to assume it to be known. Our model can be applied to community detection in both dense and sparse networks, with both categorical and continuous covariates, and our MCMC algorithm is very efficient with good mixing properties. We demonstrate the superior performance of our model over existing models in a comprehensive simulation study and an application to two real datasets.

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

↑