SPLADE_doc: SPLADEv2 model with document encoder only
SPLADE v2: Sparse Lexical and Expansion Model for Information Retrieval (Thibault Formal, Carlos Lassance, Benjamin Piwowarski, Stéphane Clinchant). 2021. https://arxiv.org/abs/2109.10086
Using the model
The model can be loaded with experimaestro IR
from xpmir.models import AutoModel
# Model that can be re-used in experiments
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_doc")
# Use this if you want to actually use the model
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_doc", as_instance=True)
model.initialize() model.rsv("walgreens store sales average", "The average
Walgreens salary ranges...")
Results
| Dataset | AP | P@20 | RR | RR@10 | nDCG | nDCG@10 | nDCG@20 |
|---|---|---|---|---|---|---|---|
| trec2019 | 0.3875 | 0.6314 | 0.9355 | 0.9336 | 0.6332 | 0.6663 | 0.6277 |
| msmarco_dev | 0.3269 | 0.0351 | 0.3318 | 0.3204 | 0.4492 | 0.3797 | 0.4037 |