CoolFace
Modelpublic

sinequa/passage-ranker.apricot

sourceHugging Faceupdated 8mo agoView on Hugging Face
0likes19downloads
Model Card

Model Card for passage-ranker.apricot

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.apricot

Supported Languages

The model was trained and tested in the following languages:

  • —English
  • —French
  • —German
  • —Spanish
  • —Italian
  • —Dutch
  • —Japanese
  • —Portuguese
  • —Chinese (simplified)
  • —Polish
  • —Arabic
  • —Korean

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
English Relevance (NDCG@10)0.449
Arabic Relevance (NDCG@10)0.251
Korean Relevance (NDCG@10)0.234

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

Inference Times

GPUQuantization typeBatch size 1Batch size 32
NVIDIA A10FP161 ms5 ms
NVIDIA A10FP322 ms22 ms
NVIDIA T4FP161 ms13 ms
NVIDIA T4FP323 ms64 ms
NVIDIA L4FP162 ms6 ms
NVIDIA L4FP322 ms30 ms

GPU Memory usage

Quantization typeMemory
FP16550 MiB
FP321100 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

English

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.449
Arguana0.515
CLIMATE-FEVER0.170
DBPedia Entity0.332
FEVER0.723
FiQA-20180.291
HotpotQA0.660
MS MARCO0.400
NFCorpus0.279
NQ0.427
Quora0.755
SCIDOCS0.129
SciFact0.626
TREC-COVID0.687
Webis-Touche-20200.291
Arabic

This model has arabic capacities, that are being evaluated over a home made translation of Msmarco with BM25 as the first stage retrieval.

DatasetNDCG@10
msmarco-ar0.251
Korean

This model has korean capacities, that are being evaluated over a home made translation of Msmarco with BM25 as the first stage retrieval.

DatasetNDCG@10
msmarco-ko0.233
Other languages

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
French0.391
German0.338
Spanish0.424
Japanese0.489
Chinese (simplified)0.423