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

Download:

Current browse context:

cs.CL
new | recent | 1612

Change to browse by:

References & Citations

Computer Science > Computation and Language

Title:FastText.zip: Compressing text classification models

Abstract: We consider the problem of producing compact architectures for text classification, such that the full model fits in a limited amount of memory. After considering different solutions inspired by the hashing literature, we propose a method built upon product quantization to store word embeddings. While the original technique leads to a loss in accuracy, we adapt this method to circumvent quantization artefacts. Our experiments carried out on several benchmarks show that our approach typically requires two orders of magnitude less memory than fastText while being only slightly inferior with respect to accuracy. As a result, it outperforms the state of the art by a good margin in terms of the compromise between memory usage and accuracy.
Comments: Submitted to ICLR 2017
Subjects: Computation and Language (cs.CL); Machine Learning (cs.LG)
Cite as: arXiv:1612.03651 [cs.CL]
  (or arXiv:1612.03651v1 [cs.CL] for this version)

Submission history

From: Armand Joulin [view email]
[v1] Mon, 12 Dec 2016 12:51:03 UTC (76 KB)