arXiv:2607.18626cs.IRcs.CL2026-07

用聚类中心向量实现轻量级伪相关反馈,提升检索精度。

PLAID-PRF: Pseudo-Relevance Feedback with Centroid-like Tokens in PLAID

论文配图:PLAID-PRF: Pseudo-Relevance Feedback with Centroid-like Tokens in PLAID
图 1 · 摘自论文原文
  • 基于PLAID的聚类中心向量进行伪相关反馈,避免高计算开销。
  • 在MSMARCO和BEIR上提升4.3% nDCG@10,7.3% MRR@10。
  • 适合需要高效精准检索的场景,如工业级搜索系统。

多向量稠密检索模型如ColBERT通过建模查询与文档间的细粒度词级交互实现强大检索效果。PLAID等方法利用词向量的聚类中心量化技术降低索引大小并加速检索,同时保持高性能。本文提出PLAID-PRF,针对PLAID执行伪相关反馈(PRF),基于召回结果重构查询向量。与以往在多向量模型上进行PRF的方法不同,PLAID-PRF利用内部聚类中心向量,将其类比为传统PRF中的词项,仅需少量且多样化的高价值扩展向量,将其附加至原查询后重运行PLAID,从而优化候选生成与最终打分。在标准域MSMARCO及四个跨域BEIR基准上的实验表明,PLAID-PRF持续优于各类基线。特别地,相比PLAID,其在nDCG@10上最高提升4.3%,在MRR@10上提升7.3%,同时引入的计算开销远低于先前方法。结果证明,所提出的聚类中心感知的伪相关反馈机制是一种有效且轻量的提升前排名结果质量的方法。整体上,该工作实现了无需昂贵查询时文档-词项聚类的高效反馈感知晚期交互检索。

原文摘要 · Abstract (English)

Multi-vector dense retrieval models, such as ColBERT, achieve strong retrieval effectiveness by modelling fine-grained token-level interactions between queries and documents. Methods such as PLAID use centroid-based quantisation of each token's vector to reduce the index size and speed up retrieval while maintaining strong effectiveness. In this work, we introduce PLAID-PRF, a method that performs Pseudo-Relevance Feedback (PRF) over PLAID to reformulate ColBERT's query vectors based on the top-retrieved results. In contrast with prior methods that perform PRF on multi-vector retrieval models, PLAID-PRF keeps computational costs low by leveraging the internal PLAID centroid vectors, treating them similarly to tokens in traditional PRF methods. The method selects a small and diverse set of high-utility expansion vectors and appends them to the original query, rerunning PLAID to refine both candidate generation and final scoring. Extensive experiments on the standard in-domain MSMARCO and four out-of-domain BEIR benchmarks show that PLAID-PRF consistently improves retrieval effectiveness over various baselines. In particular, PLAID-PRF improves over PLAID by up to 4.3% nDCG@10 and 7.3% MRR@10, while introducing substantially less computation overhead than prior PRF methods. The results demonstrate that our proposed centroid-aware PRF method offers an effective and lightweight mechanism to improve the quality of top-ranked retrieved results. Overall, this work enables effective and efficient feedback-aware late-interaction retrieval without expensive query-time document-token clustering.

检索增强伪相关反馈稠密检索PLAID

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