selfconstruct3d commited on
Commit
db446b7
·
verified ·
1 Parent(s): ecf15f8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +37 -8
README.md CHANGED
@@ -5,6 +5,11 @@ tags:
5
  - mpnet
6
  - classification
7
  - fine-tuned
 
 
 
 
 
8
  ---
9
 
10
  # AttackGroup-MPNET - Model Card for MPNet Cybersecurity Classifier
@@ -190,14 +195,38 @@ To be anounced...
190
 
191
  ## Evaluation Results
192
 
193
- | Model | Embedding Variability | Accuracy |
194
- |-----------------------------------------------|-----------------------|----------|
195
- | Original MPNet | 0.085554 | 0.9964 |
196
- | MLM Fine-tuned MPNet | 0.034983 | 0.6536 |
197
- | ** AttackGroup-MPNET ** | 0.236336 | 0.9600 |
198
- | SecBERT | 0.591303 | 0.9886 |
199
- | ATTACK-BERT | 0.096108 | 0.9678 |
200
- | SecureBERT | 0.007100 | 0.4931 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
201
 
202
 
203
  ### Single Prediction Example
 
5
  - mpnet
6
  - classification
7
  - fine-tuned
8
+ license: creativeml-openrail-m
9
+ language:
10
+ - en
11
+ base_model:
12
+ - sentence-transformers/all-mpnet-base-v2
13
  ---
14
 
15
  # AttackGroup-MPNET - Model Card for MPNet Cybersecurity Classifier
 
195
 
196
  ## Evaluation Results
197
 
198
+ | Model | Accuracy | F1 Macro | F1 Weighted | Embedding Variability |
199
+ |-----------------------|----------|----------|-------------|-----------------------|
200
+ | **AttackGroup-MPNET** | **0.85** | **0.759**| **0.847** | 0.234 |
201
+ | GTE Large | 0.66 | 0.571 | 0.667 | 0.266 |
202
+ | E5 Large v2 | 0.64 | 0.541 | 0.650 | 0.355 |
203
+ | Original MPNet | 0.63 | 0.534 | 0.619 | 0.092 |
204
+ | BGE Large | 0.53 | 0.418 | 0.519 | 0.366 |
205
+ | SupSimCSE | 0.50 | 0.373 | 0.479 | 0.227 |
206
+ | MLM Fine-tuned MPNet | 0.44 | 0.272 | 0.411 | 0.125 |
207
+ | SecBERT | 0.41 | 0.315 | 0.410 | 0.591 |
208
+ | SecureBERT_Plus | 0.36 | 0.252 | 0.349 | 0.267 |
209
+ | CySecBERT | 0.34 | 0.235 | 0.323 | 0.229 |
210
+ | ATTACK-BERT | 0.33 | 0.240 | 0.316 | 0.096 |
211
+ | Secure_BERT | 0.00 | 0.000 | 0.000 | 0.007 |
212
+ | CyBERT | 0.00 | 0.000 | 0.000 | 0.015 |
213
+
214
+
215
+ | Model | Similarity Search Recall@5 | Few-shot Accuracy | In-dist Similarity | OOD Similarity | Robustness Similarity |
216
+ |----------------------|----------------------------|-------------------|--------------------|----------------|-----------------------|
217
+ | **AttackGroup-MPNET**| **0.934** | **0.857** | 0.235 | 0.017 | 0.948 |
218
+ | Original MPNet | 0.786 | 0.643 | 0.217 | -0.004 | 0.941 |
219
+ | E5 Large v2 | 0.778 | 0.679 | 0.727 | 0.013 | 0.977 |
220
+ | GTE Large | 0.746 | 0.786 | 0.845 | 0.002 | 0.984 |
221
+ | BGE Large | 0.632 | 0.750 | 0.533 | -0.006 | 0.970 |
222
+ | SupSimCSE | 0.616 | 0.571 | 0.683 | -0.015 | 0.978 |
223
+ | SecBERT | 0.468 | 0.429 | 0.586 | -0.001 | 0.970 |
224
+ | CyBERT | 0.452 | 0.250 | 1.000 | -0.001 | 1.000 |
225
+ | ATTACK-BERT | 0.362 | 0.571 | 0.157 | -0.005 | 0.950 |
226
+ | CySecBERT | 0.424 | 0.500 | 0.734 | -0.015 | 0.954 |
227
+ | Secure_BERT | 0.424 | 0.250 | 0.990 | 0.050 | 0.998 |
228
+ | SecureBERT_Plus | 0.406 | 0.464 | 0.981 | 0.040 | 0.998 |
229
+
230
 
231
 
232
  ### Single Prediction Example