seregadgl commited on
Commit
64bf783
·
verified ·
1 Parent(s): 477cec3

Add new SentenceTransformer model

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
1_Pooling/config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 768,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false,
7
+ "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
+ }
2_SparseLayer/config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"dim": 768, "vocab_size": 262144, "k": 64}
2_SparseLayer/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac1c2461440ae590d870e31d3cad68b22661d441f692cc456960cd2200d5d949
3
+ size 806356949
README.md ADDED
@@ -0,0 +1,519 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sentence-similarity
5
+ - feature-extraction
6
+ - dense
7
+ - generated_from_trainer
8
+ - dataset_size:102127
9
+ - loss:SpladeLoss
10
+ - loss:SparseMultipleNegativesRankingLoss
11
+ - loss:FlopsLoss
12
+ base_model: seregadgl/splade_gemma_google_base_checkpoint_100_clear
13
+ widget:
14
+ - source_sentence: 'query: 6460338 acdelco'
15
+ sentences:
16
+ - 'document: очиститель тормозов rsqprofessional арт 072589767pl volkswagen id buzz
17
+ янтарный'
18
+ - 'document: гтц 6460338 для chevrolet traverse'
19
+ - 'document: гтц 6960358 для chevrolet traverse'
20
+ - source_sentence: 'query: audioquest cinnamon usb 0 7500 см '
21
+ sentences:
22
+ - 'document: кабель usb аудиоквест cinnamon 0 7500 см 8712516'
23
+ - 'document: задняя камера рамке номерного знака интерпауэр ip616 54785862'
24
+ - 'document: аудиокабель soundwave 200 см'
25
+ - source_sentence: 'query: акустическое пианино weber w 121 pw '
26
+ sentences:
27
+ - 'document: акустическое пианино steinway model s'
28
+ - 'document: инструмент для игры на пианино вебер w 121 pw'
29
+ - 'document: велосипед сильвербек strela sport 700c 54 см blue 60097000435025'
30
+ - source_sentence: 'query: шкаф шрм24'
31
+ sentences:
32
+ - 'document: wardrobe shrm 24 4348563'
33
+ - 'document: духовой шкаф бертаццони f6011provtn'
34
+ - 'document: шкаф мдф30'
35
+ - source_sentence: 'query: 1452634 santool jawa 300 cl'
36
+ sentences:
37
+ - 'document: смартфон эппл iphone xs max 512gb'
38
+ - 'document: 1453934 santool съемник для сальников jawa 300 cl'
39
+ - 'document: 1452634 santool съемник для сальников jawa 300 cl'
40
+ datasets:
41
+ - seregadgl/car_and_product_triplet_103k
42
+ pipeline_tag: sentence-similarity
43
+ library_name: sentence-transformers
44
+ metrics:
45
+ - cosine_accuracy@1
46
+ - cosine_precision@1
47
+ - cosine_precision@3
48
+ - cosine_precision@5
49
+ - cosine_precision@10
50
+ - cosine_recall@1
51
+ - cosine_recall@3
52
+ - cosine_recall@5
53
+ - cosine_recall@10
54
+ - cosine_ndcg@10
55
+ - cosine_mrr@10
56
+ - cosine_map@100
57
+ model-index:
58
+ - name: SentenceTransformer based on seregadgl/splade_gemma_google_base_checkpoint_100_clear
59
+ results:
60
+ - task:
61
+ type: information-retrieval
62
+ name: Information Retrieval
63
+ dataset:
64
+ name: val set fine
65
+ type: val_set_fine
66
+ metrics:
67
+ - type: cosine_accuracy@1
68
+ value: 0.749
69
+ name: Cosine Accuracy@1
70
+ - type: cosine_precision@1
71
+ value: 0.749
72
+ name: Cosine Precision@1
73
+ - type: cosine_precision@3
74
+ value: 0.27433333333333326
75
+ name: Cosine Precision@3
76
+ - type: cosine_precision@5
77
+ value: 0.17060000000000003
78
+ name: Cosine Precision@5
79
+ - type: cosine_precision@10
80
+ value: 0.0886
81
+ name: Cosine Precision@10
82
+ - type: cosine_recall@1
83
+ value: 0.749
84
+ name: Cosine Recall@1
85
+ - type: cosine_recall@3
86
+ value: 0.823
87
+ name: Cosine Recall@3
88
+ - type: cosine_recall@5
89
+ value: 0.853
90
+ name: Cosine Recall@5
91
+ - type: cosine_recall@10
92
+ value: 0.886
93
+ name: Cosine Recall@10
94
+ - type: cosine_ndcg@10
95
+ value: 0.816961550160875
96
+ name: Cosine Ndcg@10
97
+ - type: cosine_mrr@10
98
+ value: 0.7949674603174605
99
+ name: Cosine Mrr@10
100
+ - type: cosine_map@100
101
+ value: 0.7988750755190056
102
+ name: Cosine Map@100
103
+ ---
104
+
105
+ # SentenceTransformer based on seregadgl/splade_gemma_google_base_checkpoint_100_clear
106
+
107
+ This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [seregadgl/splade_gemma_google_base_checkpoint_100_clear](https://huggingface.co/seregadgl/splade_gemma_google_base_checkpoint_100_clear) on the [car_and_product_triplet_103k](https://huggingface.co/datasets/seregadgl/car_and_product_triplet_103k) dataset. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
108
+
109
+ ## Model Details
110
+
111
+ ### Model Description
112
+ - **Model Type:** Sentence Transformer
113
+ - **Base model:** [seregadgl/splade_gemma_google_base_checkpoint_100_clear](https://huggingface.co/seregadgl/splade_gemma_google_base_checkpoint_100_clear) <!-- at revision 20c38a098901bc44c1031a7537d0e3bf0aa93063 -->
114
+ - **Maximum Sequence Length:** 512 tokens
115
+ - **Output Dimensionality:** 768 dimensions
116
+ - **Similarity Function:** Cosine Similarity
117
+ - **Training Dataset:**
118
+ - [car_and_product_triplet_103k](https://huggingface.co/datasets/seregadgl/car_and_product_triplet_103k)
119
+ <!-- - **Language:** Unknown -->
120
+ <!-- - **License:** Unknown -->
121
+
122
+ ### Model Sources
123
+
124
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
125
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
126
+ - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
127
+
128
+ ### Full Model Architecture
129
+
130
+ ```
131
+ SentenceTransformer(
132
+ (0): Transformer({'max_seq_length': 512, 'do_lower_case': False, 'architecture': 'Gemma3TextModel'})
133
+ (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
134
+ (2): SparseLayer(
135
+ (linear): Linear(in_features=768, out_features=262144, bias=True)
136
+ )
137
+ )
138
+ ```
139
+
140
+ ## Usage
141
+
142
+ ### Direct Usage (Sentence Transformers)
143
+
144
+ First install the Sentence Transformers library:
145
+
146
+ ```bash
147
+ pip install -U sentence-transformers
148
+ ```
149
+
150
+ Then you can load this model and run inference.
151
+ ```python
152
+ from sentence_transformers import SentenceTransformer
153
+
154
+ # Download from the 🤗 Hub
155
+ model = SentenceTransformer("seregadgl/splade_gemma_google_base_checkpoint_100_ver2_checkpoint100")
156
+ # Run inference
157
+ sentences = [
158
+ 'query: 1452634 santool jawa 300 cl',
159
+ 'document: 1452634 santool съемник для сальников jawa 300 cl',
160
+ 'document: 1453934 santool съемник для сальников jawa 300 cl',
161
+ ]
162
+ embeddings = model.encode(sentences)
163
+ print(embeddings.shape)
164
+ # [3, 768]
165
+
166
+ # Get the similarity scores for the embeddings
167
+ similarities = model.similarity(embeddings, embeddings)
168
+ print(similarities)
169
+ # tensor([[1.0000, 0.1749, 0.1724],
170
+ # [0.1749, 1.0000, 0.7309],
171
+ # [0.1724, 0.7309, 1.0000]])
172
+ ```
173
+
174
+ <!--
175
+ ### Direct Usage (Transformers)
176
+
177
+ <details><summary>Click to see the direct usage in Transformers</summary>
178
+
179
+ </details>
180
+ -->
181
+
182
+ <!--
183
+ ### Downstream Usage (Sentence Transformers)
184
+
185
+ You can finetune this model on your own dataset.
186
+
187
+ <details><summary>Click to expand</summary>
188
+
189
+ </details>
190
+ -->
191
+
192
+ <!--
193
+ ### Out-of-Scope Use
194
+
195
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
196
+ -->
197
+
198
+ ## Evaluation
199
+
200
+ ### Metrics
201
+
202
+ #### Information Retrieval
203
+
204
+ * Dataset: `val_set_fine`
205
+ * Evaluated with [<code>InformationRetrievalEvaluator</code>](https://sbert.net/docs/package_reference/sentence_transformer/evaluation.html#sentence_transformers.evaluation.InformationRetrievalEvaluator)
206
+
207
+ | Metric | Value |
208
+ |:--------------------|:----------|
209
+ | cosine_accuracy@1 | 0.749 |
210
+ | cosine_precision@1 | 0.749 |
211
+ | cosine_precision@3 | 0.2743 |
212
+ | cosine_precision@5 | 0.1706 |
213
+ | cosine_precision@10 | 0.0886 |
214
+ | cosine_recall@1 | 0.749 |
215
+ | cosine_recall@3 | 0.823 |
216
+ | cosine_recall@5 | 0.853 |
217
+ | cosine_recall@10 | 0.886 |
218
+ | **cosine_ndcg@10** | **0.817** |
219
+ | cosine_mrr@10 | 0.795 |
220
+ | cosine_map@100 | 0.7989 |
221
+
222
+ <!--
223
+ ## Bias, Risks and Limitations
224
+
225
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
226
+ -->
227
+
228
+ <!--
229
+ ### Recommendations
230
+
231
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
232
+ -->
233
+
234
+ ## Training Details
235
+
236
+ ### Training Dataset
237
+
238
+ #### car_and_product_triplet_103k
239
+
240
+ * Dataset: [car_and_product_triplet_103k](https://huggingface.co/datasets/seregadgl/car_and_product_triplet_103k) at [3519181](https://huggingface.co/datasets/seregadgl/car_and_product_triplet_103k/tree/35191818e272dc373544bd86903a5146c6f993e2)
241
+ * Size: 102,127 training samples
242
+ * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
243
+ * Approximate statistics based on the first 1000 samples:
244
+ | | anchor | positive | negative |
245
+ |:--------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:---------------------------------------------------------------------------------|
246
+ | type | string | string | string |
247
+ | details | <ul><li>min: 5 tokens</li><li>mean: 16.27 tokens</li><li>max: 44 tokens</li></ul> | <ul><li>min: 4 tokens</li><li>mean: 23.62 tokens</li><li>max: 77 tokens</li></ul> | <ul><li>min: 6 tokens</li><li>mean: 23.2 tokens</li><li>max: 47 tokens</li></ul> |
248
+ * Samples:
249
+ | anchor | positive | negative |
250
+ |:--------------------------------------------------------------|:--------------------------------------------------------------------------------------|:--------------------------------------------------------------------------------|
251
+ | <code>query: погружной блендер tefal optichef hb64f810</code> | <code>document: погружной блендер тефаль optichef hb64f810</code> | <code>document: погружной миксер tefal mixchef hb64f850</code> |
252
+ | <code>query: 375675836 niteo</code> | <code>document: тосол 375675836 для ford f350 полуночный синий</code> | <code>document: тосол 375625836 для ford f350 полуночный синий фиалковый</code> |
253
+ | <code>query: накидка с подогревом dodge viper pink</code> | <code>document: накидка с подогревом acdelco арт 787327sx dodge viper розовый</code> | <code>document: 787327sx накидка с подогревом indian challenger лаймовый</code> |
254
+ * Loss: [<code>SpladeLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#spladeloss) with these parameters:
255
+ ```json
256
+ {
257
+ "loss": "SparseMultipleNegativesRankingLoss(scale=1.0, similarity_fct='dot_score', gather_across_devices=False)",
258
+ "document_regularizer_weight": 1e-05,
259
+ "query_regularizer_weight": 1e-05
260
+ }
261
+ ```
262
+
263
+ ### Evaluation Dataset
264
+
265
+ #### car_and_product_triplet_103k
266
+
267
+ * Dataset: [car_and_product_triplet_103k](https://huggingface.co/datasets/seregadgl/car_and_product_triplet_103k) at [3519181](https://huggingface.co/datasets/seregadgl/car_and_product_triplet_103k/tree/35191818e272dc373544bd86903a5146c6f993e2)
268
+ * Size: 1,000 evaluation samples
269
+ * Columns: <code>anchor</code>, <code>positive</code>, and <code>negative</code>
270
+ * Approximate statistics based on the first 1000 samples:
271
+ | | anchor | positive | negative |
272
+ |:--------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
273
+ | type | string | string | string |
274
+ | details | <ul><li>min: 5 tokens</li><li>mean: 16.73 tokens</li><li>max: 74 tokens</li></ul> | <ul><li>min: 5 tokens</li><li>mean: 23.54 tokens</li><li>max: 80 tokens</li></ul> | <ul><li>min: 5 tokens</li><li>mean: 22.66 tokens</li><li>max: 65 tokens</li></ul> |
275
+ * Samples:
276
+ | anchor | positive | negative |
277
+ |:------------------------------------------------------------------|:-------------------------------------------------------------------------------------------------------------|:----------------------------------------------------------------------------------|
278
+ | <code>query: зеркала для 'слепых' зон volkswagen arteon</code> | <code>document: зеркала для 'слепых' зон 86635985zz для volkswagen arteon перламутровочёрный</code> | <code>document: 86635985zz зеркала для 'слепых' зон иж юпитер2 голубой</code> |
279
+ | <code>query: elf bar lux 1500 лимонад голубой малины 1500 </code> | <code>document: одноразовая электронная сигарета эльф бар 1 5000 мл lemonade blue raspberry 340440526</code> | <code>document: elf bar vibe 1000 мохито зелёного яблока 1000</code> |
280
+ | <code>query: удалитель наклеек chevrolet corvette onyx</code> | <code>document: удалитель наклеек 20810588pl для chevrolet corvette оникс</code> | <code>document: удалитель наклеек 20810588pl для maserati levante янтарный</code> |
281
+ * Loss: [<code>SpladeLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#spladeloss) with these parameters:
282
+ ```json
283
+ {
284
+ "loss": "SparseMultipleNegativesRankingLoss(scale=1.0, similarity_fct='dot_score', gather_across_devices=False)",
285
+ "document_regularizer_weight": 1e-05,
286
+ "query_regularizer_weight": 1e-05
287
+ }
288
+ ```
289
+
290
+ ### Training Hyperparameters
291
+ #### Non-Default Hyperparameters
292
+
293
+ - `eval_strategy`: steps
294
+ - `gradient_accumulation_steps`: 16
295
+ - `learning_rate`: 0.0001
296
+ - `num_train_epochs`: 1
297
+ - `warmup_steps`: 10
298
+ - `fp16`: True
299
+ - `load_best_model_at_end`: True
300
+ - `router_mapping`: {'query': 'anchor', 'document': 'positive'}
301
+
302
+ #### All Hyperparameters
303
+ <details><summary>Click to expand</summary>
304
+
305
+ - `overwrite_output_dir`: False
306
+ - `do_predict`: False
307
+ - `eval_strategy`: steps
308
+ - `prediction_loss_only`: True
309
+ - `per_device_train_batch_size`: 8
310
+ - `per_device_eval_batch_size`: 8
311
+ - `per_gpu_train_batch_size`: None
312
+ - `per_gpu_eval_batch_size`: None
313
+ - `gradient_accumulation_steps`: 16
314
+ - `eval_accumulation_steps`: None
315
+ - `torch_empty_cache_steps`: None
316
+ - `learning_rate`: 0.0001
317
+ - `weight_decay`: 0.0
318
+ - `adam_beta1`: 0.9
319
+ - `adam_beta2`: 0.999
320
+ - `adam_epsilon`: 1e-08
321
+ - `max_grad_norm`: 1.0
322
+ - `num_train_epochs`: 1
323
+ - `max_steps`: -1
324
+ - `lr_scheduler_type`: linear
325
+ - `lr_scheduler_kwargs`: {}
326
+ - `warmup_ratio`: 0.0
327
+ - `warmup_steps`: 10
328
+ - `log_level`: passive
329
+ - `log_level_replica`: warning
330
+ - `log_on_each_node`: True
331
+ - `logging_nan_inf_filter`: True
332
+ - `save_safetensors`: True
333
+ - `save_on_each_node`: False
334
+ - `save_only_model`: False
335
+ - `restore_callback_states_from_checkpoint`: False
336
+ - `no_cuda`: False
337
+ - `use_cpu`: False
338
+ - `use_mps_device`: False
339
+ - `seed`: 42
340
+ - `data_seed`: None
341
+ - `jit_mode_eval`: False
342
+ - `bf16`: False
343
+ - `fp16`: True
344
+ - `fp16_opt_level`: O1
345
+ - `half_precision_backend`: auto
346
+ - `bf16_full_eval`: False
347
+ - `fp16_full_eval`: False
348
+ - `tf32`: None
349
+ - `local_rank`: 0
350
+ - `ddp_backend`: None
351
+ - `tpu_num_cores`: None
352
+ - `tpu_metrics_debug`: False
353
+ - `debug`: []
354
+ - `dataloader_drop_last`: False
355
+ - `dataloader_num_workers`: 0
356
+ - `dataloader_prefetch_factor`: None
357
+ - `past_index`: -1
358
+ - `disable_tqdm`: False
359
+ - `remove_unused_columns`: True
360
+ - `label_names`: None
361
+ - `load_best_model_at_end`: True
362
+ - `ignore_data_skip`: False
363
+ - `fsdp`: []
364
+ - `fsdp_min_num_params`: 0
365
+ - `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
366
+ - `fsdp_transformer_layer_cls_to_wrap`: None
367
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
368
+ - `parallelism_config`: None
369
+ - `deepspeed`: None
370
+ - `label_smoothing_factor`: 0.0
371
+ - `optim`: adamw_torch_fused
372
+ - `optim_args`: None
373
+ - `adafactor`: False
374
+ - `group_by_length`: False
375
+ - `length_column_name`: length
376
+ - `project`: huggingface
377
+ - `trackio_space_id`: trackio
378
+ - `ddp_find_unused_parameters`: None
379
+ - `ddp_bucket_cap_mb`: None
380
+ - `ddp_broadcast_buffers`: False
381
+ - `dataloader_pin_memory`: True
382
+ - `dataloader_persistent_workers`: False
383
+ - `skip_memory_metrics`: True
384
+ - `use_legacy_prediction_loop`: False
385
+ - `push_to_hub`: False
386
+ - `resume_from_checkpoint`: None
387
+ - `hub_model_id`: None
388
+ - `hub_strategy`: every_save
389
+ - `hub_private_repo`: None
390
+ - `hub_always_push`: False
391
+ - `hub_revision`: None
392
+ - `gradient_checkpointing`: False
393
+ - `gradient_checkpointing_kwargs`: None
394
+ - `include_inputs_for_metrics`: False
395
+ - `include_for_metrics`: []
396
+ - `eval_do_concat_batches`: True
397
+ - `fp16_backend`: auto
398
+ - `push_to_hub_model_id`: None
399
+ - `push_to_hub_organization`: None
400
+ - `mp_parameters`:
401
+ - `auto_find_batch_size`: False
402
+ - `full_determinism`: False
403
+ - `torchdynamo`: None
404
+ - `ray_scope`: last
405
+ - `ddp_timeout`: 1800
406
+ - `torch_compile`: False
407
+ - `torch_compile_backend`: None
408
+ - `torch_compile_mode`: None
409
+ - `include_tokens_per_second`: False
410
+ - `include_num_input_tokens_seen`: no
411
+ - `neftune_noise_alpha`: None
412
+ - `optim_target_modules`: None
413
+ - `batch_eval_metrics`: False
414
+ - `eval_on_start`: False
415
+ - `use_liger_kernel`: False
416
+ - `liger_kernel_config`: None
417
+ - `eval_use_gather_object`: False
418
+ - `average_tokens_across_devices`: True
419
+ - `prompts`: None
420
+ - `batch_sampler`: batch_sampler
421
+ - `multi_dataset_batch_sampler`: proportional
422
+ - `router_mapping`: {'query': 'anchor', 'document': 'positive'}
423
+ - `learning_rate_mapping`: {}
424
+
425
+ </details>
426
+
427
+ ### Training Logs
428
+ | Epoch | Step | Validation Loss | val_set_fine_cosine_ndcg@10 |
429
+ |:------:|:----:|:---------------:|:---------------------------:|
430
+ | 0.0125 | 10 | 0.8461 | 0.7841 |
431
+ | 0.0251 | 20 | 0.8195 | 0.8009 |
432
+ | 0.0376 | 30 | 0.7884 | 0.7967 |
433
+ | 0.0501 | 40 | 0.7641 | 0.8097 |
434
+ | 0.0627 | 50 | 0.7503 | 0.8146 |
435
+ | 0.0752 | 60 | 0.7140 | 0.8151 |
436
+ | 0.0877 | 70 | 0.7165 | 0.8180 |
437
+ | 0.1003 | 80 | 0.6955 | 0.8131 |
438
+ | 0.1128 | 90 | 0.6866 | 0.8157 |
439
+ | 0.1253 | 100 | 0.6735 | 0.8170 |
440
+
441
+
442
+ ### Framework Versions
443
+ - Python: 3.12.12
444
+ - Sentence Transformers: 5.2.2
445
+ - Transformers: 4.57.1
446
+ - PyTorch: 2.8.0+cu126
447
+ - Accelerate: 1.11.0
448
+ - Datasets: 4.4.2
449
+ - Tokenizers: 0.22.1
450
+
451
+ ## Citation
452
+
453
+ ### BibTeX
454
+
455
+ #### Sentence Transformers
456
+ ```bibtex
457
+ @inproceedings{reimers-2019-sentence-bert,
458
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
459
+ author = "Reimers, Nils and Gurevych, Iryna",
460
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
461
+ month = "11",
462
+ year = "2019",
463
+ publisher = "Association for Computational Linguistics",
464
+ url = "https://arxiv.org/abs/1908.10084",
465
+ }
466
+ ```
467
+
468
+ #### SpladeLoss
469
+ ```bibtex
470
+ @misc{formal2022distillationhardnegativesampling,
471
+ title={From Distillation to Hard Negative Sampling: Making Sparse Neural IR Models More Effective},
472
+ author={Thibault Formal and Carlos Lassance and Benjamin Piwowarski and Stéphane Clinchant},
473
+ year={2022},
474
+ eprint={2205.04733},
475
+ archivePrefix={arXiv},
476
+ primaryClass={cs.IR},
477
+ url={https://arxiv.org/abs/2205.04733},
478
+ }
479
+ ```
480
+
481
+ #### SparseMultipleNegativesRankingLoss
482
+ ```bibtex
483
+ @misc{henderson2017efficient,
484
+ title={Efficient Natural Language Response Suggestion for Smart Reply},
485
+ author={Matthew Henderson and Rami Al-Rfou and Brian Strope and Yun-hsuan Sung and Laszlo Lukacs and Ruiqi Guo and Sanjiv Kumar and Balint Miklos and Ray Kurzweil},
486
+ year={2017},
487
+ eprint={1705.00652},
488
+ archivePrefix={arXiv},
489
+ primaryClass={cs.CL}
490
+ }
491
+ ```
492
+
493
+ #### FlopsLoss
494
+ ```bibtex
495
+ @article{paria2020minimizing,
496
+ title={Minimizing flops to learn efficient sparse representations},
497
+ author={Paria, Biswajit and Yeh, Chih-Kuan and Yen, Ian EH and Xu, Ning and Ravikumar, Pradeep and P{'o}czos, Barnab{'a}s},
498
+ journal={arXiv preprint arXiv:2004.05665},
499
+ year={2020}
500
+ }
501
+ ```
502
+
503
+ <!--
504
+ ## Glossary
505
+
506
+ *Clearly define terms in order to be accessible across audiences.*
507
+ -->
508
+
509
+ <!--
510
+ ## Model Card Authors
511
+
512
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
513
+ -->
514
+
515
+ <!--
516
+ ## Model Card Contact
517
+
518
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
519
+ -->
config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_sliding_window_pattern": 6,
3
+ "architectures": [
4
+ "Gemma3TextModel"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "attn_logit_softcapping": null,
9
+ "bos_token_id": 2,
10
+ "dtype": "float32",
11
+ "eos_token_id": 1,
12
+ "final_logit_softcapping": null,
13
+ "head_dim": 256,
14
+ "hidden_activation": "gelu_pytorch_tanh",
15
+ "hidden_size": 768,
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 1152,
18
+ "layer_types": [
19
+ "sliding_attention",
20
+ "sliding_attention",
21
+ "sliding_attention",
22
+ "sliding_attention",
23
+ "sliding_attention",
24
+ "full_attention",
25
+ "sliding_attention",
26
+ "sliding_attention",
27
+ "sliding_attention",
28
+ "sliding_attention",
29
+ "sliding_attention",
30
+ "full_attention",
31
+ "sliding_attention",
32
+ "sliding_attention",
33
+ "sliding_attention",
34
+ "sliding_attention",
35
+ "sliding_attention",
36
+ "full_attention",
37
+ "sliding_attention",
38
+ "sliding_attention",
39
+ "sliding_attention",
40
+ "sliding_attention",
41
+ "sliding_attention",
42
+ "full_attention"
43
+ ],
44
+ "max_position_embeddings": 2048,
45
+ "model_type": "gemma3_text",
46
+ "num_attention_heads": 3,
47
+ "num_hidden_layers": 24,
48
+ "num_key_value_heads": 1,
49
+ "pad_token_id": 0,
50
+ "query_pre_attn_scalar": 256,
51
+ "rms_norm_eps": 1e-06,
52
+ "rope_local_base_freq": 10000.0,
53
+ "rope_parameters": {
54
+ "full_attention": {
55
+ "rope_theta": 1000000.0,
56
+ "rope_type": "default"
57
+ },
58
+ "sliding_attention": {
59
+ "rope_theta": 10000.0,
60
+ "rope_type": "default"
61
+ }
62
+ },
63
+ "rope_scaling": null,
64
+ "rope_theta": 1000000.0,
65
+ "sliding_window": 65,
66
+ "transformers_version": "4.57.1",
67
+ "use_bidirectional_attention": true,
68
+ "use_cache": true,
69
+ "vocab_size": 262144
70
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "SentenceTransformer",
3
+ "__version__": {
4
+ "sentence_transformers": "5.2.2",
5
+ "transformers": "4.57.1",
6
+ "pytorch": "2.8.0+cu126"
7
+ },
8
+ "prompts": {
9
+ "query": "",
10
+ "document": ""
11
+ },
12
+ "default_prompt_name": null,
13
+ "similarity_fn_name": "cosine"
14
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cbf5a78393b6a033e0b8a63a57549964f7ed5c6fbeb4ba0694214f36123f2fd2
3
+ size 1211486072
modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_SparseLayer",
18
+ "type": "__main__.SparseLayer"
19
+ }
20
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 512,
3
+ "do_lower_case": false
4
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<bos>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<eos>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "mask_token": {
17
+ "content": "<mask>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "pad_token": {
24
+ "content": "<pad>",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "unk_token": {
31
+ "content": "<unk>",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ }
37
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc07dcf4350cd709c4e37bdb92d6f7b723f1396d28fb4307ec6a6d2c867f5861
3
+ size 33384696
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff