Edit model card

Model Card for Model ID

This model is finetuned starting from the well-known ms-marco-MiniLM-L-6-v2 using KL distillation techniques as described here, using bge-reranker-v2-m3 as teacher

Usage

Usage with Transformers

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
model = AutoModelForSequenceClassification.from_pretrained("juanluisdb/MiniLM-L-6-rerank-m3")
tokenizer = AutoTokenizer.from_pretrained("juanluisdb/MiniLM-L-6-rerank-m3")
features = tokenizer(['How many people live in Berlin?', 'How many people live in Berlin?'], ['Berlin has a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.', 'New York City is famous for the Metropolitan Museum of Art.'],  padding=True, truncation=True, return_tensors="pt")
model.eval()
with torch.no_grad():
    scores = model(**features).logits
    print(scores)

Usage with SentenceTransformers

from sentence_transformers import CrossEncoder
model = CrossEncoder("juanluisdb/MiniLM-L-6-rerank-m3", max_length=512)
scores = model.predict([('Query', 'Paragraph1'), ('Query', 'Paragraph2') , ('Query', 'Paragraph3')])

Evaluation

BEIR (NDCG@10)

I've run tests on different BEIR datasets. Cross Encoders rerank top100 BM25 results.

bm25 jina-reranker-v1-turbo-en bge-reranker-v2-m3 mxbai-rerank-base-v1 ms-marco-MiniLM-L-6-v2 MiniLM-L-6-rerank-m3
nq* 0.305 0.533 0.597 0.535 0.523 0.580
fever* 0.638 0.852 0.857 0.767 0.801 0.867
fiqa 0.238 0.336 0.397 0.382 0.349 0.364
trec-covid 0.589 0.774 0.784 0.830 0.741 0.738
scidocs 0.15 0.166 0.169 0.171 0.164 0.165
scifact 0.676 0.739 0.731 0.719 0.688 0.750
nfcorpus 0.318 0.353 0.336 0.353 0.349 0.350
hotpotqa 0.629 0.745 0.794 0.668 0.724 0.775
dbpedia-entity 0.319 0.421 0.445 0.416 0.445 0.444
quora 0.787 0.858 0.858 0.747 0.825 0.871
climate-fever 0.163 0.233 0.314 0.253 0.244 0.309

* Training splits of NQ and Fever were used as part of the training data.

Comparison with ablated model trained only on MSMarco:

ms-marco-MiniLM-L-6-v2 MiniLM-L-6-rerank-m3-ablated
nq 0.5234 0.5412
fever 0.8007 0.8221
fiqa 0.349 0.3598
trec-covid 0.741 0.7331
scidocs 0.1638 0.163
scifact 0.688 0.7376
nfcorpus 0.3493 0.3495
hotpotqa 0.7235 0.7583
dbpedia-entity 0.4445 0.4382
quora 0.8251 0.8619
climate-fever 0.2438 0.2449

Datasets Used

~900k queries with 32-way triplets were used from these datasets:

  • MSMarco
  • TriviaQA
  • Natural Questions
  • FEVER
Downloads last month
77
Safetensors
Model size
22.7M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for juanluisdb/MiniLM-L-6-rerank-m3

Finetuned
(6)
this model

Datasets used to train juanluisdb/MiniLM-L-6-rerank-m3