An iterative clustering algorithm for the Contextual Stochastic Block Model with optimality guarantees - Archive ouverte HAL Access content directly
Conference Papers Year :

An iterative clustering algorithm for the Contextual Stochastic Block Model with optimality guarantees

(1) , (1) , (1)
1
Guillaume Braun
  • Function : Author
  • PersonId : 1089525
Hemant Tyagi
  • Function : Author
  • PersonId : 1087784
Christophe Biernacki
  • Function : Author
  • PersonId : 923939

Abstract

Real-world networks often come with side information that can help to improve the performance of network analysis tasks such as clustering. Despite a large number of empirical and theoretical studies conducted on network clustering methods during the past decade, the added value of side information and the methods used to incorporate it optimally in clustering algorithms are relatively less understood. We propose a new iterative algorithm to cluster networks with side information for nodes (in the form of covariates) and show that our algorithm is optimal under the Contextual Symmetric Stochastic Block Model. Our algorithm can be applied to general Contextual Stochastic Block Models and avoids hyperparameter tuning in contrast to previously proposed methods. We confirm our theoretical results on synthetic data experiments where our algorithm significantly outperforms other methods, and show that it can also be applied to signed graphs. Finally we demonstrate the practical interest of our method on real data.
Fichier principal
Vignette du fichier
CSBM_Braun21.pdf (1.45 Mo) Télécharger le fichier
Origin : Files produced by the author(s)

Dates and versions

hal-03526257 , version 1 (14-01-2022)

Identifiers

Cite

Guillaume Braun, Hemant Tyagi, Christophe Biernacki. An iterative clustering algorithm for the Contextual Stochastic Block Model with optimality guarantees. Proceedings of the 39th International Conference on Machine Learning, Jul 2022, Baltimore, United States. ⟨hal-03526257⟩
58 View
47 Download

Altmetric

Share

Gmail Facebook Twitter LinkedIn More