ML Anthology

Unsupervised Learning of Joint Embeddings for Node Representation and Community Detection

Rayyan Ahmad Khan, Muhammad Umer Anwaar, Omran Kaddah, Zhiwei Han, Martin Kleinsteuber
ECML-PKDD 2021 pp. 19-35
doi:10.1007/978-3-030-86520-7_2 /ecmlpkdd/2021/khan2021ecmlpkdd-unsupervised/

Abstract

In graph analysis community detection and node representation learning are two highly correlated tasks. In this work, we propose an efficient generative model called J-ENC for learning J oint E mbedding for N ode representation and C ommunity detection. J-ENC learns a community-aware node representation, i.e., learning of the node embeddings are constrained in such a way that connected nodes are not only “closer” to each other but also share similar community assignments. This joint learning framework leverages community-aware node embeddings for better performance on these tasks: node classification, overlapping community detection and non-overlapping community detection. We demonstrate on several graph datasets that J-ENC effectively outperforms many competitive baselines on these tasks. Furthermore, we show that J-ENC not only has quite robust performance with varying hyperparameters but also is computationally efficient than its competitors.

PDF ECML-PKDD Semantic Scholar

Cite

Text

Khan et al. "Unsupervised Learning of Joint Embeddings for Node Representation and Community Detection." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2021. doi:10.1007/978-3-030-86520-7_2

Markdown

[Khan et al. "Unsupervised Learning of Joint Embeddings for Node Representation and Community Detection." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2021.](https://mlanthology.org/ecmlpkdd/2021/khan2021ecmlpkdd-unsupervised/) doi:10.1007/978-3-030-86520-7_2

BibTeX

@inproceedings{khan2021ecmlpkdd-unsupervised,
  title     = {{Unsupervised Learning of Joint Embeddings for Node Representation and Community Detection}},
  author    = {Khan, Rayyan Ahmad and Anwaar, Muhammad Umer and Kaddah, Omran and Han, Zhiwei and Kleinsteuber, Martin},
  booktitle = {European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases},
  year      = {2021},
  pages     = {19-35},
  doi       = {10.1007/978-3-030-86520-7_2},
  url       = {https://mlanthology.org/ecmlpkdd/2021/khan2021ecmlpkdd-unsupervised/}
}