LotusFilter: Fast Diverse Nearest Neighbor Search via a Learned Cutoff Table

Fuente: arXiv
Saved in:
Bibliographic Details
Main Author: Matsui, Yusuke
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915327744409600
author Matsui, Yusuke
author_facet Matsui, Yusuke
contents Approximate nearest neighbor search (ANNS) is an essential building block for applications like RAG but can sometimes yield results that are overly similar to each other. In certain scenarios, search results should be similar to the query and yet diverse. We propose LotusFilter, a post-processing module to diversify ANNS results. We precompute a cutoff table summarizing vectors that are close to each other. During the filtering, LotusFilter greedily looks up the table to delete redundant vectors from the candidates. We demonstrated that the LotusFilter operates fast (0.02 [ms/query]) in settings resembling real-world RAG applications, utilizing features such as OpenAI embeddings. Our code is publicly available at https://github.com/matsui528/lotf.
format Preprint
id arxiv_https___arxiv_org_abs_2506_04790
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle LotusFilter: Fast Diverse Nearest Neighbor Search via a Learned Cutoff Table
Matsui, Yusuke
Computer Vision and Pattern Recognition
Information Retrieval
Machine Learning
Approximate nearest neighbor search (ANNS) is an essential building block for applications like RAG but can sometimes yield results that are overly similar to each other. In certain scenarios, search results should be similar to the query and yet diverse. We propose LotusFilter, a post-processing module to diversify ANNS results. We precompute a cutoff table summarizing vectors that are close to each other. During the filtering, LotusFilter greedily looks up the table to delete redundant vectors from the candidates. We demonstrated that the LotusFilter operates fast (0.02 [ms/query]) in settings resembling real-world RAG applications, utilizing features such as OpenAI embeddings. Our code is publicly available at https://github.com/matsui528/lotf.
title LotusFilter: Fast Diverse Nearest Neighbor Search via a Learned Cutoff Table
topic Computer Vision and Pattern Recognition
Information Retrieval
Machine Learning
url https://arxiv.org/abs/2506.04790