A0885
Title: Variational inference: Posterior threshold improves network clustering accuracy in sparse regimes
Authors: Can Minh Le - University of California, Davis (United States) [presenting]
Abstract: The variational inference has been widely used in machine learning literature to fit various Bayesian models. This method has been successfully applied in network analysis to solve community detection problems. Although these results are promising, their theoretical support is only for relatively dense networks, an assumption that may not hold for real networks. In addition, it has been shown recently that the variational loss surface has many saddle points, which may severely affect its performance, especially when applied to sparse networks. The aim is to propose a simple way to improve the variational inference method by hard thresholding the posterior of the community assignment after each iteration. Using a random initialization that correlates with the true community assignment, it is shown that the proposed method converges and can accurately recover the true community labels, even when the average node degree of the network is bounded. The extensive numerical study further confirms the advantage of the proposed method over classical variational inference and another state-of-the-art algorithm.