Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

RocketQA: An Optimized Training Approach to Dense Passage Retrieval for Open-Domain Question Answering

About

In open-domain question answering, dense passage retrieval has become a new paradigm to retrieve relevant passages for finding answers. Typically, the dual-encoder architecture is adopted to learn dense representations of questions and passages for semantic matching. However, it is difficult to effectively train a dual-encoder due to the challenges including the discrepancy between training and inference, the existence of unlabeled positives and limited training data. To address these challenges, we propose an optimized training approach, called RocketQA, to improving dense passage retrieval. We make three major technical contributions in RocketQA, namely cross-batch negatives, denoised hard negatives and data augmentation. The experiment results show that RocketQA significantly outperforms previous state-of-the-art models on both MSMARCO and Natural Questions. We also conduct extensive experiments to examine the effectiveness of the three strategies in RocketQA. Besides, we demonstrate that the performance of end-to-end QA can be improved based on our RocketQA retriever.

Yingqi Qu, Yuchen Ding, Jing Liu, Kai Liu, Ruiyang Ren, Wayne Xin Zhao, Daxiang Dong, Hua Wu, Haifeng Wang• 2020

Related benchmarks

TaskDatasetResultRank
Passage retrievalMsMARCO (dev)
MRR@1037
116
RetrievalMS MARCO (dev)
MRR@100.37
84
Passage RankingMS MARCO (dev)
MRR@1040.9
73
RetrievalNatural Questions (test)
Top-5 Recall74
62
Passage retrievalNatural Questions (NQ) (test)
Top-20 Accuracy82.7
45
Information RetrievalMS MARCO in-domain--
18
Open-domain Question AnsweringNaturalQuestions (NQ) v1.0 (test)
Acc@2082.7
11
Passage ReadingNQ
EM42.8
10
Passage RerankingMS-MARCO passage ranking August 11, 2021 (test)
MRR@1042.6
3
Showing 9 of 9 rows

Other info

Code

Follow for update