用无学习二值编码让大模型文本检索快16.7倍、内存降16倍
LLMs Meet Isolation Kernel: Lightweight, Learning-free Binary Embeddings for Fast Retrieval
- 基于隔离核的无学习二值化方法,直接转换大模型嵌入为二进制码
- 在多个数据集上实现16.7倍加速、16倍内存压缩,精度接近原模型
- 适合需要低延迟、低存储的实时检索场景,兼容图索引结构
大型语言模型(LLMs)虽显著提升了文本表征能力,但其高维嵌入带来巨大存储与检索开销。尽管马特罗什卡表示学习(MRL)和对比稀疏表示(CSR)等方法部分缓解此问题,仍存在检索精度下降的缺陷。本文提出无学习的隔离核嵌入(IKE),通过隔离核(IK)将LLM嵌入转化为二值嵌入。该方法轻量且基于二进制编码,内存占用极低,支持快速位运算,显著降低检索延迟。在多个文本检索数据集上的实验表明,IKE相较原始LLM嵌入实现最高16.7倍的检索加速与16倍的内存降低,同时保持相当的准确性。理论分析证明,IKE满足有效二值哈希的四个关键条件,而其他方法不具备。相比CSR,IKE在效率与效果上均更优,且与基于图的索引兼容,在近似最近邻(ANN)搜索中展现出更好的精度-延迟平衡性能。
原文摘要 · Abstract (English)
Large language models (LLMs) have recently enabled remarkable progress in text representation. However, their embeddings are typically high-dimensional, leading to substantial storage and retrieval overhead. Although recent approaches such as Matryoshka Representation Learning (MRL) and Contrastive Sparse Representation (CSR) alleviate these issues to some extent, they still suffer from retrieval accuracy degradation. This paper proposes Isolation Kernel Embedding or IKE, a learning-free method that transforms an LLM embedding into a binary embedding using Isolation Kernel (IK). Lightweight and based on binary encoding, IKE offers a low memory footprint and fast bitwise computation, lowering retrieval latency. Experiments on multiple text retrieval datasets demonstrate that IKE offers up to 16.7x faster retrieval and 16x lower memory usage than the original LLM embeddings, while maintaining comparable accuracy. Theoretically, we show that IKE works because it satisfies four essential criteria for effective binary hashing that other methods do not possess. Compared to CSR, IKE consistently achieves better retrieval efficiency and effectiveness. IKE also works effectively with graph-based indexing, demonstrating its superiority in balancing accuracy and latency compared to alternative compression techniques in the approximate nearest neighbor (ANN) search setting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。