Martin Fajcik, Martin Docekal, Karel Ondrej, Pavel Smrz
This work presents a novel four-stage open-domain QA pipeline R2-D2 (Rank twice, reaD twice). The pipeline is composed of a retriever, passage reranker, extractive reader, generative reader and a mechanism that aggregates the final prediction from all system's components. We demonstrate its strength across three open-domain QA datasets: NaturalQuestions, TriviaQA and EfficientQA, surpassing state-of-the-art on the first two. Our analysis demonstrates that: (i) combining extractive and generative reader yields absolute improvements up to 5 exact match and it is at least twice as effective as the posterior averaging ensemble of the same models with different parameters, (ii) the extractive reader with fewer parameters can match the performance of the generative reader on extractive QA datasets.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Question Answering | Natural Questions | EM | 55.9 | R2-D2 (full) |
| Question Answering | Natural Questions (long) | EM | 55.9 | R2-D2 \w HN-DPR |
| Question Answering | Natural Questions | Exact Match | 55.9 | R2-D2 \w HN-DPR |
| Information Retrieval | Natural Questions | Precision@100 | 88.25 | DPR+ELECTRA-large-extreader-reranker |
| Information Retrieval | Natural Questions | Precision@20 | 85.26 | DPR+ELECTRA-large-extreader-reranker |
| Information Retrieval | Natural Questions | Precision@100 | 88.03 | DPR+RoBERTa-base-crossencoder-reranker |
| Information Retrieval | Natural Questions | Precision@20 | 84.46 | DPR+RoBERTa-base-crossencoder-reranker |
| Open-Domain Question Answering | Natural Questions | Exact Match | 55.9 | R2-D2 \w HN-DPR |