Command Palette
Search for a command to run...
Rodrigo Nogueira; Wei Yang; Jimmy Lin; Kyunghyun Cho

Abstract
One technique to improve the retrieval effectiveness of a search engine is to expand documents with terms that are related or representative of the documents' content.From the perspective of a question answering system, this might comprise questions the document can potentially answer. Following this observation, we propose a simple method that predicts which queries will be issued for a given document and then expands it with those predictions with a vanilla sequence-to-sequence model, trained using datasets consisting of pairs of query and relevant documents. By combining our method with a highly-effective re-ranking component, we achieve the state of the art in two retrieval tasks. In a latency-critical regime, retrieval results alone (without re-ranking) approach the effectiveness of more computationally expensive neural re-rankers but are much faster.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| passage-re-ranking-on-ms-marco | BERT + Doc2query | MRR: 0.368 |
| passage-re-ranking-on-trec-pm | BERT + Doc2query | mAP: 36.5 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.