Spectral Clustering for Directed Graphs via Likelihood Estimation on Stochastic Block Models

Ning Zhang, Xiaowen Dong, Mihai Cucuringu

Abstract: Graph clustering is a fundamental task in unsupervised learning with broad real-world applications. While spectral clustering methods for undirected graphs are well-established and guided by a minimum cut optimization consensus, their extension to directed graphs remains relatively underexplored due to the additional complexity introduced by edge directions. In this paper, we leverage statistical inference on stochastic block models to guide the development of a spectral clustering algorithm for directed graphs. Specifically, we study the maximum likelihood estimation under a widely used directed stochastic block model, and derive a global objective function that aligns with the underlying community structure. Based on the spectral relaxation, we introduce a novel, self-adaptive spectral clustering method for directed graphs, and provide theoretical guarantees on the miscluster error. Extensive experiments on synthetic and real-world datasets demonstrate significant performance gains over existing baselines.

Submission Number: 425