Saved in:
Bibliographic Details
Main Authors: Li, Xiang, Zhao, Yunpeng, Pan, Qing, Hao, Ning
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2412.03780
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910728158445568
author Li, Xiang
Zhao, Yunpeng
Pan, Qing
Hao, Ning
author_facet Li, Xiang
Zhao, Yunpeng
Pan, Qing
Hao, Ning
contents Community detection is the task of clustering objects based on their pairwise relationships. Most of the model-based community detection methods, such as the stochastic block model and its variants, are designed for networks with binary (yes/no) edges. In many practical scenarios, edges often possess continuous weights, spanning positive and negative values, which reflect varying levels of connectivity. To address this challenge, we introduce the heterogeneous block covariance model (HBCM) that defines a community structure within the covariance matrix, where edges have signed and continuous weights. Furthermore, it takes into account the heterogeneity of objects when forming connections with other objects within a community. A novel variational expectation-maximization algorithm is proposed to estimate the group membership. The HBCM provides provable consistent estimates of memberships, and its promising performance is observed in numerical simulations with different setups. The model is applied to a single-cell RNA-seq dataset of a mouse embryo and a stock price dataset. Supplementary materials for this article are available online.
format Preprint
id arxiv_https___arxiv_org_abs_2412_03780
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Community Detection with Heterogeneous Block Covariance Model
Li, Xiang
Zhao, Yunpeng
Pan, Qing
Hao, Ning
Machine Learning
Computation
Community detection is the task of clustering objects based on their pairwise relationships. Most of the model-based community detection methods, such as the stochastic block model and its variants, are designed for networks with binary (yes/no) edges. In many practical scenarios, edges often possess continuous weights, spanning positive and negative values, which reflect varying levels of connectivity. To address this challenge, we introduce the heterogeneous block covariance model (HBCM) that defines a community structure within the covariance matrix, where edges have signed and continuous weights. Furthermore, it takes into account the heterogeneity of objects when forming connections with other objects within a community. A novel variational expectation-maximization algorithm is proposed to estimate the group membership. The HBCM provides provable consistent estimates of memberships, and its promising performance is observed in numerical simulations with different setups. The model is applied to a single-cell RNA-seq dataset of a mouse embryo and a stock price dataset. Supplementary materials for this article are available online.
title Community Detection with Heterogeneous Block Covariance Model
topic Machine Learning
Computation
url https://arxiv.org/abs/2412.03780