Advancing Local Clustering on Graphs via Compressive Sensing: Semi-supervised and Unsupervised Methods
Zhaiming Shen, Sung Ha Kang
TL;DR
This paper reframes local clustering on graphs as recovering sparse cluster indicators from a diffusion-like process tied to the graph Laplacian, enabling targeted discovery of small subgraphs with minimal labeling. It introduces semi-supervised Local Clustering (SSLC) and unsupervised Local Clustering (USLC), each with theoretical guarantees under mild perturbation and RIP-type conditions, and demonstrates that multiple clusters can be identified efficiently. Empirically, SSLC/USLC achieve state-of-the-art results in low-label settings across synthetic SBM benchmarks and real datasets (FashionMNIST, CIFAR-10, Planetoid graphs), while maintaining robustness to outliers and favorable runtimes. The methods expand practical applicability of local clustering to large graphs where full labeling is impractical, though they are best suited for sparse graphs and low-label scenarios.
Abstract
Local clustering aims to identify specific substructures within a large graph without any additional structural information of the graph. These substructures are typically small compared to the overall graph, enabling the problem to be approached by finding a sparse solution to a linear system associated with the graph Laplacian. In this work, we first propose a method for identifying specific local clusters when very few labeled data are given, which we term semi-supervised local clustering. We then extend this approach to the unsupervised setting when no prior information on labels is available. The proposed methods involve randomly sampling the graph, applying diffusion through local cluster extraction, then examining the overlap among the results to find each cluster. We establish the co-membership conditions for any pair of nodes, and rigorously prove the correctness of our methods. Additionally, we conduct extensive experiments to demonstrate that the proposed methods achieve state of the art results in the low-label rates regime.
