English

Bernoulli Embeddings for Graphs

Machine Learning 2018-03-28 v1 Artificial Intelligence Machine Learning

Abstract

Just as semantic hashing can accelerate information retrieval, binary valued embeddings can significantly reduce latency in the retrieval of graphical data. We introduce a simple but effective model for learning such binary vectors for nodes in a graph. By imagining the embeddings as independent coin flips of varying bias, continuous optimization techniques can be applied to the approximate expected loss. Embeddings optimized in this fashion consistently outperform the quantization of both spectral graph embeddings and various learned real-valued embeddings, on both ranking and pre-ranking tasks for a variety of datasets.

Keywords

Cite

@article{arxiv.1803.09211,
  title  = {Bernoulli Embeddings for Graphs},
  author = {Vinith Misra and Sumit Bhatia},
  journal= {arXiv preprint arXiv:1803.09211},
  year   = {2018}
}

Comments

The Thirty-Second AAAI Conference on Artificial Intelligence (AAAI-18)

R2 v1 2026-06-23T01:04:10.793Z