Multi-Task Retrieval for Knowledge-Intensive Tasks
Jean Maillard, Vladimir Karpukhin, Fabio Petroni, Wen-tau Yih, Barlas Oguz, Veselin Stoyanov, Gargi Ghosh
Abstract
Retrieving relevant contexts from a large corpus is a crucial step for tasks such as opendomain question answering and fact checking. Although neural retrieval outperforms traditional methods like tf-idf and BM25, its performance degrades considerably when applied to out-of-domain data. Driven by the question of whether a neural retrieval model can be universal and perform robustly on a wide variety of problems, we propose a multi-task trained model. Our approach not only surpasses previous methods in the few-shot setting, but also rivals specialised neural retrievers, even when in-domain training data is abundant. With the help of our retriever, we improve existing models for downstream tasks and closely match or improve the state of the art on multiple benchmarks. * Equal Contribution. 1 While large pre-trained neural models have been shown to incorporate real-world knowledge in their parameters and thus may skip retrieval (Petroni et al., 2019) , they still have limited capacity and suffer from a lack of explainability.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext d885214b-b801-4adc-8f91-d78e054466d6Cited by top-tier papers8
- Autoregressive Search Engines: Generating Substrings as Document IdentifiersMichele Bevilacqua, Giuseppe Ottaviano, Patrick Lewis, Scott Yih et al.NeurIPS 2022 · 242 citations
- Chain-of-Retrieval Augmented GenerationLiang Wang, Haonan Chen, Nan Yang, Xiaolong Huang et al.NeurIPS 2025 · 59 citations
- FiD-Light: Efficient and Effective Retrieval-Augmented Text GenerationSebastian Hofstätter, Jiecao Chen, Karthik Raman, Hamed ZamaniSIGIR 2023 · 47 citations
- A Unified Generative Retriever for Knowledge-Intensive Language Tasks via Prompt LearningJiangui Chen, Ruqing Zhang, Jiafeng Guo, Maarten de Rijke et al.SIGIR 2023 · 31 citations
- Robust Retrieval Augmented Generation for Zero-shot Slot FillingMichael R. Glass, Gaetano Rossiello, Md. Faisal Mahbub Chowdhury, Alfio GliozzoEMNLP 2021 · 18 citations
Builds on9
- Retrieval-Augmented Generation for Knowledge-Intensive NLP TasksPatrick Lewis, Ethan Perez, Aleksandra Piktus, Fabio Petroni et al.NeurIPS 2020 · 19,162 citations
- Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text RetrievalLee Xiong, Chenyan Xiong, Ye Li, Kwok-Fung Tang et al.ICLR 2021 · 1,547 citations
- ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERTOmar Khattab, Matei ZahariaSIGIR 2020 · 1,246 citations
- BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and ComprehensionMike Lewis, Yinhan Liu, Naman Goyal, Marjan Ghazvininejad et al.ACL 2020 · 1,224 citations
- Scalable Zero-shot Entity Linking with Dense Entity RetrievalLedell Wu, Fabio Petroni, Martin Josifoski, Sebastian Riedel et al.EMNLP 2020 · 336 citations
Related papers
- HypeR: Multitask Hyper-Prompted Training Enables Large-Scale Retrieval GeneralizationZefeng Cai, Chongyang Tao, Tao Shen, Can Xu et al.ICLR 2023
- Improving Biomedical Information Retrieval with Neural RetrieversMan Luo, Arindam Mitra, Tejas Gokhale, Chitta BaralAAAI 2022 · 42 citations
- Retrieval Augmented Language Model Pre-TrainingKelvin Guu, Kenton Lee, Zora Tung, Panupong Pasupat et al.ICML 2020 · 2,937 citations
- Chain-of-Skills: A Configurable Model for Open-Domain Question AnsweringKaixin Ma, Hao Cheng, Yu Zhang, Xiaodong Liu et al.ACL 2023 · 12 citations
- Improving Passage Retrieval with Zero-Shot Question GenerationDevendra Singh Sachan, Mike Lewis, Mandar Joshi, Armen Aghajanyan et al.EMNLP 2022 · 69 citations
