nickprock commited on
Commit
edbe5d2
·
verified ·
1 Parent(s): 8cf8580

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +164 -112
README.md CHANGED
@@ -1,91 +1,126 @@
1
- ---
2
- tags:
3
- - sentence-transformers
4
- - sentence-similarity
5
- - feature-extraction
6
- - dense
7
- - generated_from_trainer
8
- - dataset_size:45181
9
- - loss:MatryoshkaLoss
10
- - loss:CachedMultipleNegativesRankingLoss
11
- - loss:MultipleNegativesRankingLoss
12
- base_model: ibm-granite/granite-embedding-311m-multilingual-r2
13
- widget:
14
- - source_sentence: In quali anni ha partecipato Carriera ai Campionati europei?
15
- sentences:
16
- - Mokgomane è un villaggio del Botswana situato nel distretto Meridionale, sottodistretto
17
- di Barolong. Il villaggio, secondo il censimento del 2011, conta 708 abitanti.
18
- - '[''Carriera\nCon la ha disputato i Campionati europei del 2001 e i Campionati
19
- mondiali del 2002.\n\nCon la ha disputato due edizioni dei Campionati europei
20
- (2003, 2005).'', ''Carriera\nCon la ha disputato i Campionati mondiali del 2002
21
- e tre edizioni dei Campionati europei (1997, 1999, 2001).\n\nCon la ha disputato
22
- i Campionati europei del 2003.'']'
23
- - Carriera Con la ha disputato i Giochi olimpici di Montréal 1976, i Campionati
24
- mondiali del 1975 e tre edizioni dei Campionati europei (1974, 1976, 1978).
25
- - source_sentence: Quali sono le caratteristiche geografiche e demografiche della
26
- prefettura di Avé in Togo?
27
- sentences:
28
- - La prefettura di Avé è una prefettura del Togo situato nella regione Marittima
29
- con 97.830 abitanti al censimento 2010. Il capoluogo è la città di Kévé.
30
- - '[''La prefettura di Haho è una prefettura del Togo situato nella regione degli
31
- Altopiani con 247.817 abitanti al censimento 2010. Il capoluogo è la città di
32
- Notsé.'', ''La prefettura di Lacs è una prefettura del Togo situato nella regione
33
- Marittima con 172.148 abitanti al censimento 2010. Il capoluogo è la città di
34
- Aneho.'']'
35
- - Mmathethe è un villaggio del Botswana situato nel distretto Meridionale, sottodistretto
36
- di Barolong. Il villaggio, secondo il censimento del 2011, conta 5.078 abitanti.
37
- - source_sentence: Quali sono i nomi alternativi per il comune Těrlicko?
38
- sentences:
39
- - Malotwana Siding è un villaggio del Botswana situato nel distretto di Kgatleng,
40
- sottodistretto di Kgatleng. Il villaggio, secondo il censimento del 2011, conta
41
- 608 abitanti.
42
- - '[''Il toponimo Rohoznice può riferirsi a diverse località della Repubblica Ceca:\n\nRohoznice,
43
- comune del distretto di Jičín\nRohoznice, comune del distretto di Pardubice'',
44
- "Biskupice (in tedesco Biskupitz) è un comune della Repubblica Ceca facente parte
45
- del distretto di Zlín, nella regione di Zlín.\n\nQua è nato l''ex calciatore ed
46
- allenatore Milan Máčala."]'
47
- - Těrlicko (in polacco Cierlicko, in tedesco Tierlitzko) è un comune della Repubblica
48
- Ceca facente parte del distretto di Karviná, nella regione della Moravia-Slesia.
49
- - source_sentence: Quante persone habita nel villaggio Tswaaneng al censimento del
50
- 2011?
51
- sentences:
52
- - Mogoditshane è un villaggio del Botswana situato nel distretto di Kweneng, sottodistretto
53
- di Kweneng East. Il villaggio, secondo il censimento del 2011, conta 58.079 abitanti.
54
- - '[''Tswaane è un villaggio del Botswana situato nel distretto di Kweneng, sottodistretto
55
- di Kweneng West. Il villaggio, secondo il censimento del 2011, conta 603 abitanti.'',
56
- ''Tshwaane è un villaggio del Botswana situato nel distretto Meridionale, sottodistretto
57
- di Ngwaketse. Il villaggio, secondo il censimento del 2011, conta 299 abitanti.'']'
58
- - Tswaaneng è un villaggio del Botswana situato nel distretto Meridionale, sottodistretto
59
- di Barolong. Il villaggio, secondo il censimento del 2011, conta 679 abitanti.
60
- - source_sentence: In quali gare fu eliminato l'atleta nelle Olimpiadi di Parigi?
61
- sentences:
62
- - Partecipò ai Giochi della II Olimpiade di Parigi nella gara di spada individuale
63
- e sciabola individuale. In entrambe le gare fu eliminato alle semifinali.
64
- - Partecipò ai Giochi della II Olimpiade di Parigi nelle gare di fioretto per maestri,
65
- dove giunse al nono posto, e di spada per maestri, dove fu eliminato in semifinale.
66
- - '[''Partecipò ai Giochi della II Olimpiade di Parigi alla gara di fioretto individuale,
67
- in cui fu eliminato ai quarti, e alla gara di spada individuale, in cui fu eliminato
68
- al primo turno.'', ''Partecipò ai Giochi della II Olimpiade di Parigi nelle gare
69
- di fioretto per maestri, in cui fu eliminato ai ripescaggi, e di spada per maestri,
70
- in cui fu eliminato al primo turno.'']'
71
- pipeline_tag: sentence-similarity
72
- library_name: sentence-transformers
73
- metrics:
74
- - cosine_accuracy
75
- model-index:
76
- - name: SentenceTransformer based on ibm-granite/granite-embedding-311m-multilingual-r2
77
- results:
78
- - task:
79
- type: triplet
80
- name: Triplet
81
- dataset:
82
- name: triplet eval it
83
- type: triplet-eval-it
84
- metrics:
85
- - type: cosine_accuracy
86
- value: 0.9995794892311096
87
- name: Cosine Accuracy
88
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
89
 
90
  # SentenceTransformer based on ibm-granite/granite-embedding-311m-multilingual-r2
91
 
@@ -176,34 +211,51 @@ You can finetune this model on your own dataset.
176
 
177
  *List how the model may foreseeably be misused and address what users ought not to do with the model.*
178
  -->
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
179
 
180
- ## Evaluation
181
-
182
- ### Metrics
183
-
184
- #### Triplet
185
-
186
- * Dataset: `triplet-eval-it`
187
- * Evaluated with [<code>TripletEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.sentence_transformer.evaluation.TripletEvaluator)
188
-
189
- | Metric | Value |
190
- |:--------------------|:-----------|
191
- | **cosine_accuracy** | **0.9996** |
192
-
193
- <!--
194
- ## Bias, Risks and Limitations
195
-
196
- *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
197
- -->
198
-
199
- <!--
200
- ### Recommendations
201
-
202
- *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
203
- -->
204
 
205
  ## Training Details
206
 
 
 
 
 
 
 
 
 
207
  ### Training Dataset
208
 
209
  #### it-wiki-retrieval-synthetic-hn
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - dense
7
+ - generated_from_trainer
8
+ - dataset_size:45181
9
+ - loss:MatryoshkaLoss
10
+ - loss:CachedMultipleNegativesRankingLoss
11
+ - loss:MultipleNegativesRankingLoss
12
+ base_model: ibm-granite/granite-embedding-311m-multilingual-r2
13
+ widget:
14
+ - source_sentence: In quali anni ha partecipato Carriera ai Campionati europei?
15
+ sentences:
16
+ - >-
17
+ Mokgomane è un villaggio del Botswana situato nel distretto Meridionale,
18
+ sottodistretto di Barolong. Il villaggio, secondo il censimento del 2011,
19
+ conta 708 abitanti.
20
+ - >-
21
+ ['Carriera\nCon la ha disputato i Campionati europei del 2001 e i Campionati
22
+ mondiali del 2002.\n\nCon la ha disputato due edizioni dei Campionati
23
+ europei (2003, 2005).', 'Carriera\nCon la ha disputato i Campionati mondiali
24
+ del 2002 e tre edizioni dei Campionati europei (1997, 1999, 2001).\n\nCon la
25
+ ha disputato i Campionati europei del 2003.']
26
+ - >-
27
+ Carriera Con la ha disputato i Giochi olimpici di Montréal 1976, i
28
+ Campionati mondiali del 1975 e tre edizioni dei Campionati europei (1974,
29
+ 1976, 1978).
30
+ - source_sentence: >-
31
+ Quali sono le caratteristiche geografiche e demografiche della prefettura di
32
+ Avé in Togo?
33
+ sentences:
34
+ - >-
35
+ La prefettura di Avé è una prefettura del Togo situato nella regione
36
+ Marittima con 97.830 abitanti al censimento 2010. Il capoluogo è la città di
37
+ Kévé.
38
+ - >-
39
+ ['La prefettura di Haho è una prefettura del Togo situato nella regione
40
+ degli Altopiani con 247.817 abitanti al censimento 2010. Il capoluogo è la
41
+ città di Notsé.', 'La prefettura di Lacs è una prefettura del Togo situato
42
+ nella regione Marittima con 172.148 abitanti al censimento 2010. Il
43
+ capoluogo è la città di Aneho.']
44
+ - >-
45
+ Mmathethe è un villaggio del Botswana situato nel distretto Meridionale,
46
+ sottodistretto di Barolong. Il villaggio, secondo il censimento del 2011,
47
+ conta 5.078 abitanti.
48
+ - source_sentence: Quali sono i nomi alternativi per il comune Těrlicko?
49
+ sentences:
50
+ - >-
51
+ Malotwana Siding è un villaggio del Botswana situato nel distretto di
52
+ Kgatleng, sottodistretto di Kgatleng. Il villaggio, secondo il censimento
53
+ del 2011, conta 608 abitanti.
54
+ - >-
55
+ ['Il toponimo Rohoznice può riferirsi a diverse località della Repubblica
56
+ Ceca:\n\nRohoznice, comune del distretto di Jičín\nRohoznice, comune del
57
+ distretto di Pardubice', "Biskupice (in tedesco Biskupitz) è un comune della
58
+ Repubblica Ceca facente parte del distretto di Zlín, nella regione di
59
+ Zlín.\n\nQua è nato l'ex calciatore ed allenatore Milan Máčala."]
60
+ - >-
61
+ Těrlicko (in polacco Cierlicko, in tedesco Tierlitzko) è un comune della
62
+ Repubblica Ceca facente parte del distretto di Karviná, nella regione della
63
+ Moravia-Slesia.
64
+ - source_sentence: Quante persone habita nel villaggio Tswaaneng al censimento del 2011?
65
+ sentences:
66
+ - >-
67
+ Mogoditshane è un villaggio del Botswana situato nel distretto di Kweneng,
68
+ sottodistretto di Kweneng East. Il villaggio, secondo il censimento del
69
+ 2011, conta 58.079 abitanti.
70
+ - >-
71
+ ['Tswaane è un villaggio del Botswana situato nel distretto di Kweneng,
72
+ sottodistretto di Kweneng West. Il villaggio, secondo il censimento del
73
+ 2011, conta 603 abitanti.', 'Tshwaane è un villaggio del Botswana situato
74
+ nel distretto Meridionale, sottodistretto di Ngwaketse. Il villaggio,
75
+ secondo il censimento del 2011, conta 299 abitanti.']
76
+ - >-
77
+ Tswaaneng è un villaggio del Botswana situato nel distretto Meridionale,
78
+ sottodistretto di Barolong. Il villaggio, secondo il censimento del 2011,
79
+ conta 679 abitanti.
80
+ - source_sentence: In quali gare fu eliminato l'atleta nelle Olimpiadi di Parigi?
81
+ sentences:
82
+ - >-
83
+ Partecipò ai Giochi della II Olimpiade di Parigi nella gara di spada
84
+ individuale e sciabola individuale. In entrambe le gare fu eliminato alle
85
+ semifinali.
86
+ - >-
87
+ Partecipò ai Giochi della II Olimpiade di Parigi nelle gare di fioretto per
88
+ maestri, dove giunse al nono posto, e di spada per maestri, dove fu
89
+ eliminato in semifinale.
90
+ - >-
91
+ ['Partecipò ai Giochi della II Olimpiade di Parigi alla gara di fioretto
92
+ individuale, in cui fu eliminato ai quarti, e alla gara di spada
93
+ individuale, in cui fu eliminato al primo turno.', 'Partecipò ai Giochi
94
+ della II Olimpiade di Parigi nelle gare di fioretto per maestri, in cui fu
95
+ eliminato ai ripescaggi, e di spada per maestri, in cui fu eliminato al
96
+ primo turno.']
97
+ pipeline_tag: sentence-similarity
98
+ library_name: sentence-transformers
99
+ metrics:
100
+ - cosine_accuracy
101
+ model-index:
102
+ - name: >-
103
+ SentenceTransformer based on
104
+ ibm-granite/granite-embedding-311m-multilingual-r2
105
+ results:
106
+ - task:
107
+ type: triplet
108
+ name: Triplet
109
+ dataset:
110
+ name: triplet eval it
111
+ type: triplet-eval-it
112
+ metrics:
113
+ - type: cosine_accuracy
114
+ value: 0.9995794892311096
115
+ name: Cosine Accuracy
116
+ license: apache-2.0
117
+ datasets:
118
+ - nickprock/it-wiki-retrieval-synthetic-hn
119
+ - mteb/stsb_multi_mt
120
+ - crux82/squad_it
121
+ language:
122
+ - it
123
+ ---
124
 
125
  # SentenceTransformer based on ibm-granite/granite-embedding-311m-multilingual-r2
126
 
 
211
 
212
  *List how the model may foreseeably be misused and address what users ought not to do with the model.*
213
  -->
214
+ ## Evaluation & Benchmarks
215
+
216
+ To ensure full transparency, the model was evaluated across two distinct benchmarks: **Information Retrieval / RAG** (SQuAD-it) and **Semantic Textual Similarity** (STSb-it), compared against widely used Italian embedding models.
217
+
218
+ ### 1. Information Retrieval & RAG Benchmark (SQuAD-it)
219
+ * **Dataset:** `crux82/squad_it` (Test Split - 1,000 queries matched against a unique passage corpus)
220
+ * **Objective:** Evaluate how effectively the model retrieves the exact paragraph containing the answer among 1,000 candidate passages.
221
+
222
+ | Modello | Dim. Vettore | Hit@1 / Precision (%) | MRR@10 | NDCG@10 | Latency (1k Queries) |
223
+ | :--- | :---: | :---: | :---: | :---: | :---: |
224
+ | `nickprock/sentence-bert-base-italian-uncased` | 768 | 36.40% | 0.4980 | 0.5723 | 2.55s |
225
+ | `nickprock/sentence-bert-base-italian-xxl-uncased` | 768 | 40.60% | 0.5347 | 0.6026 | 2.17s |
226
+ | `nickprock/multi-sentence-BERTino` | 768 | 39.30% | 0.4933 | 0.5465 | 1.11s |
227
+ | `DeepMount00/Ita-Search` | 1024 | 43.50% | 0.5465 | 0.6087 | 17.51s |
228
+ | **`granite-311m-italiano-matryoshka (Full)`** | **768** | **65.40%** | **0.7560** | **0.8009** | **1.71s** |
229
+ | **`granite-311m-italiano-matryoshka (Truncated)`** | **256** | **65.00%** | **0.7500** | **0.7957** | **1.61s** |
230
+
231
+ ---
232
+
233
+ ### 2. Semantic Textual Similarity Benchmark (STSb-it)
234
+ * **Dataset:** `stsb_multi_mt` (Italian Test Split - 1,379 sentence pairs with human scores from 0 to 5)
235
+ * **Objective:** Measure correlation between model cosine similarity and human judgment on sentence equivalence.
236
+
237
+ | Modello | Dim. Vettore | Spearman Cosine | Pearson Cosine | Throughput (Pairs/Sec) |
238
+ | :--- | :---: | :---: | :---: | :---: |
239
+ | `nickprock/sentence-bert-base-italian-uncased` | 768 | 0.8097 | 0.8098 | 596.9 |
240
+ | `nickprock/sentence-bert-base-italian-xxl-uncased` | 768 | 0.8205 | 0.8224 | 684.1 |
241
+ | `nickprock/multi-sentence-BERTino` | 768 | 0.8137 | 0.8163 | 1206.1 |
242
+ | `DeepMount00/Ita-Search` | 1024 | 0.7608 | 0.7702 | 120.2 |
243
+ | **`granite-311m-italiano-matryoshka (Full)`** | **768** | **0.7612** | **0.7574** | **588.2** |
244
+ | **`granite-311m-italiano-matryoshka (Truncated)`** | **256** | **0.7627** | **0.7599** | **572.0** |
245
+
246
+ ---
247
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
248
 
249
  ## Training Details
250
 
251
+ ### Training Setup
252
+ - **Dataset:** `nickprock/it-wiki-retrieval-synthetic-hn` (45,181 samples)
253
+ - **Epochs:** 1
254
+ - **Batch Size:** 16 (per device) x 16 (Gradient Accumulation) = 256 effective batch size
255
+ - **Optimizer:** `adamw_8bit`
256
+ - **Learning Rate:** 2e-5 with linear warmup (0.1)
257
+ - **Precision:** `bf16`
258
+
259
  ### Training Dataset
260
 
261
  #### it-wiki-retrieval-synthetic-hn