HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

No Parameter Left Behind: How Distillation and Model Size Affect Zero-Shot Retrieval

Guilherme Moraes Rosa Luiz Bonifacio Vitor Jeronymo Hugo Abonizio Marzieh Fadaee Roberto Lotufo Rodrigo Nogueira

No Parameter Left Behind: How Distillation and Model Size Affect Zero-Shot Retrieval

Abstract

Recent work has shown that small distilled language models are strong competitors to models that are orders of magnitude larger and slower in a wide range of information retrieval tasks. This has made distilled and dense models, due to latency constraints, the go-to choice for deployment in real-world retrieval applications. In this work, we question this practice by showing that the number of parameters and early query-document interaction play a significant role in the generalization ability of retrieval models. Our experiments show that increasing model size results in marginal gains on in-domain test sets, but much larger gains in new domains never seen during fine-tuning. Furthermore, we show that rerankers largely outperform dense ones of similar size in several tasks. Our largest reranker reaches the state of the art in 12 of the 18 datasets of the Benchmark-IR (BEIR) and surpasses the previous state of the art by 3 average points. Finally, we confirm that in-domain effectiveness is not a good indicator of zero-shot effectiveness. Code is available at https://github.com/guilhermemr04/scaling-zero-shot-retrieval.git

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
question-answering-on-fiqa-2018-beirmonoT5-3B
nDCG@10: 0.513
question-answering-on-hotpotqa-beirmonoT5-3B
nDCG@10: 0.759
question-answering-on-nq-beirmonoT5-3B
nDCG@10: 0.633

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
No Parameter Left Behind: How Distillation and Model Size Affect Zero-Shot Retrieval | Papers | HyperAI