Abstract:The stochastic block model (SBM) is a probabilistic model for community structure in networks. Typically, only the adjacency matrix is used to perform SBM parameter inference. In this paper, we consider circumstances in which nodes have an associated vector of continuous attributes that are also used to learn the node-to-community assignments and corresponding SBM parameters. While this assumption is not realistic for every application, our model assumes that the attributes associated with the nodes in a network's community can be described by a common multivariate Gaussian model. In this augmented, attributed SBM, the objective is to simultaneously learn the SBM connectivity probabilities with the multivariate Gaussian parameters describing each community. While there are recent examples in the literature that combine connectivity and attribute information to inform community detection, our model is the first augmented stochastic block model to handle multiple continuous attributes. This provides the flexibility in biological data to, for example, augment connectivity information with continuous measurements from multiple experimental modalities. Because the lack of labeled network data often makes community detection results difficult to validate, we highlight the usefulness of our model for two network prediction tasks: link prediction and collaborative filtering. As a result of fitting this attributed stochastic block model, one can predict the attribute vector or connectivity patterns for a new node in the event of the complementary source of information (connectivity or attributes, respectively). We also highlight two biological examples where the attributed stochastic block model provides satisfactory performance in the link prediction and collaborative filtering tasks.

Uniform Consistency in Stochastic Block Model with Continuous Community Label

Strong Consistency of Spectral Clustering for Stochastic Block Models.

Consistency of community detection in networks under degree-corrected stochastic block models

Monitoring networks with overlapping communities based on latent mixed-membership stochastic block model

Consistency of maximum-likelihood and variational estimators in the Stochastic Block Model

Stochastic Block Models with Multiple Continuous Attributes

Regularized Stochastic Block Model for robust community detection in complex networks

On Consistency of the MLE under Finite Mixtures of Location-Scale Distributions with a Structural Parameter

Near Universal Consistency of the Maximum Pseudolikelihood Estimator for Discrete Models

Strong Consistency, Graph Laplacians, and the Stochastic Block Model

Information-Theoretic Limits and Strong Consistency on Binary Non-uniform Hypergraph Stochastic Block Models

Restricted Tweedie Stochastic Block Models

Unraveling the Detectability of Stochastic Block Model with Overlapping Communities.

Analysis of spectral clustering algorithms for community detection: the general bipartite setting

Consistent Community Detection in Inter-Layer Dependent Multi-Layer Networks

Community detection in sparse latent space models

On Equivalence of Likelihood Maximization of Stochastic Block Model and Constrained Nonnegative Matrix Factorization

Optimal recovery by maximum and integrated conditional likelihood in the general Stochastic Block Model

Degree corrected stochastic block model: excursion representation

Two-Sample Test for Stochastic Block Models via Maximum Entry-wise Deviation