SPLADE_max: SPLADEv2 with max aggregation
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_max")
# Use this if you want to actually use the model
model = AutoModel.load_from_hf_hub("xpmir/SPLADE_max", 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.4556 | 0.6802 | 0.9729 | 0.9729 | 0.7057 | 0.6944 | 0.6620 |
| msmarco_dev | 0.3522 | 0.0371 | 0.3564 | 0.3449 | 0.4750 | 0.4065 | 0.4318 |