Learning Latent Block Structure in Weighted Networks

Abstract

Community detection is an important task in network analysis, in which we aim to learn a network partition that groups together vertices with similar community-level connectivity patterns. By finding such groups of vertices with similar structural roles, we extract a compact representation of the network's large-scale structure, which can facilitate its scientific interpretation and the prediction of unknown or future interactions. Popular approaches, including the stochastic block model, assume edges are unweighted, which limits their utility by throwing away potentially useful information. We introduce the `weighted stochastic block model' (WSBM), which generalizes the stochastic block model to networks with edge weights drawn from any exponential family distribution. This model learns from both the presence and weight of edges, allowing it to discover structure that would otherwise be hidden when weights are discarded or thresholded. We describe a Bayesian variational algorithm for efficiently approximating this model's posterior distribution over latent block structures. We then evaluate the WSBM's performance on both edge-existence and edge-weight prediction tasks for a set of real-world weighted networks. In all cases, the WSBM performs as well or better than the best alternatives on these tasks.

28 Pages

References in corpus (10)

• Hierarchical structure and the prediction of missing links in networks
• Stochastic blockmodels and community structure in networks
• Reaction-diffusion processes and metapopulation models in heterogeneous networks
• Missing and spurious interactions and the reconstruction of complex networks
• Phase transition in the detection of modules in sparse networks
• A Bayesian Approach to Network Modularity
• World citation and collaboration networks: uncovering the role of geography in science
• Parsimonious module inference in large networks
• A network inference method for large-scale unsupervised identification of novel drug-drug interactions
• Comparative Study for Inference of Hidden Classes in Stochastic Block Models

Cited by in corpus (4)

• Community detection in networks: A user guide
• The many facets of community detection in complex networks
• Nonparametric weighted stochastic block models
• Model selection and hypothesis testing for large-scale network models with overlapping groups