about
Experimental Analysis of Large-Scale Learnable Vector Storage Compression (arxiv.org)
1 point by PaulHoule on Feb 22, 2024 | hide | past | pdf | discuss on HN

In plain words: A new comparison sorts 14 ways to shrink the lookup tables that store item and word vectors, then runs them all in one shared setup for a fair test. It shows which method wins at each memory budget and where each one falls short.

Abstract · Experimental Analysis of Large-scale Learnable Vector Storage Compression

Learnable embedding vector is one of the most important applications in machine learning, and is widely used in various database-related domains. However, the high dimensionality of sparse data in recommendation tasks and the huge volume of corpus in retrieval-related tasks lead to a large memory consumption of the embedding table, which poses a great challenge to the training and deployment of models. Recent research has proposed various methods to compress the embeddings at the cost of a slight decrease in model quality or the introduction of other overheads. Nevertheless, the relative performance of these methods remains unclear. Existing experimental comparisons only cover a subset of these methods and focus on limited metrics. In this paper, we perform a comprehensive comparative analysis and experimental evaluation of embedding compression. We introduce a new taxonomy that categorizes these techniques based on their characteristics and methodologies, and further develop a modular benchmarking framework that integrates 14 representative methods. Under a uniform test environment, our benchmark fairly evaluates each approach, presents their strengths and weaknesses under different memory budgets, and recommends the best method based on the use case. In addition to providing useful guidelines, our study also uncovers the limitations of current methods and suggests potential directions for future research.

Hailin Zhang, Penghao Zhao, Xupeng Miao, Yingxia Shao, Zirui Liu, Tong Yang, Bin Cui
arXiv:2311.15578 · cs.LG, cs.DB, cs.IR · submitted Nov 27, 2023 · updated Feb 13, 2024
abstract · pdf · html

add comment on HN