|国家预印本平台
首页|LotusFilter: Fast Diverse Nearest Neighbor Search via a Learned Cutoff Table

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

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

来源:Arxiv_logoArxiv
英文摘要

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.

Yusuke Matsui

计算技术、计算机技术

Yusuke Matsui.LotusFilter: Fast Diverse Nearest Neighbor Search via a Learned Cutoff Table[EB/OL].(2025-06-05)[2025-07-16].https://arxiv.org/abs/2506.04790.点此复制

评论