Near-Lossless Binarization of Word Embeddings

Authors

  • Julien Tissier Université de St Etienne
  • Christophe Gravier Université de St Etienne
  • Amaury Habrard Université de St Etienne

DOI:

https://doi.org/10.1609/aaai.v33i01.33017104

Abstract

Word embeddings are commonly used as a starting point in many NLP models to achieve state-of-the-art performances. However, with a large vocabulary and many dimensions, these floating-point representations are expensive both in terms of memory and calculations which makes them unsuitable for use on low-resource devices. The method proposed in this paper transforms real-valued embeddings into binary embeddings while preserving semantic information, requiring only 128 or 256 bits for each vector. This leads to a small memory footprint and fast vector operations. The model is based on an autoencoder architecture, which also allows to reconstruct original vectors from the binary ones. Experimental results on semantic similarity, text classification and sentiment analysis tasks show that the binarization of word embeddings only leads to a loss of ∼2% in accuracy while vector size is reduced by 97%. Furthermore, a top-k benchmark demonstrates that using these binary vectors is 30 times faster than using real-valued vectors.

Downloads

Published

2019-07-17

How to Cite

Tissier, J., Gravier, C., & Habrard, A. (2019). Near-Lossless Binarization of Word Embeddings. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 7104-7111. https://doi.org/10.1609/aaai.v33i01.33017104

Issue

Section

AAAI Technical Track: Natural Language Processing