3 years ago

GEMSEC: Graph Embedding with Self Clustering.

Benedek Rozemberczki, Ryan Davies, Rik Sarkar, Charles Sutton

Network embedding procedures localize nodes of a graph in a low dimensional feature space, which enables machine learning on graph data. In this paper we propose GEMSEC - a graph embedding algorithm which learns a clustering of the nodes simultaneously with the embedding. Clusters or communities are natural constructs in social networks, and incorporating this objective is found to produce more representative embeddings than traditional methods. We find that GEMSEC extracts high quality clusters on real world social networks and outperforms existing embedding algorithms on clustering tasks. The embedding procedure preserves the neighborhoods of individual vertices, while at the same time, it forms k different clusters for a given k. GEMSEC generalizes earlier works in the domain. It augments the skip-gram like embedding procedures and is agnostic of the neighborhood sampling strategy. In experiments, the joint embedding and clustering is found to be extremely scalable, with computation costs linear in the size of the input.

Publisher URL: http://arxiv.org/abs/1802.03997

DOI: arXiv:1802.03997v2

You might also like
Discover & Discuss Important Research

Keeping up-to-date with research can feel impossible, with papers being published faster than you'll ever be able to read them. That's where Researcher comes in: we're simplifying discovery and making important discussions happen. With over 19,000 sources, including peer-reviewed journals, preprints, blogs, universities, podcasts and Live events across 10 research areas, you'll never miss what's important to you. It's like social media, but better. Oh, and we should mention - it's free.

  • Download from Google Play
  • Download from App Store
  • Download from AppInChina

Researcher displays publicly available abstracts and doesn’t host any full article content. If the content is open access, we will direct clicks from the abstracts to the publisher website and display the PDF copy on our platform. Clicks to view the full text will be directed to the publisher website, where only users with subscriptions or access through their institution are able to view the full article.