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

Chrome logo Add to Chrome

Firefox logo Add to Firefox

Unsupervised Community Detection with Modularity-Based Attention Model

May 20, 2019
Ivan Lobov, Sergey Ivanov

Share this with someone who'll enjoy it:

In this paper we take a problem of unsupervised nodes clustering on graphs and show how recent advances in attention models can be applied successfully in a "hard" regime of the problem. We propose an unsupervised algorithm that encodes Bethe Hessian embeddings by optimizing soft modularity loss and argue that our model is competitive to both classical and Graph Neural Network (GNN) models while it can be trained on a single graph.

   Access Paper Source

Share this with someone who'll enjoy it: