Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox

Overlapping Community Detection with Graph Neural Networks

Sep 26, 2019
Oleksandr Shchur, Stephan Günnemann

Share this with someone who'll enjoy it:

Community detection is a fundamental problem in machine learning. While deep learning has shown great promise in many graphrelated tasks, developing neural models for community detection has received surprisingly little attention. The few existing approaches focus on detecting disjoint communities, even though communities in real graphs are well known to be overlapping. We address this shortcoming and propose a graph neural network (GNN) based model for overlapping community detection. Despite its simplicity, our model outperforms the existing baselines by a large margin in the task of community recovery. We establish through an extensive experimental evaluation that the proposed model is effective, scalable and robust to hyperparameter settings. We also perform an ablation study that confirms that GNN is the key ingredient to the power of the proposed model.

* The First International Workshop on Deep Learning on Graphs (In Conjunction with the 25th ACM SIGKDD Conference on Knowledge Discovery and Data Mining) 

   Access Paper Source

Share this with someone who'll enjoy it: