CoolFace
Modelpublic

sinequa/vectorizer-v1-S-multilingual

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes246downloads
Model Card

Model Card for vectorizer-v1-S-multilingual

This model is a vectorizer developed by Sinequa. It produces an embedding vector given a passage or a query. The passage vectors are stored in our vector index and the query vector is used at query time to look up relevant passages in the index.

Model name: vectorizer-v1-S-multilingual

Supported Languages

The model was trained and tested in the following languages:

  • English
  • French
  • German
  • Spanish

Scores

MetricValue
Relevance (Recall@100)0.448

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 A10FP161 ms5 ms
NVIDIA A10FP323 ms14 ms
NVIDIA T4FP161 ms12 ms
NVIDIA T4FP322 ms52 ms
NVIDIA L4FP161 ms5 ms
NVIDIA L4FP322 ms18 ms

Gpu Memory usage

Quantization typeMemory
FP16300 MiB
FP32600 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

  • Number of parameters: 39 million
  • Base language model: Homegrown Sinequa BERT-Small (Paper) pretrained in the four supported languages
  • Insensitive to casing and accents
  • Training procedure: Query-passage pairs using in-batch negatives

Training Data

  • Natural Questions (Paper, Official Page)
  • Original English dataset
  • Translated datasets for the other three supported languages

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.

DatasetRecall@100
Average0.448
Arguana0.835
CLIMATE-FEVER0.350
DBPedia Entity0.287
FEVER0.645
FiQA-20180.305
HotpotQA0.396
MS MARCO0.533
NFCorpus0.162
NQ0.701
Quora0.947
SCIDOCS0.194
SciFact0.580
TREC-COVID0.051
Webis-Touche-20200.289

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.

LanguageRecall@100
French0.583
German0.524
Spanish0.483