Pyserini: Reproducing DPR Results
July 13, 2026 · View on GitHub
Dense passage retriever (DPR) is a dense retrieval method described in the following paper:
Vladimir Karpukhin, Barlas Oğuz, Sewon Min, Patrick Lewis, Ledell Wu, Sergey Edunov, Danqi Chen, Wen-tau Yih. Dense Passage Retrieval for Open-Domain Question Answering. Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP), pages 6769-6781, 2020.
We have replicated DPR results and incorporated the technique into Pyserini. Our own efforts are described in the following paper:
Xueguang Ma, Kai Sun, Ronak Pradeep, Minghan Li, and Jimmy Lin. Another Look at DPR: Reproduction of Training and Replication of Retrieval. Proceedings of the 44th European Conference on Information Retrieval (ECIR 2022), Part I, pages 613-626, April 2021, Stavanger, Norway.
Which evolved from a previous arXiv preprint:
Xueguang Ma, Kai Sun, Ronak Pradeep, and Jimmy Lin. A Replication Study of Dense Passage Retriever. arXiv:2104.05740, April 2021.
To be clear, we started with model checkpoint releases in the official DPR repo and did not retrain the query and passage encoders from scratch. Our implementation does not share any code with the DPR repo, other than evaluation scripts to ensure that results are comparable.
This guide provides instructions to reproduce our replication study. Our efforts include both retrieval and end-to-end answer extraction, but we only cover retrieval here.
Note that we often observe minor differences in scores between different computing environments (e.g., Linux vs. macOS). However, the differences usually appear in the fifth digit after the decimal point, and do not appear to be a cause for concern from a reproducibility perspective. Thus, while the scoring script provides results to much higher precision, we have intentionally rounded to four digits after the decimal point.
Summary
Here's how our results stack up against results reported in the paper using the DPR-Multi model:
| Dataset | Method | Top-20 (orig) | Top-20 (us) | Top-100 (orig) | Top-100 (us) |
|---|---|---|---|---|---|
| NQ | DPR | 79.4 | 79.5 | 86.0 | 86.1 |
| NQ | BM25 | 59.1 | 62.9 | 73.7 | 78.3 |
| NQ | Hybrid | 78.0 | 82.6 | 83.9 | 88.6 |
| TriviaQA | DPR | 78.8 | 78.9 | 84.7 | 84.8 |
| TriviaQA | BM25 | 66.9 | 76.4 | 76.7 | 83.2 |
| TriviaQA | Hybrid | 79.9 | 82.6 | 84.4 | 86.6 |
| WQ | DPR | 75.0 | 75.1 | 82.9 | 83.0 |
| WQ | BM25 | 55.0 | 62.4 | 71.1 | 75.5 |
| WQ | Hybrid | 74.7 | 77.1 | 82.3 | 84.4 |
| CuratedTREC | DPR | 89.1 | 88.8 | 93.9 | 93.4 |
| CuratedTREC | BM25 | 70.9 | 80.7 | 84.1 | 89.9 |
| CuratedTREC | Hybrid | 88.5 | 90.1 | 94.1 | 95.0 |
| SQuAD | DPR | 51.6 | 52.0 | 67.6 | 67.7 |
| SQuAD | BM25 | 68.8 | 71.1 | 80.0 | 81.8 |
| SQuAD | Hybrid | 66.2 | 75.1 | 78.6 | 84.4 |
The hybrid results reported above for "us" capture what we call the "norm" condition (see paper for details). Note that the results below represent the current state of the code base, where there may be minor differences in effectiveness from what's reported in the paper.
Natural Questions (NQ) with DPR-Multi
DPR retrieval with brute-force index:
python -m pyserini.search.faiss \
--index wikipedia-dpr-100w.dpr-multi \
--topics dpr-nq-test \
--encoder facebook/dpr-question_encoder-multiset-base \
--output runs/run.dpr.nq-test.multi.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-nq-test \
--input runs/run.dpr.nq-test.multi.trec \
--output runs/run.dpr.nq-test.multi.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.nq-test.multi.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.7947
Top100 accuracy: 0.8609
BM25 retrieval:
python -m pyserini.search.lucene \
--index wikipedia-dpr-100w \
--topics dpr-nq-test \
--output runs/run.dpr.nq-test.bm25.trec
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-nq-test \
--input runs/run.dpr.nq-test.bm25.trec \
--output runs/run.dpr.nq-test.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.nq-test.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.6294
Top100 accuracy: 0.7825
Hybrid dense-sparse retrieval (combining above two approaches):
python -m pyserini.search.hybrid \
dense --index wikipedia-dpr-100w.dpr-multi \
--encoder facebook/dpr-question_encoder-multiset-base \
sparse --index wikipedia-dpr-100w \
fusion --alpha 1.3 \
run --topics dpr-nq-test \
--output runs/run.dpr.nq-test.multi.bm25.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-nq-test \
--input runs/run.dpr.nq-test.multi.bm25.trec \
--output runs/run.dpr.nq-test.multi.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.nq-test.multi.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.8260
Top100 accuracy: 0.8859
TriviaQA with DPR-Multi
DPR retrieval with brute-force index:
python -m pyserini.search.faiss \
--index wikipedia-dpr-100w.dpr-multi \
--topics dpr-trivia-test \
--encoder facebook/dpr-question_encoder-multiset-base \
--output runs/run.dpr.trivia-test.multi.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-trivia-test \
--input runs/run.dpr.trivia-test.multi.trec \
--output runs/run.dpr.trivia-test.multi.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.trivia-test.multi.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.7887
Top100 accuracy: 0.8479
BM25 retrieval:
python -m pyserini.search.lucene \
--index wikipedia-dpr-100w \
--topics dpr-trivia-test \
--output runs/run.dpr.trivia-test.bm25.trec
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-trivia-test \
--input runs/run.dpr.trivia-test.bm25.trec \
--output runs/run.dpr.trivia-test.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.trivia-test.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.7641
Top100 accuracy: 0.8315
Hybrid dense-sparse retrieval (combining above two approaches):
python -m pyserini.search.hybrid \
dense --index wikipedia-dpr-100w.dpr-multi \
--encoder facebook/dpr-question_encoder-multiset-base \
sparse --index wikipedia-dpr-100w \
fusion --alpha 0.95 \
run --topics dpr-trivia-test \
--output runs/run.dpr.trivia-test.multi.bm25.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-trivia-test \
--input runs/run.dpr.trivia-test.multi.bm25.trec \
--output runs/run.dpr.trivia-test.multi.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.trivia-test.multi.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.8264
Top100 accuracy: 0.8655
WebQuestions (WQ) with DPR-Multi
DPR retrieval with brute-force index:
python -m pyserini.search.faiss \
--index wikipedia-dpr-100w.dpr-multi \
--topics dpr-wq-test \
--encoder facebook/dpr-question_encoder-multiset-base \
--output runs/run.dpr.wq-test.multi.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-wq-test \
--input runs/run.dpr.wq-test.multi.trec \
--output runs/run.dpr.wq-test.multi.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.wq-test.multi.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.7505
Top100 accuracy: 0.8297
BM25 retrieval:
python -m pyserini.search.lucene \
--index wikipedia-dpr-100w \
--topics dpr-wq-test \
--output runs/run.dpr.wq-test.bm25.trec
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-wq-test \
--input runs/run.dpr.wq-test.bm25.trec \
--output runs/run.dpr.wq-test.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.wq-test.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.6240
Top100 accuracy: 0.7549
Hybrid dense-sparse retrieval (combining above two approaches):
python -m pyserini.search.hybrid \
dense --index wikipedia-dpr-100w.dpr-multi \
--encoder facebook/dpr-question_encoder-multiset-base \
sparse --index wikipedia-dpr-100w \
fusion --alpha 0.95 \
run --topics dpr-wq-test \
--output runs/run.dpr.wq-test.multi.bm25.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-wq-test \
--input runs/run.dpr.wq-test.multi.bm25.trec \
--output runs/run.dpr.wq-test.multi.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.wq-test.multi.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.7712
Top100 accuracy: 0.8440
CuratedTREC with DPR-Multi
DPR retrieval with brute-force index:
python -m pyserini.search.faiss \
--index wikipedia-dpr-100w.dpr-multi \
--topics dpr-curated-test \
--encoder facebook/dpr-question_encoder-multiset-base \
--output runs/run.dpr.curated-test.multi.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-curated-test \
--input runs/run.dpr.curated-test.multi.trec \
--output runs/run.dpr.curated-test.multi.json \
--regex
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.curated-test.multi.json \
--topk 20 100 \
--regex
And the expected results:
Top20 accuracy: 0.8876
Top100 accuracy: 0.9337
BM25 retrieval:
python -m pyserini.search.lucene \
--index wikipedia-dpr-100w \
--topics dpr-curated-test \
--output runs/run.dpr.curated-test.bm25.trec
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-curated-test \
--input runs/run.dpr.curated-test.bm25.trec \
--output runs/run.dpr.curated-test.bm25.json \
--regex
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.curated-test.bm25.json \
--topk 20 100 \
--regex
And the expected results:
Top20 accuracy: 0.8069
Top100 accuracy: 0.8991
Hybrid dense-sparse retrieval (combining above two approaches):
python -m pyserini.search.hybrid \
dense --index wikipedia-dpr-100w.dpr-multi \
--encoder facebook/dpr-question_encoder-multiset-base \
sparse --index wikipedia-dpr-100w \
fusion --alpha 1.05 \
run --topics dpr-curated-test \
--output runs/run.dpr.curated-test.multi.bm25.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-curated-test \
--input runs/run.dpr.curated-test.multi.bm25.trec \
--output runs/run.dpr.curated-test.multi.bm25.json \
--regex
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.curated-test.multi.bm25.json \
--topk 20 100 \
--regex
And the expected results:
Top20 accuracy: 0.9006
Top100 accuracy: 0.9496
SQuAD with DPR-Multi
DPR retrieval with brute-force index:
python -m pyserini.search.faiss \
--index wikipedia-dpr-100w.dpr-multi \
--topics dpr-squad-test \
--encoder facebook/dpr-question_encoder-multiset-base \
--output runs/run.dpr.squad-test.multi.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-squad-test \
--input runs/run.dpr.squad-test.multi.trec \
--output runs/run.dpr.squad-test.multi.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.squad-test.multi.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.5199
Top100 accuracy: 0.6773
BM25 retrieval:
python -m pyserini.search.lucene \
--index wikipedia-dpr-100w \
--topics dpr-squad-test \
--output runs/run.dpr.squad-test.bm25.trec
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-squad-test \
--input runs/run.dpr.squad-test.bm25.trec \
--output runs/run.dpr.squad-test.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.squad-test.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.7109
Top100 accuracy: 0.8184
Hybrid dense-sparse retrieval (combining above two approaches):
python -m pyserini.search.hybrid \
dense --index wikipedia-dpr-100w.dpr-multi \
--encoder facebook/dpr-question_encoder-multiset-base \
sparse --index wikipedia-dpr-100w \
fusion --alpha 2.00 \
run --topics dpr-squad-test \
--output runs/run.dpr.squad-test.multi.bm25.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-squad-test \
--input runs/run.dpr.squad-test.multi.bm25.trec \
--output runs/run.dpr.squad-test.multi.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.squad-test.multi.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.7511
Top100 accuracy: 0.8436
Natural Questions (NQ) with DPR-Single
DPR retrieval with brute-force index:
python -m pyserini.search.faiss \
--index wikipedia-dpr-100w.dpr-single-nq \
--topics dpr-nq-test \
--encoder facebook/dpr-question_encoder-single-nq-base \
--output runs/run.dpr.nq-test.single.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--index wikipedia-dpr-100w \
--topics dpr-nq-test \
--input runs/run.dpr.nq-test.single.trec \
--output runs/run.dpr.nq-test.single.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.nq-test.single.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.8006
Top100 accuracy: 0.8609
Hybrid dense-sparse retrieval:
python -m pyserini.search.hybrid \
dense --index wikipedia-dpr-100w.dpr-single-nq \
--encoder facebook/dpr-question_encoder-single-nq-base \
sparse --index wikipedia-dpr-100w \
fusion --alpha 1.2 \
run --topics dpr-nq-test \
--output runs/run.dpr.nq-test.single.bm25.trec \
--batch-size 512 --threads 16
To evaluate, first convert the TREC output format to DPR's json format:
python -m pyserini.eval.convert_trec_run_to_dpr_retrieval_run \
--topics dpr-nq-test \
--index wikipedia-dpr-100w \
--input runs/run.dpr.nq-test.single.bm25.trec \
--output runs/run.dpr.nq-test.single.bm25.json
python -m pyserini.eval.evaluate_dpr_retrieval \
--retrieval runs/run.dpr.nq-test.single.bm25.json \
--topk 20 100
And the expected results:
Top20 accuracy: 0.8288
Top100 accuracy: 0.8837
Reproduction Log*
- Results reproduced by @lintool on 2021-02-12 (commit
52a1e7) - Results reproduced by @lintool on 2021-04-21 (commit
2adbf1) - Results reproduced by @ArthurChen189 on 2021-06-09 (commit
5e8b91) - Results reproduced by @mayankanand007 on 2021-07-28 (commit
b2b353) - Results reproduced by @vivianliu0 on 2022-01-20 (commit
67d0a6) - Results reproduced by @manveertamber on 2022-01-22 (commit
ef70c6) - Results reproduced by @lintool on 2022-12-25 (commit
0c495c) - Results reproduced by @lintool on 2023-01-10 (commit
7dafc4) - Results reproduced by @lintool on 2024-10-07 (commit
3f7609) - Results reproduced by @lintool on 2026-06-22 (commit
65b1bbb) - Results reproduced by @lintool on 2026-07-02 (commit
ef7398b)