Improved Learned Sparse Retrieval with Corpus-Specific Vocabularies
Preprint
Abstract
We explore leveraging corpus-specific vocabularies that improve both efficiency and effectiveness of learned sparse retrieval systems. We find that pre-training the underlying BERT model on the target corpus, specifically targeting different vocabulary sizes incorporated into the document expansion process, improves retrieval quality by up to 12% while in some scenarios decreasing latency by up to 50%. Our experiments show that adopting corpus-specific vocabulary and increasing vocabulary size decreases average postings list length which in turn reduces latency. Ablation studies show interesting interactions between custom vocabularies, document expansion techniques, and sparsification objectives of sparse models. Both effectiveness and efficiency improvements transfer to different retrieval approaches such as uniCOIL and SPLADE and offer a simple yet effective approach to providing new efficiency-effectiveness trade-offs for learned sparse retrieval systems.Keywords
All Related Versions
This publication has 16 references indexed in Scilit:
- Pretraining without wordpieces: learning over a vocabulary of millions of wordsInternational Journal of Machine Learning and Cybernetics, 2024
- An Experimental Study on Pretraining Transformers from Scratch for IR2023
- Shallow pooling for sparse labelsInformation Retrieval Journal, 2022
- COIL: Revisit Exact Lexical Match in Information Retrieval with Contextualized Inverted List2021
- Techniques for Inverted Index CompressionACM Computing Surveys, 2020
- Modeling and solving term mismatch for full-text retrievalACM SIGIR Forum, 2012
- The Probabilistic Relevance Framework: BM25 and BeyondFoundations and Trends® in Information Retrieval, 2009
- Inverted files for text search enginesACM Computing Surveys, 2006
- Query evaluation: Strategies and optimizationsInformation Processing & Management, 1995
- On Information and SufficiencyThe Annals of Mathematical Statistics, 1951