Add new SentenceTransformer model

1_Pooling/config.json

@@ -0,0 +1,10 @@
+{
+  "word_embedding_dimension": 1024,
+  "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
+}

README.md

@@ -0,0 +1,124 @@
+---
+language:
+- de
+library_name: sentence-transformers
+tags:
+- sentence-transformers
+- sentence-similarity
+- feature-extraction
+- loss:MatryoshkaLoss
+base_model: aari1995/gbert-large-2
+metrics:
+- spearman_cosine
+widget:
+- source_sentence: Ein Mann übt Boxen
+  sentences:
+  - Ein Affe praktiziert Kampfsportarten.
+  - Eine Person faltet ein Blatt Papier.
+  - Eine Frau geht mit ihrem Hund spazieren.
+- source_sentence: Zwei Frauen laufen.
+  sentences:
+  - Frauen laufen.
+  - Die Frau prüft die Augen des Mannes.
+  - Ein Mann ist auf einem Dach
+pipeline_tag: sentence-similarity
+license: apache-2.0
+---
+
+# 🇩🇪 German Semantic V3b 🇩🇪
+### (and [**German_Semantic_V3**](https://huggingface.co/aari1995/German_Semantic_V3))
+
+The successors of [German_Semantic_STS_V2](https://huggingface.co/aari1995/German_Semantic_STS_V2) are here and come with loads of cool new features! While [German_Semantic_V3](https://huggingface.co/aari1995/German_Semantic_V3) is really knowledge-heavy, V3b is more focused on performance. Feel free to provide feedback on the model and what you would like to see next.
+
+**Note:** To run this model properly, see "Usage".
+
+# Major updates and USPs:
+
+- **Flexibility:** Trained with flexible sequence-length and embedding truncation, flexibility is a core feature of the model. Yet, smaller dimensions bring a minor trade-off in quality. 
+- **Sequence length:** Embed up to 8192 tokens (16 times more than V2 and other models)
+- **Matryoshka Embeddings:** The model is trained for embedding sizes from 1024 down to 64, allowing you to store much smaller embeddings with little quality loss.
+- **German only:** This model is German-only, it has rich cultural knowledge about Germany and German topics. Therefore, also the model to learn more efficient thanks to its tokenizer, deal better with shorter queries and generally be more nuanced in many scenarios.
+- **Typo and Casing**: This model was trained to be robust against minor typos and casing, leading to slightly weaker benchmark performance and learning during training, but higher robustness of the embeddings.
+- **Pooling Function:** Moving away from mean pooling towards using the CLS token. Generally seems to learn better after the stage-2 pretraining and allows for more flexibility.
+- **License:** Apache 2.0
+
+
+# Usage:
+
+This model has some build-in functionality that is rather hidden. To profit from it, use this code:
+
+```python
+from sentence_transformers import SentenceTransformer
+
+
+matryoshka_dim = 1024 # How big your embeddings should be, choose from: 64, 128, 256, 512, 768, 1024
+model = SentenceTransformer("aari1995/German_Semantic_V3", trust_remote_code=True, truncate_dim=matryoshka_dim)
+
+# model.truncate_dim = 64 # truncation dimensions can also be changed after loading
+# model.max_seq_length = 512 #optionally, set your maximum sequence length lower if your hardware is limited 
+
+# Run inference
+sentences = [
+    'Eine Flagge weht.',
+    'Die Flagge bewegte sich in der Luft.',
+    'Zwei Personen beobachten das Wasser.',
+]
+
+# For FP16 embeddings (half space, no quality loss)
+embeddings = model.encode(sentences, convert_to_tensor=True).half()
+
+# For FP32 embeddings (takes more space)
+# embeddings = model.encode(sentences)
+
+# Get the similarity scores for the embeddings
+similarities = model.similarity(embeddings, embeddings)
+
+```
+
+
+# FAQ
+
+**Q: Is this Model better than V2?**
+
+**A:** In terms of flexibility, this model is better. Performance wise, in most of the experiments this model is also better.
+
+
+**Q: What is the difference between V3 and V3b?**
+
+**A:** V3 is slightly worse on benchmarks, while V3b has a knowledge cutoff by 2020, so it really depends on your use-case what model to use.
+
+If you want peak performance and do not worry too much about recent developments, take this one (V3b).
+
+If you are fine with sacrificing a few points on benchmarks and want the model to know what happened from 2020 on (elections, covid, other cultural events etc.), I'd suggest you use [German_Semantic_V3](https://huggingface.co/aari1995/German_Semantic_V3).
+
+Another noticable difference is that V3 has a broader cosine_similarity spectrum, reaching from -1 to 1 (but mostly, the least is over -0.2). On the other side, V3b is more aligned with V2 and the similarity spectrum is around 0 to 1. Also, V3 uses cls_pooling while V3b uses mean_pooling.
+
+**Q: How does the model perform vs. multilingual models?**
+
+**A:** There are really great multilingual models that will be very useful for many use-cases. This model shines with its cultural knowledge and knowledge about German people and behaviour. 
+
+
+**Q: What is the trade-off when reducing the embedding size?**
+
+**A:** Broadly speaking, when going from 1024 to 512 dimensions, there is very little trade-off (1 percent). When going down to 64 dimensions, you may face a decrease of up to 3 percent.
+
+
+# Evaluation
+
+Storage comparison:
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/5f3801ab7e583543386217ac/Aa5WzHanj-DXc86AKxpEz.png)
+
+Benchmarks: soon.
+
+# Up next:
+German_Semantic_V3_Instruct: Guiding your embeddings towards self-selected aspects. - planned: 2024.
+
+# Thank You and Credits
+
+- To [jinaAI](https://huggingface.co/jinaai) for their BERT implementation that is used, especially ALiBi
+- To [deepset](https://huggingface.co/deepset) for the gbert-large, which is a really great model
+- To [occiglot](https://huggingface.co/occiglot) and OSCAR for their data used to pre-train the model
+- To [Tom](https://huggingface.co/tomaarsen), especially for sentence-transformers, [Björn and Jan from ellamind](https://ellamind.com/de/) for the consultation
+- To [Meta](https://huggingface.co/facebook) for XNLI which is used in variations
+
+Idea, Training and Implementation by Aaron Chibb

config.json

@@ -0,0 +1,33 @@
+{
+  "_name_or_path": "aari1995/German_Semantic_V3b",
+  "architectures": [
+    "BertModel"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "auto_map": {
+    "AutoConfig": "aari1995/German_Semantic_V3b--configuration_bert.JinaBertConfig",
+    "AutoModel": "aari1995/German_Semantic_V3b--modeling_bert.JinaBertModel",
+    "AutoModelForMaskedLM": "aari1995/German_Semantic_V3b--modeling_bert.JinaBertForMaskedLM",
+    "AutoModelForSequenceClassification": "aari1995/German_Semantic_V3b--modeling_bert.JinaBertForSequenceClassification"
+  },
+  "classifier_dropout": null,
+  "emb_pooler": null,
+  "feed_forward_type": "original",
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 1024,
+  "initializer_range": 0.02,
+  "intermediate_size": 4096,
+  "layer_norm_eps": 1e-12,
+  "max_position_embeddings": 8192,
+  "model_type": "bert",
+  "num_attention_heads": 16,
+  "num_hidden_layers": 24,
+  "pad_token_id": 0,
+  "position_embedding_type": "alibi",
+  "torch_dtype": "float32",
+  "transformers_version": "4.48.2",
+  "type_vocab_size": 2,
+  "use_cache": true,
+  "vocab_size": 31102
+}

config_sentence_transformers.json

@@ -0,0 +1,10 @@
+{
+  "__version__": {
+    "sentence_transformers": "3.4.1",
+    "transformers": "4.48.2",
+    "pytorch": "2.5.1+cu124"
+  },
+  "prompts": {},
+  "default_prompt_name": null,
+  "similarity_fn_name": "cosine"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b442c2a8627438b6013b20d642c2b5d3e1d40e78a0a9e4ef03908925a68661a
+size 1374445424

modules.json

@@ -0,0 +1,14 @@
+[
+  {
+    "idx": 0,
+    "name": "0",
+    "path": "",
+    "type": "sentence_transformers.models.Transformer"
+  },
+  {
+    "idx": 1,
+    "name": "1",
+    "path": "1_Pooling",
+    "type": "sentence_transformers.models.Pooling"
+  }
+]

sentence_bert_config.json

@@ -0,0 +1,4 @@
+{
+  "max_seq_length": 8192,
+  "do_lower_case": false
+}

special_tokens_map.json

@@ -0,0 +1,37 @@
+{
+  "cls_token": {
+    "content": "[CLS]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "mask_token": {
+    "content": "[MASK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "[PAD]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "sep_token": {
+    "content": "[SEP]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "[UNK]",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,66 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "101": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "102": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "103": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "104": {
+      "content": "[MASK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "[CLS]",
+  "do_basic_tokenize": true,
+  "do_lower_case": false,
+  "extra_special_tokens": {},
+  "mask_token": "[MASK]",
+  "max_len": 9999999999,
+  "max_length": 8192,
+  "model_max_length": 8192,
+  "never_split": null,
+  "pad_to_multiple_of": null,
+  "pad_token": "[PAD]",
+  "pad_token_type_id": 0,
+  "padding_side": "right",
+  "sep_token": "[SEP]",
+  "stride": 0,
+  "strip_accents": false,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "BertTokenizer",
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
+  "unk_token": "[UNK]"
+}