Litcius/Paper detail

Questions Are All You Need to Train a Dense Passage Retriever

Devendra Singh Sachan, Mike Lewis, Dani Yogatama, Luke Zettlemoyer, Joëlle Pineau, Manzil Zaheer

2023Transactions of the Association for Computational Linguistics28 citationsDOIOpen Access PDF

Abstract

Abstract We introduce ART, a new corpus-level autoencoding approach for training dense retrieval models that does not require any labeled training data. Dense retrieval is a central challenge for open-domain tasks, such as Open QA, where state-of-the-art methods typically require large supervised datasets with custom hard-negative mining and denoising of positive examples. ART, in contrast, only requires access to unpaired inputs and outputs (e.g., questions and potential answer passages). It uses a new passage-retrieval autoencoding scheme, where (1) an input question is used to retrieve a set of evidence passages, and (2) the passages are then used to compute the probability of reconstructing the original question. Training for retrieval based on question reconstruction enables effective unsupervised learning of both passage and question encoders, which can be later incorporated into complete Open QA systems without any further finetuning. Extensive experiments demonstrate that ART obtains state-of-the-art results on multiple QA retrieval benchmarks with only generic initialization from a pre-trained language model, removing the need for labeled data and task-specific losses.1 Our code and model checkpoints are available at: https://github.com/DevSinghSachan/art.

Topics & Concepts

Computer scienceInitializationCode (set theory)Task (project management)Set (abstract data type)EncoderArtificial intelligenceInformation retrievalScheme (mathematics)Training setDomain (mathematical analysis)Language modelNatural language processingMachine learningProgramming languageMathematical analysisManagementMathematicsEconomicsOperating systemTopic ModelingNatural Language Processing TechniquesMultimodal Machine Learning Applications
Questions Are All You Need to Train a Dense Passage Retriever | Litcius