Revisiting Instance-Optimal Cluster Recovery in the Labeled Stochastic Block Model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ariu, Kaito, Proutiere, Alexandre, Yun, Se-Young
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911128301338624
author Ariu, Kaito
Proutiere, Alexandre
Yun, Se-Young
author_facet Ariu, Kaito
Proutiere, Alexandre
Yun, Se-Young
contents In this paper, we investigate the problem of recovering hidden communities in the Labeled Stochastic Block Model (LSBM) with a finite number of clusters whose sizes grow linearly with the total number of nodes. We derive the necessary and sufficient conditions under which the expected number of misclassified nodes is less than $ s $, for any number $ s = o(n) $. To achieve this, we propose IAC (Instance-Adaptive Clustering), the first algorithm whose performance matches the instance-specific lower bounds both in expectation and with high probability. IAC is a novel two-phase algorithm that consists of a one-shot spectral clustering step followed by iterative likelihood-based cluster assignment improvements. This approach is based on the instance-specific lower bound and notably does not require any knowledge of the model parameters, including the number of clusters. By performing the spectral clustering only once, IAC maintains an overall computational complexity of $ \mathcal{O}(n\, \text{polylog}(n)) $, making it scalable and practical for large-scale problems.
format Preprint
id arxiv_https___arxiv_org_abs_2306_12968
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Revisiting Instance-Optimal Cluster Recovery in the Labeled Stochastic Block Model
Ariu, Kaito
Proutiere, Alexandre
Yun, Se-Young
Social and Information Networks
Machine Learning
In this paper, we investigate the problem of recovering hidden communities in the Labeled Stochastic Block Model (LSBM) with a finite number of clusters whose sizes grow linearly with the total number of nodes. We derive the necessary and sufficient conditions under which the expected number of misclassified nodes is less than $ s $, for any number $ s = o(n) $. To achieve this, we propose IAC (Instance-Adaptive Clustering), the first algorithm whose performance matches the instance-specific lower bounds both in expectation and with high probability. IAC is a novel two-phase algorithm that consists of a one-shot spectral clustering step followed by iterative likelihood-based cluster assignment improvements. This approach is based on the instance-specific lower bound and notably does not require any knowledge of the model parameters, including the number of clusters. By performing the spectral clustering only once, IAC maintains an overall computational complexity of $ \mathcal{O}(n\, \text{polylog}(n)) $, making it scalable and practical for large-scale problems.
title Revisiting Instance-Optimal Cluster Recovery in the Labeled Stochastic Block Model
topic Social and Information Networks
Machine Learning
url https://arxiv.org/abs/2306.12968