Abhaykoul commited on
Commit
2d676af
·
verified ·
1 Parent(s): df56cdf

Upload model card README for vtx-embed-1M

Browse files
Files changed (1) hide show
  1. README.md +194 -0
README.md ADDED
@@ -0,0 +1,194 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ license: mit
5
+ library_name: tokenizers
6
+ tags:
7
+ - sentence-similarity
8
+ - feature-extraction
9
+ - embeddings
10
+ - rag
11
+ - quantized
12
+ - 4-bit
13
+ - matryoshka
14
+ - ultra-lightweight
15
+ - code-search
16
+ - retrieval
17
+ pipeline_tag: feature-extraction
18
+ metrics:
19
+ - spearman_cosine
20
+ model-index:
21
+ - name: vtx-embed-1M
22
+ results:
23
+ - task:
24
+ type: sts
25
+ name: Semantic Textual Similarity
26
+ dataset:
27
+ name: STSBenchmark
28
+ type: mteb/stsbenchmark-sts
29
+ metrics:
30
+ - type: cosine_spearman
31
+ value: 0.7149
32
+ - task:
33
+ type: sts
34
+ name: Semantic Textual Similarity
35
+ dataset:
36
+ name: SICK-R
37
+ type: mteb/sickr-sts
38
+ metrics:
39
+ - type: cosine_spearman
40
+ value: 0.5841
41
+ - task:
42
+ type: classification
43
+ name: Classification
44
+ dataset:
45
+ name: Banking77Classification
46
+ type: mteb/banking77
47
+ metrics:
48
+ - type: accuracy
49
+ value: 0.6420
50
+ - task:
51
+ type: classification
52
+ name: Classification
53
+ dataset:
54
+ name: AmazonCounterfactualClassification
55
+ type: mteb/amazon_counterfactual_classification
56
+ metrics:
57
+ - type: accuracy
58
+ value: 0.6185
59
+ - task:
60
+ type: clustering
61
+ name: Clustering
62
+ dataset:
63
+ name: TwentyNewsgroupsClustering
64
+ type: mteb/twentynewsgroups-clustering
65
+ metrics:
66
+ - type: v_measure
67
+ value: 0.2512
68
+ - task:
69
+ type: clustering
70
+ name: Clustering
71
+ dataset:
72
+ name: RedditClustering
73
+ type: mteb/reddit-clustering
74
+ metrics:
75
+ - type: v_measure
76
+ value: 0.3140
77
+ ---
78
+
79
+ <div align="center">
80
+
81
+ # 🚀 vtx-embed-1M
82
+
83
+ **Sub-megabyte pure data-free distilled embedding model.**
84
+ Native 4-Bit quantization · 0.57 MB RAM · 1.05M Parameters · 1.14M Tokens/sec CPU Latency
85
+
86
+ [![HuggingFace](https://img.shields.io/badge/🤗%20HuggingFace-VTXAI%2Fvtx--embed--1M-blue)](https://huggingface.co/VTXAI/vtx-embed-1M)
87
+ [![License: MIT](https://img.shields.io/badge/License-MIT-green.svg)](https://opensource.org/licenses/MIT)
88
+ [![Python 3.8+](https://img.shields.io/badge/Python-3.8%2B-blue)](https://python.org)
89
+
90
+ </div>
91
+
92
+ ---
93
+
94
+ ## 📄 Model Details
95
+
96
+ | Property | Value |
97
+ | :--- | :--- |
98
+ | **Model Type** | Native 4-Bit Data-Less Distilled Embedding (LF4) |
99
+ | **Total Parameters** | **1.05M (1,048,576 parameters)** |
100
+ | **Tensor Storage Format** | `lf4` — 4-bit per-block with FP16 scale + zero |
101
+ | **In-RAM Memory** | **0.57 MB (570 KB)** |
102
+ | **On-Disk Size** | **0.57 MB (570 KB)** |
103
+ | **Vocabulary Size** | 16,384 |
104
+ | **Max Sequence Length** | 512 tokens |
105
+ | **Output Dimensions** | 64 *(Matryoshka: also 32)* |
106
+ | **Pooling** | SIF IDF-weighted + PC-1 removal |
107
+ | **Distillation Type** | 100% Data-Free SVD Subspace Reconstruction |
108
+ | **License** | MIT |
109
+
110
+ ---
111
+
112
+ ## 📊 Comparative MTEB & STS Benchmark Results
113
+
114
+ Comparing **vtx-embed-1M** side-by-side with **vtx-embed-7M**, **MiniLM-L6-v2**, and **bge-small-en-v1.5**:
115
+
116
+ ### Semantic Textual Similarity (STS)
117
+
118
+ | Dataset | Metric | **vtx-embed-1M (0.57 MB)** | vtx-embed-7M (4.72 MB) | MiniLM-L6-v2 (90 MB) | bge-small-en-v1.5 (134 MB) |
119
+ | :--- | :---: | :---: | :---: | :---: | :---: |
120
+ | **STSBenchmark** | Spearman ρ | **0.7149** | 0.7918 | 0.8284 | 0.8278 |
121
+ | **SICK-R** | Spearman ρ | **0.5841** | 0.6294 | 0.7572 | 0.7460 |
122
+
123
+ ### Classification
124
+
125
+ | Dataset | Metric | **vtx-embed-1M (0.57 MB)** | vtx-embed-7M (4.72 MB) | MiniLM-L6-v2 (90 MB) | bge-small-en-v1.5 (134 MB) |
126
+ | :--- | :---: | :---: | :---: | :---: | :---: |
127
+ | **Banking77Classification** | Accuracy | **0.6420** | 0.7043 | 0.7451 | 0.7884 |
128
+ | **AmazonCounterfactualClassification** | Accuracy | **0.6185** | 0.6679 | 0.7371 | 0.7279 |
129
+
130
+ ### Clustering
131
+
132
+ | Dataset | Metric | **vtx-embed-1M (0.57 MB)** | vtx-embed-7M (4.72 MB) | MiniLM-L6-v2 (90 MB) | bge-small-en-v1.5 (134 MB) |
133
+ | :--- | :---: | :---: | :---: | :---: | :---: |
134
+ | **TwentyNewsgroupsClustering** | V-Measure | **0.2512** | 0.2936 | 0.3529 | 0.4419 |
135
+ | **RedditClustering** | V-Measure | **0.3140** | 0.3880 | 0.4342 | 0.5376 |
136
+
137
+ ---
138
+
139
+ ## ⚡ Efficiency & Throughput Benchmark
140
+
141
+ *Evaluated directly on CPU execution over 1,158 codebase files (900k+ tokens):*
142
+
143
+ | Model | RAM Size | Throughput | Single File Latency | Tool Routing Accuracy |
144
+ | :--- | :-: | :-: | :-: | :-: |
145
+ | **vtx-embed-1M (64-dim)** | **0.57 MB** | **1,139,267 Tokens/sec** | **0.682 ms** | **100.0% (20/20)** |
146
+ | **vtx-embed-7M (256-dim)** | **4.72 MB** | 878,578 Tokens/sec | 0.841 ms | **100.0% (20/20)** |
147
+ | LiquidAI/LFM2.5-Embedding-350M | ~700 MB | ~137 texts/sec | 7.30 ms | — |
148
+ | sentence-transformers/all-MiniLM-L6-v2 | 90 MB | ~80 texts/sec | 12.4 ms | — |
149
+
150
+ > **vtx-embed-1M is 157× smaller than MiniLM-L6-v2 while maintaining a 100% Agent Tool Search Accuracy.**
151
+
152
+ ---
153
+
154
+ ## 💻 Quickstart
155
+
156
+ ### Installation
157
+
158
+ ```bash
159
+ pip install numpy tokenizers safetensors huggingface_hub
160
+ ```
161
+
162
+ ### Python Inference
163
+
164
+ ```python
165
+ from vortex_embed.src.lf4_v4_5 import VortexEmbedV4_5
166
+
167
+ # Load 1M model from Hugging Face
168
+ model = VortexEmbedV4_5.from_pretrained("VTXAI/vtx-embed-1M")
169
+
170
+ sentences = ["Execute terminal command", "Run bash shell process in background"]
171
+ embeddings = model.encode(sentences, normalize=True) # (2, 64) float32
172
+
173
+ sim = (embeddings[0] * embeddings[1]).sum()
174
+ print(f"Similarity: {sim:.4f}")
175
+ ```
176
+
177
+ ---
178
+
179
+ ## 📜 Citation
180
+
181
+ ```bibtex
182
+ @misc{vtx-embed-1M,
183
+ title = {vtx-embed-1M: Data-Free Matrix Distillation for Sub-Megabyte Embeddings},
184
+ author = {VTXAI},
185
+ year = {2026},
186
+ url = {https://huggingface.co/VTXAI/vtx-embed-1M}
187
+ }
188
+ ```
189
+
190
+ ---
191
+
192
+ ## 📄 License
193
+
194
+ MIT License