Exponential Family Graph Embeddings

Description

We emphasize on exponential family distributions to capture rich interaction patterns between nodes in random walk sequences. We introduce the generic exponential family graph embedding model, that generalizes random walk-based network representation learning techniques to exponential family conditional distributions. We study three particular instances of this model, analyzing their properties and showing their relationship to existing unsupervised learning models. Our experimental evaluation on real-world datasets demonstrates that the proposed techniques outperform well-known baseline methods in two downstream machine learning tasks.
Dolphins Network
DeepWalk
EFGE-Bern
EFGE-Pois
EFGE-Norm
The Dolphins network composed by 2 communities and the corresponding embeddings for d=2

Code

An implementation of the project in C++ can be reached at the Github repository.

Contributors

Abdulkadir Ƈelikkanat and Fragkiskos D. Malliaros

References

A. Celikkanat and F. D. Malliaros, Exponential Family Graph Embeddings, The Thirty-Fourth AAAI Conference on Artificial Intelligence (AAAI-20), New York City, New York, 2020.