English

Improved clustering algorithms for the Bipartite Stochastic Block Model

Statistics Theory 2021-04-26 v3 Statistics Theory

Abstract

We establish sufficient conditions of exact and almost full recovery of the node partition in Bipartite Stochastic Block Model (BSBM) using polynomial time algorithms. First, we improve upon the known conditions of almost full recovery by spectral clustering algorithms in BSBM. Next, we propose a new computationally simple and fast procedure achieving exact recovery under milder conditions than the state of the art. Namely, if the vertex sets V1V_1 and V2V_2 in BSBM have sizes n1n_1 and n2n_2, we show that the condition p=Ω(max(logn1n1n2,logn1n2))p = \Omega\left(\max\left(\sqrt{\frac{\log{n_1}}{n_1n_2}},\frac{\log{n_1}}{n_2}\right)\right) on the edge intensity pp is sufficient for exact recovery witin V1V_1. This condition exhibits an elbow at n2n1logn1n_{2} \asymp n_1\log{n_1} between the low-dimensional and high-dimensional regimes. The suggested procedure is a variant of Lloyd's iterations initialized with a well-chosen spectral estimator leading to what we expect to be the optimal condition for exact recovery in BSBM. {The optimality conjecture is supported by showing that, for a supervised oracle procedure, such a condition is necessary to achieve exact recovery.} The key elements of the proof techniques are different from classical community detection tools on random graphs. Numerical studies confirm our theory, and show that the suggested algorithm is both very fast and achieves {almost the same} performance as the supervised oracle. Finally, using the connection between planted satisfiability problems and the BSBM, we improve upon the sufficient number of clauses to completely recover the planted assignment.

Keywords

Cite

@article{arxiv.1911.07987,
  title  = {Improved clustering algorithms for the Bipartite Stochastic Block Model},
  author = {Mohamed Ndaoud and Suzanne Sigalla and Alexandre B. Tsybakov},
  journal= {arXiv preprint arXiv:1911.07987},
  year   = {2021}
}
R2 v1 2026-06-23T12:20:01.230Z