Sentence Similarity
sentence-transformers
Safetensors
bert
multilingual
layer-pruning
vocab-pruning
knowledge-distillation
minilm-l12
text-embeddings-inference
Instructions to use gomyk/minilm-student-L6_uniform_distilled with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use gomyk/minilm-student-L6_uniform_distilled with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("gomyk/minilm-student-L6_uniform_distilled") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
L6_uniform_distilled (Distilled)
Lightweight sentence encoder created from sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 via layer pruning + vocabulary pruning + knowledge distillation.
Model Details
| Property | Value |
|---|---|
| Teacher | sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 |
| Architecture | MiniLM-L12 (pruned) |
| Hidden dim | 384 |
| Layers | 6 / 12 |
| Layer indices | [0, 2, 4, 7, 9, 11] |
| Strategy | 6 layers, evenly spaced (general-purpose) |
| Parameters | 106,825,344 |
| Model size (FP32) | 98.1MB |
| Distilled | Yes |
Architecture
==============================================================
TEACHER: MiniLM-L12 โ STUDENT: 6L / 38,775 vocab
==============================================================
TEACHER STUDENT
โโโโโโโโโโโโโโโโโโโโโโโโโโโ โโโโโโโโโโโโโโโโโโโโโโโโโโโ
โโโโโโโโโโโโโโโโโโโโโโโโโโโ โโโโโโโโโโโโโโโโโโโโโโโโโโโ
โ Input Tokens โ โ Input Tokens โ
โโโโโโโโโโโโโโฌโโโโโโโโโโโโโ โโโโโโโโโโโโโโฌโโโโโโโโโโโโโ
โ โ
โโโโโโโโโโโโโโดโโโโโโโโโโโโโ โโโโโโโโโโโโโโดโโโโโโโโโโโโโ
โ Embeddings โ โ Embeddings (pruned) โ
โ vocab: 250,002 โ โ vocab: 38,775 โ
โ dim: 384 โ โ dim: 384 โ
โโโโโโโโโโโโโโฌโโโโโโโโโโโโโ โโโโโโโโโโโโโโฌโโโโโโโโโโโโโ
โ โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโ โโโโโโโโโโโโโโโโโโโโโโโโโโโ
โ Layer 0 โ โโโบ โ Layer 0 โ L0 โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 1 โ โณ โ โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 2 โ โโโบ โ Layer 1 โ L2 โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 3 โ โณ โ โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 4 โ โโโบ โ Layer 2 โ L4 โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 5 โ โณ โ โ
โ โ โ โ โ โ โ โ โ โ โ โโค โ โ
โ Layer 6 โ โณ โ โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 7 โ โโโบ โ Layer 3 โ L7 โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 8 โ โณ โ โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 9 โ โโโบ โ Layer 4 โ L9 โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 10 โ โณ โ โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโค โโโโโโโโโโโโโโโโโโโโโโโโโโโค
โ Layer 11 โ โโโบ โ Layer 5 โ L11 โ
โโโโโโโโโโโโโโฌโโโโโโโโโโโโโ โโโโโโโโโโโโโโฌโโโโโโโโโโโโโ
โ โ
โโโโโโโโโโโโโโดโโโโโโโโโโโโโ โโโโโโโโโโโโโโดโโโโโโโโโโโโโ
โ Mean Pooling โ โ Mean Pooling โ
โ โ 384d embedding โ โ โ 384d embedding โ
โโโโโโโโโโโโโโโโโโโโโโโโโโโ โโโโโโโโโโโโโโโโโโโโโโโโโโโ
Size: 448.0MB (FP32) โ 98.1MB (FP32)
Params: 117,451,392 โ 25,714,176
Reduction: 78.1%
==============================================================
Quick Start
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("L6_uniform_distilled", trust_remote_code=True)
sentences = [
"Hello, how are you?",
"์๋
ํ์ธ์",
"Bonjour, comment allez-vous?",
]
embeddings = model.encode(sentences)
print(embeddings.shape) # (3, 384)
MTEB Evaluation Results
Overall Average: 56.93%
| Task Group | Average |
|---|---|
| Classification | 59.41% |
| Clustering | 38.07% |
| STS | 71.48% |
Classification
| Task | Average | Details |
|---|---|---|
| AmazonCounterfactualClassification | 67.92% | de: 70.3%, en: 70.06%, en-ext: 69.5% |
| Banking77Classification | 79.34% | default: 79.34% |
| ImdbClassification | 58.94% | default: 58.94% |
| MTOPDomainClassification | 76.3% | en: 85.94%, es: 79.58%, th: 78.32% |
| MassiveIntentClassification | 35.04% | en: 66.34%, zh-CN: 62.56%, ja: 62.27% |
| MassiveScenarioClassification | 40.77% | en: 71.95%, zh-CN: 69.09%, ja: 68.11% |
| ToxicConversationsClassification | 60.34% | default: 60.34% |
| TweetSentimentExtractionClassification | 56.67% | default: 56.67% |
Clustering
| Task | Average | Details |
|---|---|---|
| ArXivHierarchicalClusteringP2P | 51.94% | default: 51.94% |
| ArXivHierarchicalClusteringS2S | 48.06% | default: 48.06% |
| BiorxivClusteringP2P.v2 | 30.65% | default: 30.65% |
| MedrxivClusteringP2P.v2 | 31.34% | default: 31.34% |
| MedrxivClusteringS2S.v2 | 28.24% | default: 28.24% |
| StackExchangeClustering.v2 | 48.14% | default: 48.14% |
| StackExchangeClusteringP2P.v2 | 35.9% | default: 35.9% |
| TwentyNewsgroupsClustering.v2 | 30.3% | default: 30.3% |
STS
| Task | Average | Details |
|---|---|---|
| BIOSSES | 60.1% | default: 60.1% |
| SICK-R | 77.0% | default: 77.0% |
| STS12 | 72.99% | default: 72.99% |
| STS13 | 79.03% | default: 79.03% |
| STS14 | 76.54% | default: 76.54% |
| STS15 | 84.27% | default: 84.27% |
| STS17 | 58.61% | en-en: 84.68%, es-es: 78.41%, nl-en: 64.48% |
| STS22.v2 | 51.39% | fr: 70.62%, es-en: 67.53%, zh: 64.99% |
| STSBenchmark | 83.35% | default: 83.35% |
Distillation Impact
| Task | Before | After | Delta |
|---|---|---|---|
| AmazonCounterfactualClassification | 68.27% | 67.92% | -0.35%p |
| ArXivHierarchicalClusteringP2P | 50.12% | 51.94% | +1.82%p |
| ArXivHierarchicalClusteringS2S | 46.66% | 48.06% | +1.4%p |
| Banking77Classification | 73.53% | 79.34% | +5.81%p |
| BiorxivClusteringP2P.v2 | 25.42% | 30.65% | +5.23%p |
| BIOSSES | 57.32% | 60.1% | +2.78%p |
| ImdbClassification | 60.64% | 58.94% | -1.7%p |
| MassiveIntentClassification | 37.62% | 35.04% | -2.58%p |
| MassiveScenarioClassification | 41.45% | 40.77% | -0.68%p |
| MedrxivClusteringP2P.v2 | 28.32% | 31.34% | +3.02%p |
| MedrxivClusteringS2S.v2 | 25.33% | 28.24% | +2.91%p |
| MTOPDomainClassification | 75.11% | 76.3% | +1.19%p |
| SICK-R | 69.91% | 77.0% | +7.09%p |
| StackExchangeClustering.v2 | 44.13% | 48.14% | +4.01%p |
| StackExchangeClusteringP2P.v2 | 33.07% | 35.9% | +2.83%p |
| STS12 | 66.88% | 72.99% | +6.11%p |
| STS13 | 71.42% | 79.03% | +7.61%p |
| STS14 | 68.52% | 76.54% | +8.02%p |
| STS15 | 79.84% | 84.27% | +4.43%p |
| STS17 | 53.52% | 58.61% | +5.09%p |
| STS22.v2 | 40.57% | 51.39% | +10.82%p |
| STSBenchmark | 74.69% | 83.35% | +8.66%p |
| ToxicConversationsClassification | 61.36% | 60.34% | -1.02%p |
| TweetSentimentExtractionClassification | 53.21% | 56.67% | +3.46%p |
| TwentyNewsgroupsClustering.v2 | 22.01% | 30.3% | +8.29%p |
Training
Stage 1: Layer Pruning
- Teacher:
sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2(12 layers, 384d) - Selected layers:
[0, 2, 4, 7, 9, 11](6 layers, evenly spaced (general-purpose)) - Vocabulary pruning applied
Stage 2: Knowledge Distillation
- Method: MSE + Cosine Similarity loss
- Data: MTEB Classification/Clustering/STS task datasets
- Optimizer: AdamW (lr=2e-5, weight_decay=0.01)
- Schedule: Cosine annealing over 3 epochs
Supported Languages (18)
ko, en, ja, zh, es, fr, de, pt, it, ru, ar, hi, th, vi, id, tr, nl, pl
- Downloads last month
- 47