LEANN: A Low-Storage Vector Index

NLP VDIT RecSys / IR DRSA Neural Search AI Systems and Hardware EDIO VDRAD
基于嵌入的向量搜索支撑着许多重要应用,例如推荐系统和检索增强生成(RAG)。该技术依赖向量索引来实现高效搜索。然而,这些索引需要存储高维嵌入向量以及庞大的索引元数据,其总体规模可能达到原始数据(如文本片段)大小的数倍。如此高的存储开销使得在个人设备或大规模数据集上部署向量搜索变得困难甚至不切实际。为解决这一问题,我们提出了LEANN——一种面向向量搜索的高效存储索引方案,它通过即时重新计算嵌入向量而非预先存储,并对最先进的近似图索引进行压缩,同时保持搜索精度。LEANN能够在仅使用极小部分存储空间(例如仅为原始数据的5%)的情况下实现高质量的向量搜索,并支持存储高效的索引构建与更新。在真实场景的基准测试中,与传统索引相比,LEANN可将索引大小减少高达50倍,同时在RAG应用中保持最先进的搜索精度和相当的响应延迟。
Embedding-based vector search underpins many important applications, such as recommendation and retrieval-augmented generation (RAG). It relies on vector indices to enable efficient search. However, these indices require storing high-dimensional embeddings and large index metadata, whose total size can be several times larger than the original data (e.g., text chunks). Such high storage overhead makes it difficult, or even impractical, to deploy vector search on personal devices or large-scale datasets. To tackle this problem, we propose LEANN, a storage-efficient index for vector search that recomputes embeddings on the fly instead of storing them, and compresses state-of-the-art proximity graph indices while preserving search accuracy. LEANN delivers high-quality vector search while using only a fraction of the storage (e.g., 5% of the original data) and supporting storage-efficient index construction and updates. On real-world benchmarks, LEANN reduces index size by up to 50x compared with conventional indices, while maintaining SOTA accuracy and comparable latency for RAG applications.
许愿