sinequa/passage-ranker.pistachio

Model Card for passage-ranker.pistachio

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

Supported Languages

The model was trained and tested in the following languages:

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

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

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

Inference Times

GPU Memory usage

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

• —Minimal Sinequa version: 11.10.0
• —CUDA compute capability: above 7.5

Model Details

Overview

• —Number of parameters: 167 million
• —Base language model: Multilingual BERT-Base
• —Insensitive to casing and accents
• —Training procedure: MonoBERT

Training Data

• —MS MARCO Passage Ranking (Paper, Official Page, English & translated datasets on the HF dataset hub, translated dataset in Polish on the HF dataset hub)
• —Original English dataset
• —Translated datasets for the other nine supported languages

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.

Polish

This model has polish capacities, that are being evaluated over a subset of the PIRBenchmark with BM25 as the first stage retrieval.

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.