No More K-means: Single-Stage Sparse Coding for Efficient Multi-Vector Retrieval
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 51 of its 67 references that are in the map. First posted on arXiv 28 May 2026.
Builds on
51 of 67 references.
References · 51
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2016MS MARCO: A Human Generated MAchine Reading COmprehension Datasetcited by 1,591
- 2018FEVER: a large-scale dataset for Fact Extraction and VERificationcited by 1,362
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2021BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Modelscited by 1,225
- 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,175
- 2020ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERTcited by 1,160
- 2023Towards General Text Embeddings with Multi-stage Contrastive Learningcited by 705
- 2022MTEB: Massive Text Embedding Benchmarkcited by 623
- 2020Fact or Fiction: Verifying Scientific Claimscited by 546
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2021ColBERTv2: Effective and Efficient Retrieval via Lightweight Late Interactioncited by 464
38 more
- 2020SPECTER: Document-level Representation Learning using Citation-informed Transformerscited by 426
- 2024The Faiss librarycited by 370
- 2023Improving Text Embeddings with Large Language Modelscited by 295
- 2021SPLADE: Sparse Lexical and Expansion Model for First Stage Rankingcited by 283
- 2024Jumping Ahead: Improving Reconstruction Fidelity with JumpReLU Sparse Autoencoderscited by 283
- 2022Matryoshka Representation Learningcited by 268
- 2020CLIMATE-FEVER: A Dataset for Verification of Real-World Climate Claimscited by 195
- 2024Improving Dictionary Learning with Gated Sparse Autoencoderscited by 171
- 2024jina-embeddings-v3: Multilingual Embeddings With Task LoRAcited by 138
- 2021SPLADE v2: Sparse Lexical and Expansion Model for Information Retrievalcited by 126
- 2024BatchTopK Sparse Autoencoderscited by 126
- 2021COIL: Revisit Exact Lexical Match in Information Retrieval with Contextualized Inverted Listcited by 113
- 2025On the Theoretical Limitations of Embedding-Based Retrievalcited by 98
- 2022PLAID: An Efficient Engine for Late Interaction Retrievalcited by 95
- 2025Llama-Embed-Nemotron-8B: A Universal Text Embedding Model for Multilingual and Cross-Lingual Taskscited by 46
- 2025KaLM-Embedding-V2: Superior Training Techniques and Data Inspire A Versatile Embedding Modelcited by 35
- 2024Calibrating Multi-modal Representations: A Pursuit of Group Robustness without Annotationscited by 30
- 2024MUVERA: Multi-Vector Retrieval via Fixed Dimensional Encodingscited by 30
- 2024Counting Ability of Large Language Models and Impact of Tokenizationcited by 29
- 2023Rethinking the Role of Token Retrieval in Multi-Vector Retrievalcited by 28
- 2024Answer is All You Need: Instruction-following Text Embedding via Answering the Questioncited by 26
- 2025Beyond Matryoshka: Revisiting Sparse Coding for Adaptive Representationcited by 25
- 2025Uncovering Memorization Effect in the Presence of Spurious Correlationscited by 20
- 2024Non-negative Contrastive Learningcited by 18
- 2024Linq-Embed-Mistral Technical Reportcited by 18
- 2023Attention Calibration for Transformer-based Sequential Recommendationcited by 15
- 2025WARP: An Efficient Engine for Multi-Vector Retrievalcited by 15
- 2022Aligning Source Visual and Target Language Domains for Unpaired Video Captioningcited by 14
- 2023Multi-task Item-attribute Graph Pre-training for Strict Cold-start Item Recommendationcited by 13
- 2022Multi-Vector Retrieval as Sparse Alignmentcited by 12
- 2022DESSERT: An Efficient Algorithm for Vector Set Search with Vector Set Queriescited by 10
- 2025CRISP: Clustering Multi-Vector Representations for Denoising and Pruningcited by 8
- 2024Efficient Document Ranking with Learnable Late Interactionscited by 6
- 2024UniHGKR: Unified Instruction-aware Heterogeneous Knowledge Retrieverscited by 5
- 2026CSRv2: Unlocking Ultra-Sparse Embeddingscited by 5
- 2025Memory Retrieval and Consolidation in Large Language Models through Function Tokenscited by 4
- 2026When to Think, When to Speak: Learning Disclosure Policies for LLM Reasoningcited by 4
- 2024Efficient Multi-Vector Dense Retrieval Using Bit Vectorscited by 1
50 most cited of 51 All 51 (JSON)
Built on by
Not cited yet.