rdxtremity commited on
Commit
b737adb
·
verified ·
1 Parent(s): 19d4258

add SPLADE v6_v2 (qwen30-pairs A/B winner, sparse nDCG 0.8189)

Browse files
splade_v6_v2/1_SpladePooling/config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "pooling_strategy": "max",
3
+ "activation_function": "relu",
4
+ "embedding_dimension": 119547
5
+ }
splade_v6_v2/README.md ADDED
@@ -0,0 +1,497 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - sentence-transformers
4
+ - sparse-encoder
5
+ - sparse
6
+ - splade
7
+ - generated_from_trainer
8
+ - dataset_size:1298215
9
+ - loss:SpladeLoss
10
+ - loss:SparseMultipleNegativesRankingLoss
11
+ - loss:FlopsLoss
12
+ base_model: distilbert/distilbert-base-multilingual-cased
13
+ widget:
14
+ - text: '[Clan London] Women''s Logo Tote Bag Olive (Olive) | حقيبة توت بشعار العلامة
15
+ التجارية للنساء، زيتي (زيتي). Category: Bags > Shoppers & Totes.'
16
+ - text: '[Fire & Glory] Women''s Animal Print Midi Dress Light Gray (Light Gray) |
17
+ فستان ميدي بأكمام طويلة وطبعة جرافيك للنساء، أسود (رمادي فاتح). Category: Dresses
18
+ > Midi Dresses.'
19
+ - text: '[melissa] Melissa Becky AD Pink (Pink) | ميليسا بيكي إيه دي (وردي). Category:
20
+ Shoes > Slides & Flip-Flops.'
21
+ - text: '[H&M] Curvy Ultra High-Rise Wide-Leg Jeans (Pastel Blue) | جينز كيرفي بخصر
22
+ مرتفع جداً وأرجل واسعة (أزرق باستيل). Category: Denim > Jeans.'
23
+ - text: '[Splash Basics] Flexi Comfort Drawstring A-line Midi Skirt Blue (Blue) |
24
+ الرباط ألف خط تنورة ميدي (أزرق). Category: Bottoms > Skirts.'
25
+ pipeline_tag: feature-extraction
26
+ library_name: sentence-transformers
27
+ ---
28
+
29
+ # SPLADE Sparse Encoder
30
+
31
+ This is a [SPLADE Sparse Encoder](https://www.sbert.net/docs/sparse_encoder/usage/usage.html) model finetuned from [distilbert/distilbert-base-multilingual-cased](https://huggingface.co/distilbert/distilbert-base-multilingual-cased) on the pairs and triplets datasets using the [sentence-transformers](https://www.SBERT.net) library. It maps sentences & paragraphs to a 119547-dimensional sparse vector space and can be used for semantic search and sparse retrieval.
32
+ ## Model Details
33
+
34
+ ### Model Description
35
+ - **Model Type:** SPLADE Sparse Encoder
36
+ - **Base model:** [distilbert/distilbert-base-multilingual-cased](https://huggingface.co/distilbert/distilbert-base-multilingual-cased) <!-- at revision 45c032ab32cc946ad88a166f7cb282f58c753c2e -->
37
+ - **Maximum Sequence Length:** 128 tokens
38
+ - **Output Dimensionality:** 119547 dimensions
39
+ - **Similarity Function:** Dot Product
40
+ - **Supported Modality:** Text
41
+ - **Training Datasets:**
42
+ - pairs
43
+ - triplets
44
+ <!-- - **Language:** Unknown -->
45
+ <!-- - **License:** Unknown -->
46
+
47
+ ### Model Sources
48
+
49
+ - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
50
+ - **Documentation:** [Sparse Encoder Documentation](https://www.sbert.net/docs/sparse_encoder/usage/usage.html)
51
+ - **Repository:** [Sentence Transformers on GitHub](https://github.com/huggingface/sentence-transformers)
52
+ - **Hugging Face:** [Sparse Encoders on Hugging Face](https://huggingface.co/models?library=sentence-transformers&other=sparse-encoder)
53
+
54
+ ### Full Model Architecture
55
+
56
+ ```
57
+ SparseEncoder(
58
+ (0): Transformer({'transformer_task': 'fill-mask', 'modality_config': {'text': {'method': 'forward', 'method_output_name': 'logits'}}, 'module_output_name': 'token_embeddings', 'architecture': 'DistilBertForMaskedLM'})
59
+ (1): SpladePooling({'pooling_strategy': 'max', 'activation_function': 'relu', 'embedding_dimension': 119547})
60
+ )
61
+ ```
62
+
63
+ ## Usage
64
+
65
+ ### Direct Usage (Sentence Transformers)
66
+
67
+ First install the Sentence Transformers library:
68
+
69
+ ```bash
70
+ pip install -U sentence-transformers
71
+ ```
72
+ Then you can load this model and run inference.
73
+ ```python
74
+ from sentence_transformers import SparseEncoder
75
+
76
+ # Download from the 🤗 Hub
77
+ model = SparseEncoder("sparse_encoder_model_id")
78
+ # Run inference
79
+ sentences = [
80
+ 'شنطة زهّانة',
81
+ '[PANDORA] Aquarius Zodiac Charm (Silver) | دلاية برج الدلو (فضي). Category: > Accessories.',
82
+ "[Hurley] Women's Crew Neck Solid Sweatshirt Dark Salmon (Dark Salmon) | سويت شيرت نسائي سادة بياقة دائرية، برتقالي (سلموني غامق). Category: Hoodies & Sweatshirt > Sweatshirts.",
83
+ ]
84
+ embeddings = model.encode(sentences)
85
+ print(embeddings.shape)
86
+ # [3, 119547]
87
+
88
+ # Get the similarity scores for the embeddings
89
+ similarities = model.similarity(embeddings, embeddings)
90
+ print(similarities)
91
+ # tensor([[ 11.9930, 13.0235, 7.8521],
92
+ # [ 13.0235, 491.0651, 65.4858],
93
+ # [ 7.8521, 65.4858, 246.8788]])
94
+ ```
95
+ <!--
96
+ ### Direct Usage (Transformers)
97
+
98
+ <details><summary>Click to see the direct usage in Transformers</summary>
99
+
100
+ </details>
101
+ -->
102
+
103
+ <!--
104
+ ### Downstream Usage (Sentence Transformers)
105
+
106
+ You can finetune this model on your own dataset.
107
+
108
+ <details><summary>Click to expand</summary>
109
+
110
+ </details>
111
+ -->
112
+
113
+ <!--
114
+ ### Out-of-Scope Use
115
+
116
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
117
+ -->
118
+
119
+ <!--
120
+ ## Bias, Risks and Limitations
121
+
122
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
123
+ -->
124
+
125
+ <!--
126
+ ### Recommendations
127
+
128
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
129
+ -->
130
+
131
+ ## Training Details
132
+
133
+ ### Training Datasets
134
+
135
+ #### pairs
136
+
137
+ * Dataset: pairs
138
+ * Size: 1,292,307 training samples
139
+ * Columns: <code>query</code> and <code>positive</code>
140
+ * Approximate statistics based on the first 100 samples:
141
+ | | query | positive |
142
+ |:---------|:----------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|
143
+ | type | string | string |
144
+ | modality | text | text |
145
+ | details | <ul><li>min: 6 tokens</li><li>mean: 10.33 tokens</li><li>max: 20 tokens</li></ul> | <ul><li>min: 28 tokens</li><li>mean: 51.14 tokens</li><li>max: 124 tokens</li></ul> |
146
+ * Samples:
147
+ | query | positive |
148
+ |:--------------------------------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
149
+ | <code>men's watch analog</code> | <code>[CASIO] Leather Strap Analog Watch (Mint Green) \| ساعة انالوج جلد اصلي (أخضر نعناعي). Category: > .</code> |
150
+ | <code>black hoodie women</code> | <code>[Tribe of 6] Women's Logo Hooded Sweatshirt Black (Black) \| سويت شيرت نسائي سادة بأكمام طويلة وغطاء رأس، أسود (أسود). Category: Hoodies & Sweatshirt > Hoodies.</code> |
151
+ | <code>وشاح مخطط للنساء</code> | <code>[MANGO] Geometric Stripe Scarf Beige (Beige) \| وشاح بنقوش هندسية مخططة (بيج). Category: Accessories > Scarves.</code> |
152
+ * Loss: [<code>SpladeLoss</code>](https://sbert.net/docs/package_reference/sparse_encoder/losses.html#spladeloss) with these parameters:
153
+ ```json
154
+ {
155
+ "loss": "SparseMultipleNegativesRankingLoss(scale=1.0, similarity_fct='dot_score', gather_across_devices=False, directions=('query_to_doc',), partition_mode='joint', hardness_mode=None, hardness_strength=0.0)",
156
+ "document_regularizer_weight": 3e-05,
157
+ "query_regularizer_weight": 5e-05
158
+ }
159
+ ```
160
+
161
+ #### triplets
162
+
163
+ * Dataset: triplets
164
+ * Size: 5,908 training samples
165
+ * Columns: <code>query</code>, <code>positive</code>, and <code>negative</code>
166
+ * Approximate statistics based on the first 100 samples:
167
+ | | query | positive | negative |
168
+ |:---------|:---------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------|:-----------------------------------------------------------------------------------|
169
+ | type | string | string | string |
170
+ | modality | text | text | text |
171
+ | details | <ul><li>min: 5 tokens</li><li>mean: 8.75 tokens</li><li>max: 17 tokens</li></ul> | <ul><li>min: 39 tokens</li><li>mean: 57.33 tokens</li><li>max: 82 tokens</li></ul> | <ul><li>min: 38 tokens</li><li>mean: 54.78 tokens</li><li>max: 88 tokens</li></ul> |
172
+ * Samples:
173
+ | query | positive | negative |
174
+ |:----------------------------------------------------------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
175
+ | <code>don't call me jennyfer jumpsuits & bodysuits</code> | <code>[Don't Call Me Jennyfer] Women's One Sleeve Bodysuit Green (Green) \| بودي سوت سادة بأكمام قصيرة للنساء، أخضر (أخضر). Category: Jumpsuits & Bodysuits > Body Suits.</code> | <code>[Ashita Fernandes] Kenza Cupro V-Neck Jumpsuit Royal Blue (Facebook Blue) \| كينزا جمبسوت كوبرو بفتحة رقبة V أزرق ملكي (أزرق فيسبوك). Category: Jumpsuits & Bodysuits > Jumpsuits.</code> |
176
+ | <code>don't call me jennyfer jumpsuits & bodysuits</code> | <code>[Don't Call Me Jennyfer] Women's One Sleeve Bodysuit Green (Green) \| بودي سوت سادة بأكمام قصيرة للنساء، أخضر (أخضر). Category: Jumpsuits & Bodysuits > Body Suits.</code> | <code>[Splash Intimates] Mesh Panel Shaping Bodysuit (Black) \| بودي سوت مشد بأجزاء شبكية (أسود). Category: Jumpsuits & Bodysuits > Body Suits.</code> |
177
+ | <code>don't call me jennyfer jumpsuits & bodysuits</code> | <code>[Don't Call Me Jennyfer] Women's One Sleeve Bodysuit Green (Green) \| بودي سوت سادة بأكمام قصيرة للنساء، أخضر (أخضر). Category: Jumpsuits & Bodysuits > Body Suits.</code> | <code>[Ashita Fernandes] Kenza Cupro V-Neck Jumpsuit Royal Blue (Facebook Blue) \| كينزا جمبسوت كوبرو بفتحة رقبة V أزرق ملكي (أزرق فيسبوك). Category: Jumpsuits & Bodysuits > Jumpsuits.</code> |
178
+ * Loss: [<code>SpladeLoss</code>](https://sbert.net/docs/package_reference/sparse_encoder/losses.html#spladeloss) with these parameters:
179
+ ```json
180
+ {
181
+ "loss": "SparseMultipleNegativesRankingLoss(scale=1.0, similarity_fct='dot_score', gather_across_devices=False, directions=('query_to_doc',), partition_mode='joint', hardness_mode=None, hardness_strength=0.0)",
182
+ "document_regularizer_weight": 3e-05,
183
+ "query_regularizer_weight": 5e-05
184
+ }
185
+ ```
186
+
187
+ ### Training Hyperparameters
188
+ #### Non-Default Hyperparameters
189
+
190
+ - `per_device_train_batch_size`: 16
191
+ - `num_train_epochs`: 1
192
+ - `learning_rate`: 2e-05
193
+ - `warmup_steps`: 0.1
194
+ - `gradient_accumulation_steps`: 4
195
+ - `bf16`: True
196
+ - `dataloader_num_workers`: 2
197
+
198
+ #### All Hyperparameters
199
+ <details><summary>Click to expand</summary>
200
+
201
+ - `per_device_train_batch_size`: 16
202
+ - `num_train_epochs`: 1
203
+ - `max_steps`: -1
204
+ - `learning_rate`: 2e-05
205
+ - `lr_scheduler_type`: linear
206
+ - `lr_scheduler_kwargs`: None
207
+ - `warmup_steps`: 0.1
208
+ - `optim`: adamw_torch
209
+ - `optim_args`: None
210
+ - `weight_decay`: 0.0
211
+ - `adam_beta1`: 0.9
212
+ - `adam_beta2`: 0.999
213
+ - `adam_epsilon`: 1e-08
214
+ - `optim_target_modules`: None
215
+ - `gradient_accumulation_steps`: 4
216
+ - `average_tokens_across_devices`: True
217
+ - `max_grad_norm`: 1.0
218
+ - `label_smoothing_factor`: 0.0
219
+ - `bf16`: True
220
+ - `fp16`: False
221
+ - `bf16_full_eval`: False
222
+ - `fp16_full_eval`: False
223
+ - `tf32`: None
224
+ - `gradient_checkpointing`: False
225
+ - `gradient_checkpointing_kwargs`: None
226
+ - `torch_compile`: False
227
+ - `torch_compile_backend`: None
228
+ - `torch_compile_mode`: None
229
+ - `use_liger_kernel`: False
230
+ - `liger_kernel_config`: None
231
+ - `use_cache`: False
232
+ - `neftune_noise_alpha`: None
233
+ - `torch_empty_cache_steps`: None
234
+ - `auto_find_batch_size`: False
235
+ - `log_on_each_node`: True
236
+ - `logging_nan_inf_filter`: True
237
+ - `include_num_input_tokens_seen`: no
238
+ - `log_level`: passive
239
+ - `log_level_replica`: warning
240
+ - `disable_tqdm`: False
241
+ - `project`: huggingface
242
+ - `trackio_space_id`: None
243
+ - `trackio_bucket_id`: None
244
+ - `trackio_static_space_id`: None
245
+ - `per_device_eval_batch_size`: 8
246
+ - `prediction_loss_only`: True
247
+ - `eval_on_start`: False
248
+ - `eval_do_concat_batches`: True
249
+ - `eval_use_gather_object`: False
250
+ - `eval_accumulation_steps`: None
251
+ - `include_for_metrics`: []
252
+ - `batch_eval_metrics`: False
253
+ - `save_only_model`: False
254
+ - `save_on_each_node`: False
255
+ - `enable_jit_checkpoint`: False
256
+ - `push_to_hub`: False
257
+ - `hub_private_repo`: None
258
+ - `hub_model_id`: None
259
+ - `hub_strategy`: every_save
260
+ - `hub_always_push`: False
261
+ - `hub_revision`: None
262
+ - `load_best_model_at_end`: False
263
+ - `ignore_data_skip`: False
264
+ - `restore_callback_states_from_checkpoint`: False
265
+ - `full_determinism`: False
266
+ - `seed`: 42
267
+ - `data_seed`: None
268
+ - `use_cpu`: False
269
+ - `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
270
+ - `parallelism_config`: None
271
+ - `dataloader_drop_last`: False
272
+ - `dataloader_num_workers`: 2
273
+ - `dataloader_pin_memory`: True
274
+ - `dataloader_persistent_workers`: False
275
+ - `dataloader_prefetch_factor`: None
276
+ - `remove_unused_columns`: True
277
+ - `label_names`: None
278
+ - `train_sampling_strategy`: random
279
+ - `length_column_name`: length
280
+ - `ddp_find_unused_parameters`: None
281
+ - `ddp_bucket_cap_mb`: None
282
+ - `ddp_broadcast_buffers`: False
283
+ - `ddp_static_graph`: None
284
+ - `ddp_backend`: None
285
+ - `ddp_timeout`: 1800
286
+ - `fsdp`: None
287
+ - `fsdp_config`: None
288
+ - `deepspeed`: None
289
+ - `debug`: []
290
+ - `skip_memory_metrics`: True
291
+ - `do_predict`: False
292
+ - `resume_from_checkpoint`: None
293
+ - `warmup_ratio`: None
294
+ - `local_rank`: -1
295
+ - `prompts`: None
296
+ - `batch_sampler`: batch_sampler
297
+ - `multi_dataset_batch_sampler`: proportional
298
+ - `router_mapping`: {}
299
+ - `learning_rate_mapping`: {}
300
+
301
+ </details>
302
+
303
+ ### Training Logs
304
+ <details><summary>Click to expand</summary>
305
+
306
+ | Epoch | Step | Training Loss |
307
+ |:------:|:-----:|:-------------:|
308
+ | 0.0099 | 200 | 679.9589 |
309
+ | 0.0197 | 400 | 1.7857 |
310
+ | 0.0296 | 600 | 1.0238 |
311
+ | 0.0394 | 800 | 0.8982 |
312
+ | 0.0493 | 1000 | 0.8221 |
313
+ | 0.0592 | 1200 | 0.7875 |
314
+ | 0.0690 | 1400 | 0.7405 |
315
+ | 0.0789 | 1600 | 0.7131 |
316
+ | 0.0887 | 1800 | 0.6761 |
317
+ | 0.0986 | 2000 | 0.6732 |
318
+ | 0.1085 | 2200 | 0.6397 |
319
+ | 0.1183 | 2400 | 0.6303 |
320
+ | 0.1282 | 2600 | 0.6021 |
321
+ | 0.1380 | 2800 | 0.5794 |
322
+ | 0.1479 | 3000 | 0.5552 |
323
+ | 0.1578 | 3200 | 0.5638 |
324
+ | 0.1676 | 3400 | 0.5633 |
325
+ | 0.1775 | 3600 | 0.5550 |
326
+ | 0.1873 | 3800 | 0.5473 |
327
+ | 0.1972 | 4000 | 0.5281 |
328
+ | 0.2070 | 4200 | 0.5189 |
329
+ | 0.2169 | 4400 | 0.5282 |
330
+ | 0.2268 | 4600 | 0.5080 |
331
+ | 0.2366 | 4800 | 0.5046 |
332
+ | 0.2465 | 5000 | 0.5111 |
333
+ | 0.2563 | 5200 | 0.4957 |
334
+ | 0.2662 | 5400 | 0.4932 |
335
+ | 0.2761 | 5600 | 0.5045 |
336
+ | 0.2859 | 5800 | 0.5019 |
337
+ | 0.2958 | 6000 | 0.4789 |
338
+ | 0.3056 | 6200 | 0.4866 |
339
+ | 0.3155 | 6400 | 0.4810 |
340
+ | 0.3254 | 6600 | 0.4891 |
341
+ | 0.3352 | 6800 | 0.4800 |
342
+ | 0.3451 | 7000 | 0.4659 |
343
+ | 0.3549 | 7200 | 0.4612 |
344
+ | 0.3648 | 7400 | 0.4644 |
345
+ | 0.3747 | 7600 | 0.4364 |
346
+ | 0.3845 | 7800 | 0.4672 |
347
+ | 0.3944 | 8000 | 0.4546 |
348
+ | 0.4042 | 8200 | 0.4631 |
349
+ | 0.4141 | 8400 | 0.4565 |
350
+ | 0.4240 | 8600 | 0.4242 |
351
+ | 0.4338 | 8800 | 0.4425 |
352
+ | 0.4437 | 9000 | 0.4597 |
353
+ | 0.4535 | 9200 | 0.4292 |
354
+ | 0.4634 | 9400 | 0.4406 |
355
+ | 0.4733 | 9600 | 0.4331 |
356
+ | 0.4831 | 9800 | 0.4342 |
357
+ | 0.4930 | 10000 | 0.4483 |
358
+ | 0.5028 | 10200 | 0.4254 |
359
+ | 0.5127 | 10400 | 0.4182 |
360
+ | 0.5226 | 10600 | 0.4173 |
361
+ | 0.5324 | 10800 | 0.4120 |
362
+ | 0.5423 | 11000 | 0.4197 |
363
+ | 0.5521 | 11200 | 0.4029 |
364
+ | 0.5620 | 11400 | 0.4016 |
365
+ | 0.5719 | 11600 | 0.4159 |
366
+ | 0.5817 | 11800 | 0.4033 |
367
+ | 0.5916 | 12000 | 0.4250 |
368
+ | 0.6014 | 12200 | 0.4013 |
369
+ | 0.6113 | 12400 | 0.4234 |
370
+ | 0.6211 | 12600 | 0.4103 |
371
+ | 0.6310 | 12800 | 0.4046 |
372
+ | 0.6409 | 13000 | 0.4123 |
373
+ | 0.6507 | 13200 | 0.4074 |
374
+ | 0.6606 | 13400 | 0.4098 |
375
+ | 0.6704 | 13600 | 0.4192 |
376
+ | 0.6803 | 13800 | 0.3887 |
377
+ | 0.6902 | 14000 | 0.4017 |
378
+ | 0.7000 | 14200 | 0.3956 |
379
+ | 0.7099 | 14400 | 0.3971 |
380
+ | 0.7197 | 14600 | 0.3964 |
381
+ | 0.7296 | 14800 | 0.3921 |
382
+ | 0.7395 | 15000 | 0.3947 |
383
+ | 0.7493 | 15200 | 0.3795 |
384
+ | 0.7592 | 15400 | 0.3813 |
385
+ | 0.7690 | 15600 | 0.3931 |
386
+ | 0.7789 | 15800 | 0.3792 |
387
+ | 0.7888 | 16000 | 0.3878 |
388
+ | 0.7986 | 16200 | 0.3703 |
389
+ | 0.8085 | 16400 | 0.3836 |
390
+ | 0.8183 | 16600 | 0.3871 |
391
+ | 0.8282 | 16800 | 0.3751 |
392
+ | 0.8381 | 17000 | 0.4005 |
393
+ | 0.8479 | 17200 | 0.3887 |
394
+ | 0.8578 | 17400 | 0.3873 |
395
+ | 0.8676 | 17600 | 0.3811 |
396
+ | 0.8775 | 17800 | 0.3841 |
397
+ | 0.8874 | 18000 | 0.3841 |
398
+ | 0.8972 | 18200 | 0.3701 |
399
+ | 0.9071 | 18400 | 0.3913 |
400
+ | 0.9169 | 18600 | 0.3848 |
401
+ | 0.9268 | 18800 | 0.3885 |
402
+ | 0.9367 | 19000 | 0.3648 |
403
+ | 0.9465 | 19200 | 0.3842 |
404
+ | 0.9564 | 19400 | 0.3989 |
405
+ | 0.9662 | 19600 | 0.3930 |
406
+ | 0.9761 | 19800 | 0.3944 |
407
+ | 0.9860 | 20000 | 0.3712 |
408
+ | 0.9958 | 20200 | 0.3762 |
409
+
410
+ </details>
411
+
412
+ ### Training Time
413
+ - **Training**: 48.5 minutes
414
+
415
+ ### Framework Versions
416
+ - Python: 3.11.12
417
+ - Sentence Transformers: 5.6.0
418
+ - Transformers: 5.12.1
419
+ - PyTorch: 2.7.0+cu128
420
+ - Accelerate: 1.14.0
421
+ - Datasets: 5.0.0
422
+ - Tokenizers: 0.22.2
423
+
424
+ ## Additional Resources
425
+
426
+ - [Training and Finetuning Sparse Embedding Models with Sentence Transformers](https://huggingface.co/blog/train-sparse-encoder): the end-to-end guide for training or finetuning SPLADE and other sparse encoder models.
427
+
428
+ ## Citation
429
+
430
+ ### BibTeX
431
+
432
+ #### Sentence Transformers
433
+ ```bibtex
434
+ @inproceedings{reimers-2019-sentence-bert,
435
+ title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
436
+ author = "Reimers, Nils and Gurevych, Iryna",
437
+ booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
438
+ month = "11",
439
+ year = "2019",
440
+ publisher = "Association for Computational Linguistics",
441
+ url = "https://arxiv.org/abs/1908.10084",
442
+ }
443
+ ```
444
+
445
+ #### SpladeLoss
446
+ ```bibtex
447
+ @misc{formal2022distillationhardnegativesampling,
448
+ title={From Distillation to Hard Negative Sampling: Making Sparse Neural IR Models More Effective},
449
+ author={Thibault Formal and Carlos Lassance and Benjamin Piwowarski and Stéphane Clinchant},
450
+ year={2022},
451
+ eprint={2205.04733},
452
+ archivePrefix={arXiv},
453
+ primaryClass={cs.IR},
454
+ url={https://arxiv.org/abs/2205.04733},
455
+ }
456
+ ```
457
+
458
+ #### SparseMultipleNegativesRankingLoss
459
+ ```bibtex
460
+ @misc{oord2019representationlearningcontrastivepredictive,
461
+ title={Representation Learning with Contrastive Predictive Coding},
462
+ author={Aaron van den Oord and Yazhe Li and Oriol Vinyals},
463
+ year={2019},
464
+ eprint={1807.03748},
465
+ archivePrefix={arXiv},
466
+ primaryClass={cs.LG},
467
+ url={https://arxiv.org/abs/1807.03748},
468
+ }
469
+ ```
470
+
471
+ #### FlopsLoss
472
+ ```bibtex
473
+ @article{paria2020minimizing,
474
+ title={Minimizing flops to learn efficient sparse representations},
475
+ author={Paria, Biswajit and Yeh, Chih-Kuan and Yen, Ian EH and Xu, Ning and Ravikumar, Pradeep and P{'o}czos, Barnab{'a}s},
476
+ journal={arXiv preprint arXiv:2004.05665},
477
+ year={2020}
478
+ }
479
+ ```
480
+
481
+ <!--
482
+ ## Glossary
483
+
484
+ *Clearly define terms in order to be accessible across audiences.*
485
+ -->
486
+
487
+ <!--
488
+ ## Model Card Authors
489
+
490
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
491
+ -->
492
+
493
+ <!--
494
+ ## Model Card Contact
495
+
496
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
497
+ -->
splade_v6_v2/config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation": "gelu",
3
+ "architectures": [
4
+ "DistilBertForMaskedLM"
5
+ ],
6
+ "attention_dropout": 0.1,
7
+ "bos_token_id": null,
8
+ "dim": 768,
9
+ "dropout": 0.1,
10
+ "dtype": "float32",
11
+ "eos_token_id": null,
12
+ "hidden_dim": 3072,
13
+ "initializer_range": 0.02,
14
+ "max_position_embeddings": 512,
15
+ "model_type": "distilbert",
16
+ "n_heads": 12,
17
+ "n_layers": 6,
18
+ "output_past": true,
19
+ "pad_token_id": 0,
20
+ "qa_dropout": 0.1,
21
+ "seq_classif_dropout": 0.2,
22
+ "sinusoidal_pos_embds": false,
23
+ "tie_weights_": true,
24
+ "tie_word_embeddings": true,
25
+ "transformers_version": "5.12.1",
26
+ "use_cache": false,
27
+ "vocab_size": 119547
28
+ }
splade_v6_v2/config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "pytorch": "2.7.0+cu128",
4
+ "sentence_transformers": "5.6.0",
5
+ "transformers": "5.12.1"
6
+ },
7
+ "default_prompt_name": null,
8
+ "model_type": "SparseEncoder",
9
+ "prompts": {
10
+ "document": "",
11
+ "query": ""
12
+ },
13
+ "similarity_fn_name": null
14
+ }
splade_v6_v2/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:011f6d4142e86a7c2509572f5e07be4d89c6c3eed532e603b8cbcf04e73c4bf2
3
+ size 541795684
splade_v6_v2/modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.sparse_encoder.modules.mlm_transformer.MLMTransformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_SpladePooling",
12
+ "type": "sentence_transformers.sparse_encoder.modules.splade_pooling.SpladePooling"
13
+ }
14
+ ]
splade_v6_v2/sentence_bert_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "transformer_task": "fill-mask",
3
+ "modality_config": {
4
+ "text": {
5
+ "method": "forward",
6
+ "method_output_name": "logits"
7
+ }
8
+ },
9
+ "module_output_name": "token_embeddings"
10
+ }
splade_v6_v2/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
splade_v6_v2/tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "cls_token": "[CLS]",
4
+ "do_lower_case": false,
5
+ "is_local": false,
6
+ "local_files_only": false,
7
+ "mask_token": "[MASK]",
8
+ "model_max_length": 128,
9
+ "pad_token": "[PAD]",
10
+ "sep_token": "[SEP]",
11
+ "strip_accents": null,
12
+ "tokenize_chinese_chars": true,
13
+ "tokenizer_class": "BertTokenizer",
14
+ "unk_token": "[UNK]"
15
+ }