arXiv:2607.29606cs.IRcs.LG2026-07

根据查询自适应调整搜索策略,提升召回率稳定性并减少计算开销。

QASP: Query-Adaptive Robust Vector Search Policy

论文配图:QASP: Query-Adaptive Robust Vector Search Policy
图 1 · 摘自论文原文
  • 基于单次预训练回归预测每条查询的召回率变化曲线。
  • 在99%召回率下实现80%的数据访问量减少,且偏差极小。
  • 适合大规模向量搜索系统,尤其适用于高维数据和分层索引场景。

向量搜索的核心挑战在于在保持高召回率的同时最小化计算成本。固定搜索参数导致不同查询间性能差异显著,传统以平均召回率评估的方法掩盖了这些个体差异。我们提出QASP(Query-Adaptive Robust Vector Search Policy),通过一次前置监督回归预测每个查询的完整召回率进展曲线,从而为任意召回目标生成搜索策略;该方法避免了搜索过程中的迭代模型调用或针对每个目标单独训练预测器。通过使用尺度不变特征和预搜索推理,QASP在不同召回目标、索引配置和数据集间具备良好泛化能力。其细粒度的进度预测还支持轻量级动态补偿机制,根据预测与实际偏差实时调整搜索深度,无需额外推理。理论证明:QASP的训练样本需求量有限,不随数据集规模和维度增长;其损失仅比任何固定策略的不可约下界略高,且相较于固定探测的数据访问节省随内在维度呈指数级增长。实验表明,QASP显著降低召回率方差与目标偏差,提升查询满足率,并可在不重新训练的情况下扩展至大规模数据和层次化索引,在99%召回率下实现80%的数据访问减少。

原文摘要 · Abstract (English)

A fundamental challenge of vector search is achieving consistently high recall while minimizing computational costs. Fixed search parameters cause significant performance variance across queries, and conventional evaluation on average recall masks these per-query disparities. We introduce QASP (Query-Adaptive robust vector Search Policy), which predicts the complete recall progression curve per query via a single upfront supervised regression, from which a search policy is derived for any recall target; this avoids iterative model invocations during search or separate predictors per target. By predicting normalized recall values with scale-invariant features and pre-search inference, QASP generalizes across recall targets, index configurations, and datasets. Its fine-grained progress predictions further enable a lightweight reactive complement that adjusts search depth based on predicted-versus-observed deviations without additional inference. We prove that QASP requires a finite training sample independent of dataset size and dimensionality, that its loss exceeds the irreducible lower bound of any fixed policy by a vanishing margin, and that its data access savings over fixed probing grow exponentially in intrinsic dimensionality. Experimentally, QASP achieves significantly lower recall variance and deviation from target, higher query satisfaction rate, and scales to large data and hierarchical indices without retraining, achieving 99% recall with 80% less data access.

向量搜索自适应策略召回率优化高效检索

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。