Improving Query Representations for Dense Retrieval with Pseudo Relevance Feedback
arXiv:2108.13454
Abstract
Dense retrieval systems conduct first-stage retrieval using embedded representations and simple similarity metrics to match a query to documents. Its effectiveness depends on encoded embeddings to capture the semantics of queries and documents, a challenging task due to the shortness and ambiguity of search queries. This paper proposes ANCE-PRF, a new query encoder that uses pseudo relevance feedback (PRF) to improve query representations for dense retrieval. ANCE-PRF uses a BERT encoder that consumes the query and the top retrieved documents from a dense retrieval model, ANCE, and it learns to produce better query embeddings directly from relevance labels. It also keeps the document index unchanged to reduce overhead. ANCE-PRF significantly outperforms ANCE and other recent dense retrieval systems on several datasets. Analysis shows that the PRF encoder effectively captures the relevant and complementary information from PRF documents, while ignoring the noise with its learned attention mechanism.
arXiv commentAccepted at CIKM 2021
References in corpus (7)
- End-to-End Neural Ad-hoc Ranking with Kernel Pooling
- Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text Retrieval
- Overview of the TREC 2020 deep learning track
- Distilling Dense Representations for Ranking using Tightly-Coupled Teachers
- Overview of the TREC 2021 deep learning track
- Learning To Retrieve: How to Train a Dense Retrieval Model Effectively and Efficiently
- Co-BERT: A Context-Aware BERT Retrieval Model Incorporating Local and Query-specific Context