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

InPars: Data Augmentation for Information Retrieval using Large Language Models

About

The information retrieval community has recently witnessed a revolution due to large pretrained transformer models. Another key ingredient for this revolution was the MS MARCO dataset, whose scale and diversity has enabled zero-shot transfer learning to various tasks. However, not all IR tasks and domains can benefit from one single dataset equally. Extensive research in various NLP tasks has shown that using domain-specific training data, as opposed to a general-purpose one, improves the performance of neural models. In this work, we harness the few-shot capabilities of large pretrained language models as synthetic data generators for IR tasks. We show that models finetuned solely on our unsupervised dataset outperform strong baselines such as BM25 as well as recently proposed self-supervised dense retrieval methods. Furthermore, retrievers finetuned on both supervised and our synthetic data achieve better zero-shot transfer than models finetuned only on supervised data. Code, models, and data are available at https://github.com/zetaalphavector/inpars .

Luiz Bonifacio, Hugo Abonizio, Marzieh Fadaee, Rodrigo Nogueira• 2022

Related benchmarks

TaskDatasetResultRank
RetrievalTREC-DL aggregate (test)
NDCG@1056.71
38
Passage RankingNQ
MRR44.49
29
Passage RankingTREC DL 2020
R@10100
28
Passage RankingWebQuestions (WQ)
R@1062.25
28
Passage retrievalNatural Questions (NQ)
Top-10 Accuracy63.51
28
Passage RankingTREC DL 2019
R@1086.66
28
Showing 6 of 6 rows

Other info

Follow for update