Partial recovery and weak consistency in the non-uniform hypergraph Stochastic Block Model
This addresses the problem of analyzing networks with higher-order interactions for researchers in machine learning and statistics, representing an incremental improvement over existing methods.
The paper tackles community detection in sparse random hypergraphs under the non-uniform hypergraph stochastic block model, providing a spectral algorithm that achieves at least a γ fraction of vertices correctly classified, with γ depending on the signal-to-noise ratio, and weak consistency when the SNR grows slowly, improving upon prior results.
We consider the community detection problem in sparse random hypergraphs under the non-uniform hypergraph stochastic block model (HSBM), a general model of random networks with community structure and higher-order interactions. When the random hypergraph has bounded expected degrees, we provide a spectral algorithm that outputs a partition with at least a $γ$ fraction of the vertices classified correctly, where $γ\in (0.5,1)$ depends on the signal-to-noise ratio (SNR) of the model. When the SNR grows slowly as the number of vertices goes to infinity, our algorithm achieves weak consistency, which improves the previous results in Ghoshdastidar and Dukkipati (2017) for non-uniform HSBMs. Our spectral algorithm consists of three major steps: (1) Hyperedge selection: select hyperedges of certain sizes to provide the maximal signal-to-noise ratio for the induced sub-hypergraph; (2) Spectral partition: construct a regularized adjacency matrix and obtain an approximate partition based on singular vectors; (3) Correction and merging: incorporate the hyperedge information from adjacency tensors to upgrade the error rate guarantee. The theoretical analysis of our algorithm relies on the concentration and regularization of the adjacency matrix for sparse non-uniform random hypergraphs, which can be of independent interest.