English

Provable Estimation of the Number of Blocks in Block Models

Machine Learning 2018-03-20 v3 Methodology

Abstract

Community detection is a fundamental unsupervised learning problem for unlabeled networks which has a broad range of applications. Many community detection algorithms assume that the number of clusters rr is known apriori. In this paper, we propose an approach based on semi-definite relaxations, which does not require prior knowledge of model parameters like many existing convex relaxation methods and recovers the number of clusters and the clustering matrix exactly under a broad parameter regime, with probability tending to one. On a variety of simulated and real data experiments, we show that the proposed method often outperforms state-of-the-art techniques for estimating the number of clusters.

Keywords

Cite

@article{arxiv.1705.08580,
  title  = {Provable Estimation of the Number of Blocks in Block Models},
  author = {Bowei Yan and Purnamrita Sarkar and Xiuyuan Cheng},
  journal= {arXiv preprint arXiv:1705.08580},
  year   = {2018}
}

Comments

12 pages, 4 figure; AISTATS 2018