Learning Continuous Hierarchies in the Lorentz Model of Hyperbolic Geometry

By Maximilian Nickel, Douwe Kiela. International Conference on Machine Learning, 2018

604 citations recorded by the source corpus.

We are concerned with the discovery of hierarchical relationships from large-scale unstructured similarity scores. For this purpose, we study different models of hyperbolic space and find that learning embeddings in the Lorentz model is substantially more efficient than in the Poincar\'e-ball model. We show that the proposed approach allows us to learn high-quality embeddings of large taxonomies which yield improvements over Poincar\'e embeddings, especially in low dimensions. Lastly, we apply our model to discover hierarchies in two real-world datasets: we show that an embedding in hyperbolic space can reveal important aspects of a company's organizational structure as well as reveal historical relationships between language families.

Subfields: Artificial Intelligence, Molecular Biology, Statistical and Nonlinear Physics

View Paper: https://www.semanticscholar.org/paper/dc6826017bcb1f39f1bdf03c2c219e9fae804a17

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Christopher R'e
  2. Jure Leskovec
  3. Douwe Kiela
  4. Ines Chami
  5. Ivan Vulic
  6. Timothy M. Hospedales
  7. Mohit Bansal
  8. Rex Ying

In Molecular Biology

  1. Rex Ying
  2. Christopher R'e
  3. Ines Chami
  4. Qi Liu
  5. Douwe Kiela
  6. Carl Allen
  7. Kevin Chen-Chuan Chang
  8. Bryan Perozzi

In Statistical and Nonlinear Physics

  1. Rex Ying
  2. Bryan Perozzi
  3. Christopher R'e
  4. Jure Leskovec
  5. Ines Chami
  6. Qi Liu
  7. Octavian-Eugen Ganea
  8. Gary Bécigneul

Citations in the corpus