Downloads · 30 days
305
3% of all-time downloads
sinequa/passage-ranker.strawberry
passage-ranker.strawberry is a text classification model from sinequa. Use it when you need a label for a piece of text. It is set up for transformers.
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.
Downloads · 30 days
305
3% of all-time downloads
All-time downloads
10.2K
Public
Repo size
873 MB
Likes
0
Public
Click a slice to open those files.
.bin428 MB · 96%
From the Hugging Face model README
passage-ranker.strawberryThis 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.strawberry
The model was trained and tested in the following languages:
Besides the aforementioned languages, basic support can be expected for additional 91 languages that were used during the pretraining of the base model (see Appendix A of XLM-R paper).
| Metric | Value |
|---|---|
| Relevance (NDCG@10) | 0.451 |
Note that the relevance score is computed as an average over 14 retrieval datasets (see details below).
| GPU | Quantization type | Batch size 1 | Batch size 32 |
|---|---|---|---|
| NVIDIA A10 | FP16 | 1 ms | 5 ms |
| NVIDIA A10 | FP32 | 2 ms | 22 ms |
| NVIDIA T4 | FP16 | 1 ms | 13 ms |
| NVIDIA T4 | FP32 | 3 ms | 64 ms |
| NVIDIA L4 | FP16 | 2 ms | 6 ms |
| NVIDIA L4 | FP32 | 2 ms | 30 ms |
| Quantization type | Memory |
|---|---|
| FP16 | 550 MiB |
| FP32 | 1100 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.
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.
| Dataset | NDCG@10 |
|---|---|
| Average | 0.451 |
| Arguana | 0.527 |
| CLIMATE-FEVER | 0.167 |
| DBPedia Entity | 0.343 |
| FEVER | 0.698 |
| FiQA-2018 | 0.297 |
| HotpotQA | 0.648 |
| MS MARCO | 0.409 |
| NFCorpus | 0.317 |
| NQ | 0.430 |
| Quora | 0.761 |
| SCIDOCS | 0.135 |
| SciFact | 0.597 |
| TREC-COVID | 0.670 |
| Webis-Touche-2020 | 0.311 |
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.
| Language | NDCG@10 |
|---|---|
| Chinese (simplified) | 0.414 |
| French | 0.382 |
| German | 0.320 |
| Japanese | 0.479 |
| Spanish | 0.418 |