Efficient Vector Search in the Wild: One Model for Multi-K Queries
提出 OMEGA 方法,用单一基础模型高效处理不同 K 值的向量检索查询,免去为每个 K 值单独训练模型的开销。
INSIGHT传统 learned top-K search 需为不同 K 值训练专用模型,OMEGA 只需训练 K=1 的基础模型再动态精炼。这对大规模 embedding 检索场景意义重大——一个模型即可服务多种召回需求。
业务关联社区搜索中 embedding-based 召回是核心组件(语义匹配、短文本召回),不同场景需要不同的 top-K(粗排 K=1000 vs 精排 K=100)。OMEGA 可显著简化向量检索的模型管理。
证据相比 SOTA learned search 方法:平均延迟降低 6-33%,预处理时间仅需 16-30%,最优延迟的 1.01-1.28x。在 Database + IR + ML 交叉领域发表。