π€ AI Summary
This work addresses the challenge of maintaining high recall in filtered vector search under varying predicates and selectivities, where traditional quantization methods suffer due to their reliance on fixed codebooks. The paper introduces, for the first time, an instance-optimized approach tailored to this setting, proposing a query- and predicate-aware dynamic codebook generation method. For each query, the method constructs a dedicated codebook by jointly leveraging the queryβs vector characteristics and its associated filter conditions, replacing the static product quantization codebook. This dynamic strategy preserves compact storage while significantly enhancing ranking accuracy in the compressed domain. Extensive experiments demonstrate consistent superiority over baseline methods across diverse selectivity levels, effectively improving the trade-off between queries per second (QPS) and recall.
π Abstract
Filtered vector search (FVS) is increasingly important in modern AI + DB systems, where vector similarity search is combined with relational predicates. Quantization plays a vital role in these systems by enabling query processing over large vector datasets. However, lossy approaches, e.g., Product Quantization (PQ), incur a precision penalty during distance calculation, thereby negatively impacting the query recall performance. This problem becomes more challenging in FVS because the relevant vector space can change with the relational predicate and selectivity. Motivated by the success of instance-optimized database system components, we introduce iFVS, an Instance-Optimized Filtered Vector Search technique. Given a fixed, quantized vector dataset, and a representative workload of filtered vector queries, iFVS adopts a query-specific codebook generation approach for FVS that is instance-optimized towards a certain dataset and query workload. Instead of using a fixed codebook for all queries, iFVS conditions distance estimation on both the query vector and the filter predicate. This enables more accurate ranking over compressed vectors while preserving compact per-vector storage. Experiments show that iFVS improves the Queries Per Second (QPS)-recall tradeoff across several filter selectivity bins compared with fixed-codebook quantized FVS baselines.