We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Learning Edge Representations via Low-Rank Asymmetric Projections

Abstract: We propose a new method for embedding graphs while preserving directed edge information. Learning such continuous-space vector representations (or embeddings) of nodes in a graph is an important first step for using network information (from social networks, user-item graphs, knowledge bases, etc.) in many machine learning tasks.
Unlike previous work, we (1) explicitly model an edge as a function of node embeddings, and we (2) propose a novel objective, the "graph likelihood", which contrasts information from sampled random walks with non-existent edges. Individually, both of these contributions improve the learned representations, especially when there are memory constraints on the total size of the embeddings. When combined, our contributions enable us to significantly improve the state-of-the-art by learning more concise representations that better preserve the graph structure.
We evaluate our method on a variety of link-prediction task including social networks, collaboration networks, and protein interactions, showing that our proposed method learn representations with error reductions of up to 76% and 55%, on directed and undirected graphs. In addition, we show that the representations learned by our method are quite space efficient, producing embeddings which have higher structure-preserving accuracy but are 10 times smaller.
Subjects: Machine Learning (cs.LG); Social and Information Networks (cs.SI); Machine Learning (stat.ML)
Journal reference: ACM International Conference on Information and Knowledge Management, 2017
DOI: 10.1145/3132847.3132959
Cite as: arXiv:1705.05615 [cs.LG]
  (or arXiv:1705.05615v4 [cs.LG] for this version)

Submission history

From: Sami Abu-El-Haija [view email]
[v1] Tue, 16 May 2017 09:44:28 GMT (764kb,D)
[v2] Wed, 17 May 2017 23:15:39 GMT (768kb,D)
[v3] Mon, 29 May 2017 12:00:41 GMT (1411kb,D)
[v4] Wed, 13 Sep 2017 18:21:14 GMT (1280kb,D)

Link back to: arXiv, form interface, contact.