Equipping SBMs with RBMs: an interpretable approach for analysis of networks with covariates

Shubham Gupta, K. Gururaj, Ambedkar Dukkipati (Corresponding author), Rui M. Castro

Research output: Contribution to journalArticleAcademicpeer-review

27 Downloads (Pure)

Abstract

Networks with node covariates offer two advantages to community detection methods, namely, (i) exploit covariates to improve the quality of communities, and more importantly, (ii) interpret the discovered communities by identifying the relative importance of different covariates in them. Recent methods have almost exclusively focused on the first point above. However, the quantitative improvements offered by them are often due to complex black-box models like deep neural networks at the expense of interpretability. Approaches that focus on the second point are either domain specific or have poor performance in practice. This article proposes interpretable, domain-independent statistical models for networks with node covariates that additionally offer good quantitative performance. The proposed models equip Stochastic Block Models with Restricted Boltzmann Machines to provide interpretable insights about the communities and they support both pure and mixed community memberships. Besides providing interpretability, our approach's main strength is that it does not explicitly assume a causal direction between community memberships and node covariates, making it applicable in diverse domains. We derive efficient inference procedures for our models, which can, in some cases, run in linear time in the number of nodes and edges. Experiments on several synthetic and real-world networks demonstrate that our models achieve close to state-of-the-art performance on community detection and link prediction tasks while also providing interpretations for the discovered communities.

Original languageEnglish
Article numbercnac009
JournalJournal of Complex Networks
Volume10
Issue number2
DOIs
Publication statusPublished - 13 Apr 2022

Bibliographical note

Funding Information:
The authors SG, GK and AD would like to thank the Science and Engineering Research Board, Department of Science and Technology, Government of India, for the generous funding towards this work through the IMPRINT Project: IMP/2019/000383. This work arose from a visit of Prof. Ambedkar Dukkipati to Eurandom (TU/e), generously supported by the Dutch Stochastics - Theoretical and Applied Research cluster and Eurandom

Keywords

  • community detection
  • covariates
  • interpretability
  • link prediction
  • stochastic block models

Fingerprint

Dive into the research topics of 'Equipping SBMs with RBMs: an interpretable approach for analysis of networks with covariates'. Together they form a unique fingerprint.

Cite this