Variational Bayesian Inference Algorithms for Infinite Relational Model of Network Data

IEEE Trans Neural Netw Learn Syst. 2015 Sep;26(9):2176-81. doi: 10.1109/TNNLS.2014.2362012. Epub 2014 Oct 28.

Abstract

Network data show the relationship among one kind of objects, such as social networks and hyperlinks on the Web. Many statistical models have been proposed for analyzing these data. For modeling cluster structures of networks, the infinite relational model (IRM) was proposed as a Bayesian nonparametric extension of the stochastic block model. In this brief, we derive the inference algorithms for the IRM of network data based on the variational Bayesian (VB) inference methods. After showing the standard VB inference, we derive the collapsed VB (CVB) inference and its variant called the zeroth-order CVB inference. We compared the performances of the inference algorithms using six real network datasets. The CVB inference outperformed the VB inference in most of the datasets, and the differences were especially larger in dense networks.