SubSearch: Robust Estimation and Outlier Detection for Stochastic Block Models via Subgraph Search

TL;DR

SubSearch is a novel algorithm that robustly estimates stochastic block model parameters and detects outliers by searching for subgraphs that best fit the model, outperforming traditional methods especially on real-world data.

Contribution

We introduce SubSearch, a new subgraph search-based approach for robust SBM parameter estimation and outlier detection, addressing deviations in real-world network data.

Findings

Effective in synthetic and real datasets

Outperforms existing robust estimation methods

Accurately identifies outlier nodes

Abstract

Community detection is a fundamental task in graph analysis, with methods often relying on fitting models like the Stochastic Block Model (SBM) to observed networks. While many algorithms can accurately estimate SBM parameters when the input graph is a perfect sample from the model, real-world graphs rarely conform to such idealized assumptions. Therefore, robust algorithms are crucial-ones that can recover model parameters even when the data deviates from the assumed distribution. In this work, we propose SubSearch, an algorithm for robustly estimating SBM parameters by exploring the space of subgraphs in search of one that closely aligns with the model's assumptions. Our approach also functions as an outlier detection method, properly identifying nodes responsible for the graph's deviation from the model and going beyond simple techniques like pruning high-degree nodes. Extensive experiments on both synthetic and real-world datasets demonstrate the effectiveness of our method.