4bit-Quantization in Vector-Embedding for RAG
Retrieval-augmented generation (RAG) is a promising technique that has shown great potential in addressing some of the limitations of large language models (LLMs). One of the major limitations of LLMs is that they can suffer from outdated information, as their training data may not always reflect the latest information. Additionally, LLMs can also be prone to hallucinations, where they generate responses that are not grounded in fact. RAG aims to mitigate these issues by leveraging a database of relevant documents, which are stored as embedding vectors in a high-dimensional space. However, one of the challenges of using high-dimensional embeddings is that they require a significant amount of memory to store. This can be a major issue, especially when dealing with large databases of documents. To alleviate this problem, we propose the use of 4-bit quantization to store the embedding vectors. This involves reducing the precision of the vectors from 32-bit floating-point numbers to 4-bit integers, which can significantly reduce the memory requirements. Our approach has several benefits. Firstly, it significantly reduces the memory storage requirements of the high-dimensional vector database, making it more feasible to deploy RAG systems in resource-constrained environments. Secondly, it speeds up the searching process, as the reduced precision of the vectors allows for faster computation. Our code is available at https://github.com/taeheej/4bit-Quantization-in-Vector-Embedding-for-RAG.
Authors
- Taehee Jeong (ORCID: https://orcid.org/0000-0002-6427-8410)
Institutions
- San Jose State University (US)
Publication Details
- Published
- 2024-12-18
- DOI
- https://doi.org/10.1109/icmla61862.2024.00156
- Citations
- 3
- Primary Topic
- Sparse and Compressive Sensing Techniques
- Type
- article
- Field-Weighted Citation Impact
- 1.12