Fast Sequence-Based Embedding with Diffusion Graphs

Benedek Rozemberczki, Rik Sarkar
1/21/2020
cs.LGcs.DCcs.SI

Official code: benedekrozemberczki/diff2vec (127 stars, GPL-3.0, TensorFlow, last commit 2022-11-06).

Abstract

A graph embedding is a representation of graph vertices in a low-dimensional space, which approximately preserves properties such as distances between nodes. Vertex sequence-based embedding procedures use features extracted from linear sequences of nodes to create embeddings using a neural network. In this paper, we propose diffusion graphs as a method to rapidly generate vertex sequences for network embedding. Its computational efficiency is superior to previous methods due to simpler sequence generation, and it produces more accurate results. In experiments, we found that the performance relative to other methods improves with increasing edge density in the graph. In a community detection task, clustering nodes in the embedding space produces better results compared to other sequence-based embedding methods.

View on arXivView PDF

Code Implementations(1)

TensorFlowGPL-3.0

Reference implementation of Diffusion2Vec (Complenet 2018) built on Gensim and NetworkX.

12714PythonOct 13, 20173 years agoGPL-3.0
complex-networksdeep-learningdeepwalkdiff2vecdiffusion+15 more

Similar Papers

Cite this paper

@article{rozemberczki2020fast,
  title  = {Fast Sequence-Based Embedding with Diffusion Graphs},
  author = {Benedek Rozemberczki and Rik Sarkar},
  year   = {2020},
  eprint = {2001.07463},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2001.07463}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2001.07463.svg)](https://codeofpaper.com/paper/2001.07463)

Discussion