VEC-SBM: Optimal Community Detection with Vectorial Edges Covariates

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Braun, Guillaume, Sugiyama, Masashi
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910348340101120
author Braun, Guillaume
Sugiyama, Masashi
author_facet Braun, Guillaume
Sugiyama, Masashi
contents Social networks are often associated with rich side information, such as texts and images. While numerous methods have been developed to identify communities from pairwise interactions, they usually ignore such side information. In this work, we study an extension of the Stochastic Block Model (SBM), a widely used statistical framework for community detection, that integrates vectorial edges covariates: the Vectorial Edges Covariates Stochastic Block Model (VEC-SBM). We propose a novel algorithm based on iterative refinement techniques and show that it optimally recovers the latent communities under the VEC-SBM. Furthermore, we rigorously assess the added value of leveraging edge's side information in the community detection process. We complement our theoretical results with numerical experiments on synthetic and semi-synthetic data.
format Preprint
id arxiv_https___arxiv_org_abs_2402_18805
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle VEC-SBM: Optimal Community Detection with Vectorial Edges Covariates
Braun, Guillaume
Sugiyama, Masashi
Social and Information Networks
Machine Learning
Social networks are often associated with rich side information, such as texts and images. While numerous methods have been developed to identify communities from pairwise interactions, they usually ignore such side information. In this work, we study an extension of the Stochastic Block Model (SBM), a widely used statistical framework for community detection, that integrates vectorial edges covariates: the Vectorial Edges Covariates Stochastic Block Model (VEC-SBM). We propose a novel algorithm based on iterative refinement techniques and show that it optimally recovers the latent communities under the VEC-SBM. Furthermore, we rigorously assess the added value of leveraging edge's side information in the community detection process. We complement our theoretical results with numerical experiments on synthetic and semi-synthetic data.
title VEC-SBM: Optimal Community Detection with Vectorial Edges Covariates
topic Social and Information Networks
Machine Learning
url https://arxiv.org/abs/2402.18805