NeurIPS 2019spotlight50 citations

Numerically Accurate Hyperbolic Embeddings Using Tiling-Based Models

Tao Yu, Christopher M De Sa

Abstract

Hyperbolic embeddings achieve excellent performance when embedding hierarchical data structures like synonym or type hierarchies, but they can be limited by numerical error when ordinary floating-point numbers are used to represent points in hyperbolic space. Standard models such as the Poincar{\'e} disk and the Lorentz model have unbounded numerical error as points get far from the origin. To address this, we propose a new model which uses an integer-based tiling to represent \emph{any} point in hyperbolic space with provably bounded numerical error. This allows us to learn high-precision embeddings without using BigFloats, and enables us to store the resulting embeddings with fewer bits. We evaluate our tiling-based model empirically, and show that it can both compress hyperbolic embeddings (down to $2\%$ of a Poincar{\'e} embedding on WordNet Nouns) and learn more accurate embeddings on real-world datasets.

BibTeX
@inproceedings{NEURIPS2019_82c25591,
 author = {Yu, Tao and De Sa, Christopher M},
 booktitle = {Advances in Neural Information Processing Systems},
 editor = {H. Wallach and H. Larochelle and A. Beygelzimer and F. d\textquotesingle Alch\'{e}-Buc and E. Fox and R. Garnett},
 pages = {},
 publisher = {Curran Associates, Inc.},
 title = {Numerically Accurate Hyperbolic Embeddings Using Tiling-Based Models},
 url = {https://proceedings.neurips.cc/paper_files/paper/2019/file/82c2559140b95ccda9c6ca4a8b981f1e-Paper.pdf},
 volume = {32},
 year = {2019}
}
Numerically Accurate Hyperbolic Embeddings Using Tiling-Based Models · NeurIPS 2019