A two-phase fine-tuning pipeline (global contrastive pretraining, then domain-specific hard-negative training) improves LLaMA-based dense retrieval on a Gemini-generated Japanese legal dataset and on a subset of MS MARCO.
PAIR: Leveraging Passage-Centric Similarity Relation for Improving Dense Passage Retrieval
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
Recently, dense passage retrieval has become a mainstream approach to finding relevant information in various natural language processing tasks. A number of studies have been devoted to improving the widely adopted dual-encoder architecture. However, most of the previous studies only consider query-centric similarity relation when learning the dual-encoder retriever. In order to capture more comprehensive similarity relations, we propose a novel approach that leverages both query-centric and PAssage-centric sImilarity Relations (called PAIR) for dense passage retrieval. To implement our approach, we make three major technical contributions by introducing formal formulations of the two kinds of similarity relations, generating high-quality pseudo labeled data via knowledge distillation, and designing an effective two-stage training procedure that incorporates passage-centric similarity relation constraint. Extensive experiments show that our approach significantly outperforms previous state-of-the-art models on both MSMARCO and Natural Questions datasets.
fields
cs.IR 1years
2024 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
Adaptive Two-Phase Finetuning LLMs for Japanese Legal Text Retrieval
A two-phase fine-tuning pipeline (global contrastive pretraining, then domain-specific hard-negative training) improves LLaMA-based dense retrieval on a Gemini-generated Japanese legal dataset and on a subset of MS MARCO.