Multi-stage Training with Improved Negative Contrast for Neural Passage Retrieval
Jing Lu, Gustavo Hernández Ábrego, Ji Ma, Jianmo Ni, Yinfei Yang
Abstract
In the context of neural passage retrieval, we study three promising techniques: synthetic data generation, negative sampling and fusion. We systematically investigate how these techniques contribute to the performance of the retrieval system and how they complement each other. We propose a multi-stage framework comprising of pre-training with synthetic data, fine-tuning with labeled data and negative sampling at both stages. We study six negative sampling strategies and apply them to the fine-tuning stage and, as a noteworthy novelty, to the synthetic data that we use for pre-training. Also, we explore fusion methods that combine negatives from different strategies. We evaluate our system using two passage retrieval tasks for open-domain QA and using MS MARCO. Our experiments show that augmenting the negative contrast in both stages is effective to improve passage retrieval accuracy and, importantly, they also show that synthetic data generation and negative sampling have additive benefits. Moreover, using fusion of different kinds allows us to reach performance that establishs a new state-of-the-art level in two of the tasks we evaluated.
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 3404255d-0f1e-4323-8dfe-1ead6d12e432Cited by top-tier papers3
- Large Dual Encoders Are Generalizable RetrieversJianmo Ni, Chen Qu, Jing Lu, Zhuyun Dai et al.EMNLP 2022 · 145 citations
- Promptagator: Few-shot Dense Retrieval From 8 ExamplesZhuyun Dai, Vincent Y. Zhao, Ji Ma, Yi Luan et al.ICLR 2023 · 46 citations
- Augmenting Zero-Shot Dense Retrievers with Plug-in Mixture-of-MemoriesSuyu Ge, Chenyan Xiong, Corby Rosset, Arnold Overwijk et al.EMNLP 2023 · 4 citations
Builds on4
- 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
- Pre-training Tasks for Embedding-based Large-scale RetrievalWei-Cheng Chang, Felix X. Yu, Yin-Wen Chang, Yiming Yang et al.ICLR 2020 · 325 citations
- Dense Passage Retrieval for Open-Domain Question AnsweringVladimir Karpukhin, Barlas Oguz, Sewon Min, Patrick Lewis et al.EMNLP 2020 · 142 citations
Related papers
- Making Retrieval-Augmented Language Models Robust to Irrelevant ContextOri Yoran, Tomer Wolfson, Ori Ram, Jonathan BerantICLR 2024 · 361 citations
- Improving Passage Retrieval with Zero-Shot Question GenerationDevendra Singh Sachan, Mike Lewis, Mandar Joshi, Armen Aghajanyan et al.EMNLP 2022 · 69 citations
- Sentence-aware Contrastive Learning for Open-Domain Passage RetrievalWu Hong, Zhuosheng Zhang, Jinyuan Wang, Hai ZhaoACL 2022
- How Does Generative Retrieval Scale to Millions of Passages?Ronak Pradeep, Kai Hui, Jai Gupta, Ádám D. Lelkes et al.EMNLP 2023 · 23 citations
- Mitigating the Impact of False Negative in Dense Retrieval with Contrastive Confidence RegularizationShiqi Wang, Yeqin Zhang, Cam-Tu NguyenAAAI 2024 · 6 citations
