CoolFace
Modelpublic

sinequa/passage-ranker.mango

sourceHugging Faceupdated 8mo agoView on Hugging Face
1likes301downloads
Model Card

Model Card for passage-ranker.mango

This model is a passage ranker developed by Sinequa. It produces a relevance score given a query-passage pair and is used to order search results.

Model name: passage-ranker.mango

Supported Languages

The model was trained and tested in the following languages:

  • Chinese (simplified)
  • Dutch
  • English
  • French
  • German
  • Italian
  • Japanese
  • Portuguese
  • Spanish

Besides the aforementioned languages, basic support can be expected for additional 93 languages that were used during the pretraining of the base model (see list of languages).

Scores

MetricValue
Relevance (NDCG@10)0.480

Note that the relevance score is computed as an average over 14 retrieval datasets (see details below).

Inference Times

GPUQuantization typeBatch size 1Batch size 32
NVIDIA A10FP162 ms28 ms
NVIDIA A10FP324 ms82 ms
NVIDIA T4FP163 ms65 ms
NVIDIA T4FP3214 ms369 ms
NVIDIA L4FP163 ms38 ms
NVIDIA L4FP325 ms123 ms

GPU Memory usage

Quantization typeMemory
FP16850 MiB
FP321200 MiB

Note that GPU memory usage only includes how much GPU memory the actual model consumes on an NVIDIA T4 GPU with a batch size of 32. It does not include the fix amount of memory that is consumed by the ONNX Runtime upon initialization which can be around 0.5 to 1 GiB depending on the used GPU.

Requirements

Model Details

Overview

Training Data

Evaluation Metrics

To determine the relevance score, we averaged the results that we obtained when evaluating on the datasets of the BEIR benchmark. Note that all these datasets are in English.

DatasetNDCG@10
Average0.480
Arguana0.537
CLIMATE-FEVER0.241
DBPedia Entity0.371
FEVER0.777
FiQA-20180.327
HotpotQA0.696
MS MARCO0.414
NFCorpus0.332
NQ0.484
Quora0.768
SCIDOCS0.143
SciFact0.648
TREC-COVID0.673
Webis-Touche-20200.310

We evaluated the model on the datasets of the MIRACL benchmark to test its multilingual capacities. Note that not all training languages are part of the benchmark, so we only report the metrics for the existing languages.

LanguageNDCG@10
Chinese (simplified)0.463
French0.447
German0.415
Japanese0.526
Spanish0.485