Estimating Stochastic Block Models in the Presence of Covariates

2024/02/26 by Kitamura, Yuichi, Laage, Louise
#Econometrics (econ.EM) #FOS: Economics and business

paper · doi:10.48550/arxiv.2402.16322

Abstract

In the standard stochastic block model for networks, the probability of a connection between two nodes, often referred to as the edge probability, depends on the unobserved communities each of these nodes belongs to. We consider a flexible framework in which each edge probability, together with the probability of community assignment, are also impacted by observed covariates. We propose a computationally tractable two-step procedure to estimate the conditional edge probabilities as well as the community assignment probabilities. The first step relies on a spectral clustering algorithm applied to a localized adjacency matrix of the network. In the second step, k-nearest neighbor regression estimates are computed on the extracted communities. We study the statistical properties of these estimators by providing non-asymptotic bounds.

Related

• Spectral inference for large Stochastic Blockmodels with nodal covariates
• On spectral algorithms for community detection in stochastic blockmodel graphs with vertex covariates
• Determining the Number of Communities in Degree-corrected Stochastic Block Models
• Stochastic Block Models with Multiple Continuous Attributes
• Edge Label Inference in Generalized Stochastic Block Models: from Spectral Theory to Impossibility Results