Quake: Adaptive Indexing for Vector Search

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Mohoney, Jason, Sarda, Devesh, Tang, Mengze, Chowdhury, Shihabur Rahman, Pacaci, Anil, Ilyas, Ihab F., Rekatsinas, Theodoros, Venkataraman, Shivaram
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910992872505344
author Mohoney, Jason
Sarda, Devesh
Tang, Mengze
Chowdhury, Shihabur Rahman
Pacaci, Anil
Ilyas, Ihab F.
Rekatsinas, Theodoros
Venkataraman, Shivaram
author_facet Mohoney, Jason
Sarda, Devesh
Tang, Mengze
Chowdhury, Shihabur Rahman
Pacaci, Anil
Ilyas, Ihab F.
Rekatsinas, Theodoros
Venkataraman, Shivaram
contents Vector search, the task of finding the k-nearest neighbors of a query vector against a database of high-dimensional vectors, underpins many machine learning applications, including retrieval-augmented generation, recommendation systems, and information retrieval. However, existing approximate nearest neighbor (ANN) methods perform poorly under dynamic and skewed workloads where data distributions evolve. We introduce Quake, an adaptive indexing system that maintains low latency and high recall in such environments. Quake employs a multi-level partitioning scheme that adjusts to updates and changing access patterns, guided by a cost model that predicts query latency based on partition sizes and access frequencies. Quake also dynamically sets query execution parameters to meet recall targets using a novel recall estimation model. Furthermore, Quake utilizes NUMA-aware intra-query parallelism for improved memory bandwidth utilization during search. To evaluate Quake, we prepare a Wikipedia vector search workload and develop a workload generator to create vector search workloads with configurable access patterns. Our evaluation shows that on dynamic workloads, Quake achieves query latency reductions of 1.5-38x and update latency reductions of 4.5-126x compared to state-of-the-art indexes such as SVS, DiskANN, HNSW, and SCANN.
format Preprint
id arxiv_https___arxiv_org_abs_2506_03437
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Quake: Adaptive Indexing for Vector Search
Mohoney, Jason
Sarda, Devesh
Tang, Mengze
Chowdhury, Shihabur Rahman
Pacaci, Anil
Ilyas, Ihab F.
Rekatsinas, Theodoros
Venkataraman, Shivaram
Information Retrieval
Vector search, the task of finding the k-nearest neighbors of a query vector against a database of high-dimensional vectors, underpins many machine learning applications, including retrieval-augmented generation, recommendation systems, and information retrieval. However, existing approximate nearest neighbor (ANN) methods perform poorly under dynamic and skewed workloads where data distributions evolve. We introduce Quake, an adaptive indexing system that maintains low latency and high recall in such environments. Quake employs a multi-level partitioning scheme that adjusts to updates and changing access patterns, guided by a cost model that predicts query latency based on partition sizes and access frequencies. Quake also dynamically sets query execution parameters to meet recall targets using a novel recall estimation model. Furthermore, Quake utilizes NUMA-aware intra-query parallelism for improved memory bandwidth utilization during search. To evaluate Quake, we prepare a Wikipedia vector search workload and develop a workload generator to create vector search workloads with configurable access patterns. Our evaluation shows that on dynamic workloads, Quake achieves query latency reductions of 1.5-38x and update latency reductions of 4.5-126x compared to state-of-the-art indexes such as SVS, DiskANN, HNSW, and SCANN.
title Quake: Adaptive Indexing for Vector Search
topic Information Retrieval
url https://arxiv.org/abs/2506.03437