Variational Embeddings for Community Detection and Node Representation
release_lxu5zjpdgzh6xp5rw2ceylju44
by
Rayyan Ahmad Khan, Muhammad Umer Anwaar, Omran Kaddah, Martin Kleinsteuber
2021
Abstract
In this paper, we study how to simultaneously learn two highly correlated
tasks of graph analysis, i.e., community detection and node representation
learning. We propose an efficient generative model called VECoDeR for jointly
learning Variational Embeddings for Community Detection and node
Representation. VECoDeR assumes that every node can be a member of one or more
communities. The node embeddings are learned in such a way that connected nodes
are not only "closer" to each other but also share similar community
assignments. A joint learning framework leverages community-aware node
embeddings for better community detection. We demonstrate on several graph
datasets that VECoDeR effectively out-performs many competitive baselines on
all three tasks i.e. node classification, overlapping community detection and
non-overlapping community detection. We also show that VECoDeR is
computationally efficient and has quite robust performance with varying
hyperparameters.
In text/plain
format
Archived Files and Locations
application/pdf 1.2 MB
file_rzna4lkxg5f73jwumhuke7zoge
|
arxiv.org (repository) web.archive.org (webarchive) |
2101.03885v1
access all versions, variants, and formats of this works (eg, pre-prints)