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

Less is More: Pre-train a Strong Text Encoder for Dense Retrieval Using a Weak Decoder

About

Dense retrieval requires high-quality text sequence embeddings to support effective search in the representation space. Autoencoder-based language models are appealing in dense retrieval as they train the encoder to output high-quality embedding that can reconstruct the input texts. However, in this paper, we provide theoretical analyses and show empirically that an autoencoder language model with a low reconstruction loss may not provide good sequence representations because the decoder may take shortcuts by exploiting language patterns. To address this, we propose a new self-learning method that pre-trains the autoencoder using a \textit{weak} decoder, with restricted capacity and attention flexibility to push the encoder to provide better text representations. Our experiments on web search, news recommendation, and open domain question answering show that our pre-trained model significantly boosts the effectiveness and few-shot ability of dense retrieval models. Our code is available at https://github.com/microsoft/SEED-Encoder/.

Shuqi Lu, Di He, Chenyan Xiong, Guolin Ke, Waleed Malik, Zhicheng Dou, Paul Bennett, Tieyan Liu, Arnold Overwijk• 2021

Related benchmarks

TaskDatasetResultRank
Passage retrievalMsMARCO (dev)
MRR@1033.9
116
RetrievalMS MARCO (dev)
MRR@100.3264
84
Information RetrievalBEIR v1.0.0 (test)
ArguAna38.9
55
Information RetrievalMS MARCO DL2019
nDCG@1060.5
26
Document RetrievalMS MARCO Document (dev)
MRR@1000.396
24
Statute RetrievalSTARD (test)
Recall@1035.55
15
Dense RetrievalBEIR zero-shot
TREC-COVID62.7
13
Dense RetrievalNatural Question (test)
Recall@1074.54
9
Information RetrievalNatural Question
Recall@1078.03
9
Showing 9 of 9 rows

Other info

Follow for update