Challenging Decoder helps in Masked Auto-Encoder Pre-training for Dense Passage Retrieval
This work addresses the challenge of enhancing retrieval performance for information retrieval systems, representing an incremental improvement over existing methods.
The paper tackles the problem of improving dense passage retrieval by proposing a token importance aware masking strategy for masked auto-encoder pre-training, which enhances encoder representation ability and shows effectiveness on large-scale supervised and zero-shot retrieval benchmarks.
Recently, various studies have been directed towards exploring dense passage retrieval techniques employing pre-trained language models, among which the masked auto-encoder (MAE) pre-training architecture has emerged as the most promising. The conventional MAE framework relies on leveraging the passage reconstruction of decoder to bolster the text representation ability of encoder, thereby enhancing the performance of resulting dense retrieval systems. Within the context of building the representation ability of the encoder through passage reconstruction of decoder, it is reasonable to postulate that a ``more demanding'' decoder will necessitate a corresponding increase in the encoder's ability. To this end, we propose a novel token importance aware masking strategy based on pointwise mutual information to intensify the challenge of the decoder. Importantly, our approach can be implemented in an unsupervised manner, without adding additional expenses to the pre-training phase. Our experiments verify that the proposed method is both effective and robust on large-scale supervised passage retrieval datasets and out-of-domain zero-shot retrieval benchmarks.