Total runs:
38.6K
Run Growth:
22.8K
Growth Rate:
58.98%
Sponsored by
APIMart
cmarkea / bloomz-560m-retriever
Model's Last Updated: December 16 2024
feature-extraction
Introduction of bloomz-560m-retriever
Model Details of bloomz-560m-retriever
Note
We now strongly recommend using the Bloomz-560m-retriever-v2 model, which offers significantly superior performance .
Bloomz-560m-retriever
We introduce the Bloomz-560m-retriever based on the Bloomz-560m-sft-chat model. This model enables the creation of an embedding representation of text and queries for a retrieval task, linking queries to documents. The model is designed to be cross-language, meaning it is language-agnostic (English/French). This model is ideal for Open Domain Question Answering (ODQA), projecting queries and text with an algebraic structure to bring them closer together.
Training
It is a bi-encoder trained on a corpus of context/query pairs, with 50% in English and 50% in French. The language distribution for queries and contexts is evenly split (1/4 French-French, 1/4 French-English, 1/4 English-French, 1/4 English-English). The learning objective is to bring the embedding representation of queries and associated contexts closer using a contrastive method. The loss function is defined in Deep Metric Learning using Triplet Network .
Benchmark
Based on the SQuAD evaluation dataset (comprising 6000 queries distributed over 1200 contexts grouped into 35 themes), we compare the performance in terms of the average top contexter value for a query (Top-mean), the standard deviation of the average top (Top-std), and the percentage of correct queries within the top-1, top-5, and top-10. We compare the model with a TF-IDF trained on the SQuAD train sub-dataset (we want a fixed algebraic structure for the vector database instead of a variable structure every time we add a new document, then the IDF part has frozen), CamemBERT, Sentence-BERT, and finally our model. We observe these performances in both monolingual and cross-language contexts (query in French and context in English).
| Model (FR/FR) | Top-mean | Top-std | Top-1 (%) | Top-5 (%) | Top-10 (%) |
|---|---|---|---|---|---|
| TF-IDF | 128 | 269 | 23 | 46 | 56 |
| CamemBERT | 417 | 347 | 1 | 2 | 3 |
| Sentence-BERT | 11 | 41 | 43 | 71 | 82 |
| Bloomz-560m-retriever | 10 | 47 | 51 | 78 | 86 |
| Bloomz-3b-retriever | 9 | 37 | 50 | 79 | 87 |
| Model (EN/FR) | Top-mean | Top-std | Top-1 (%) | Top-5 (%) | Top-10 (%) |
|---|---|---|---|---|---|
| TF-IDF | 607 | 334 | 0 | 0 | 0 |
| CamemBERT | 432 | 345 | 0 | 1 | 1 |
| Sentence-BERT | 12 | 47 | 44 | 73 | 83 |
| Bloomz-560m-retriever | 10 | 44 | 49 | 77 | 86 |
| Bloomz-3b-retriever | 9 | 38 | 50 | 78 | 87 |
We observed that TF-IDF loses robustness in cross-language scenarios (even showing lower performance than CamemBERT, which is a model specialized in French). This can be explained by the fact that a Bag-Of-Words method cannot support this type of issue because, for a given sentence between two languages, the embedding vectors will be significantly different.
CamemBERT exhibits poor performance, not because it poorly groups contexts and queries by themes, but because a meta-cluster appears, separating contexts and queries (as illustrated in the image below), making this type of modeling inappropriate in a retriever context.
How to Use Bloomz-560m-retriever
The following example utilizes the API Pipeline of the Transformers library.
import numpy as np
from transformers import pipeline
from scipy.spatial.distance import cdist
retriever = pipeline('feature-extraction', 'cmarkea/bloomz-560m-retriever')
# Inportant: take only last token!
infer = lambda x: [ii[0][-1] for ii in retriever(x)]
list_of_contexts = [...]
emb_contexts = np.concatenate(infer(list_of_contexts), axis=0)
list_of_queries = [...]
emb_queries = np.concatenate(infer(list_of_queries), axis=0)
# Important: take l2 distance!
dist = cdist(emb_queries, emb_contexts, 'euclidean')
top_k = lambda x: [
[list_of_contexts[qq] for qq in ii]
for ii in dist.argsort(axis=-1)[:,:x]
]
# top 5 nearest contexts for each queries
top_contexts = top_k(5)
Citation
@online{DeBloomzRet,
AUTHOR = {Cyrile Delestre},
ORGANIZATION = {Cr{\'e}dit Mutuel Ark{\'e}a},
URL = {https://huggingface.co/cmarkea/bloomz-560m-retriever},
YEAR = {2023},
KEYWORDS = {NLP ; Transformers ; LLM ; Bloomz},
}
Runs of cmarkea bloomz-560m-retriever on huggingface.co
22
Total runs
0
24-hour runs
0
3-day runs
-2
7-day runs
7
30-day runs
More Information About bloomz-560m-retriever huggingface.co Model
Url of bloomz-560m-retriever
bloomz-560m-retriever huggingface.co Url
Provider of bloomz-560m-retriever huggingface.co
Other API from cmarkea
Updated:October 26 2024
Total runs:
8.6K
Run Growth:
3.1K
Growth Rate:
36.21%
Total runs:
4.2K
Run Growth:
-1.6K
Growth Rate:
-38.81%
Total runs:
1.7K
Run Growth:
896
Growth Rate:
53.85%
Total runs:
459
Run Growth:
-114
Growth Rate:
-24.84%
Total runs:
173
Run Growth:
72
Growth Rate:
41.62%
Total runs:
159
Run Growth:
63
Growth Rate:
39.62%
Total runs:
144
Run Growth:
40
Growth Rate:
27.78%
Total runs:
69
Run Growth:
-50
Growth Rate:
-72.46%
Total runs:
52
Run Growth:
13
Growth Rate:
25.00%
Total runs:
40
Run Growth:
-184
Growth Rate:
-460.00%
Total runs:
33
Run Growth:
5
Growth Rate:
15.15%
Total runs:
29
Run Growth:
0
Growth Rate:
0.00%
Total runs:
26
Run Growth:
-1
Growth Rate:
-3.85%
Total runs:
25
Run Growth:
11
Growth Rate:
44.00%
Total runs:
24
Run Growth:
-45
Growth Rate:
-187.50%
Total runs:
23
Run Growth:
-1
Growth Rate:
-4.35%
Total runs:
22
Run Growth:
-25
Growth Rate:
-113.64%
Total runs:
22
Run Growth:
2
Growth Rate:
9.09%
Total runs:
20
Run Growth:
-6
Growth Rate:
-30.00%
Total runs:
18
Run Growth:
6
Growth Rate:
33.33%
Total runs:
17
Run Growth:
-2
Growth Rate:
-11.76%
Total runs:
15
Run Growth:
-1
Growth Rate:
-6.67%
Total runs:
13
Run Growth:
-3
Growth Rate:
-23.08%
Total runs:
12
Run Growth:
4
Growth Rate:
33.33%
Total runs:
11
Run Growth:
-10
Growth Rate:
-90.91%
Total runs:
9
Run Growth:
-1
Growth Rate:
-11.11%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%