Search papers, labs, and topics across Lattice.
This paper introduces Single-stage Sparse Retrieval (SSR), a novel multi-vector retrieval approach that replaces K-means clustering with sparse coding using a Sparse Autoencoder (SAE). SSR projects token embeddings into a high-dimensional, sparse representation, enabling the use of inverted indexing and bypassing the need for computationally expensive clustering. Experiments on BEIR show that SSR achieves a 15x reduction in indexing time, halves retrieval latency, and improves retrieval performance compared to ColBERTv2.
Ditch K-means: sparse coding slashes indexing time by 15x while simultaneously boosting retrieval accuracy in multi-vector retrieval.
Multi-vector retrieval (MVR) models, exemplified by ColBERT, have established new benchmarks in retrieval accuracy by preserving fine-grained token-level interactions. However, this granularity imposes prohibitive storage and retrieval efficiency bottlenecks: to manage the immense memory footprint and computational overhead of billion-scale token vectors, state-of-the-art systems are forced to rely on aggressive dimension reduction and complex clustering (e.g., K-means). This compromise introduces two critical limitations: excessive indexing latency of clustering large-scale corpora and semantic information loss inherent to compression. In this paper, we propose Single-stage Sparse Retrieval (SSR}, a paradigm shift that replaces expensive clustering with efficient sparse coding. Instead of compressing features into low-dimensional dense vectors, we utilize Sparse Autoencoder (SAE) to project token embeddings into a high-dimensional but highly sparse representation. This transformation enables us to bypass vector clustering entirely and leverage inverted indexing for precise, high-throughput retrieval. Extensive experiments on the BEIR benchmark demonstrate that SSR achieves a"trifecta"of improvements: it reduces indexing time by 15x compared to ColBERTv2, halves retrieval latency, and simultaneously improves retrieval performance over leading baselines.