LEMUR: Learned Multi-Vector Retrieval

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Jääsaari, Elias, Hyvönen, Ville, Roos, Teemu
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917518574092288
author Jääsaari, Elias
Hyvönen, Ville
Roos, Teemu
author_facet Jääsaari, Elias
Hyvönen, Ville
Roos, Teemu
contents Multi-vector representations generated by late interaction models, such as ColBERT, enable superior retrieval quality compared to single-vector representations in information retrieval applications. In multi-vector retrieval systems, both queries and documents are encoded using one embedding per token, and similarity between queries and documents is measured by the MaxSim similarity measure. However, the improved quality of multi-vector retrieval comes at the expense of significantly increased search latency. In this work, we introduce LEMUR, a simple yet efficient framework for multi-vector similarity search. LEMUR consists of two consecutive problem reductions: First, we formulate multi-vector similarity search as a supervised learning problem that can be solved using a one-hidden-layer neural network. Second, we reduce inference under this model to single-vector similarity search in its latent space, enabling the use of existing single-vector search indexes to accelerate retrieval. LEMUR is an order of magnitude faster than prior multi-vector similarity search methods. Our code is available at https://github.com/ejaasaari/lemur
format Preprint
id arxiv_https___arxiv_org_abs_2601_21853
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle LEMUR: Learned Multi-Vector Retrieval
Jääsaari, Elias
Hyvönen, Ville
Roos, Teemu
Information Retrieval
Machine Learning
Multi-vector representations generated by late interaction models, such as ColBERT, enable superior retrieval quality compared to single-vector representations in information retrieval applications. In multi-vector retrieval systems, both queries and documents are encoded using one embedding per token, and similarity between queries and documents is measured by the MaxSim similarity measure. However, the improved quality of multi-vector retrieval comes at the expense of significantly increased search latency. In this work, we introduce LEMUR, a simple yet efficient framework for multi-vector similarity search. LEMUR consists of two consecutive problem reductions: First, we formulate multi-vector similarity search as a supervised learning problem that can be solved using a one-hidden-layer neural network. Second, we reduce inference under this model to single-vector similarity search in its latent space, enabling the use of existing single-vector search indexes to accelerate retrieval. LEMUR is an order of magnitude faster than prior multi-vector similarity search methods. Our code is available at https://github.com/ejaasaari/lemur
title LEMUR: Learned Multi-Vector Retrieval
topic Information Retrieval
Machine Learning
url https://arxiv.org/abs/2601.21853