QA-DeBERTa-v3-large-binary-2
This model is a fine-tuned version of microsoft/deberta-v3-large on the saiteki-kai/Beavertails-it dataset. It achieves the following results on the evaluation set:
- Loss: 0.3122
- Accuracy: 0.8623
- Unsafe Precision: 0.8712
- Unsafe Recall: 0.8831
- Unsafe F1: 0.8771
- Unsafe Fpr: 0.1638
- Unsafe Aucpr: 0.9560
- Safe Precision: 0.8508
- Safe Recall: 0.8362
- Safe F1: 0.8435
- Safe Fpr: 0.1169
- Safe Aucpr: 0.9221
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 5e-06
- train_batch_size: 64
- eval_batch_size: 128
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- lr_scheduler_warmup_steps: 1000
- num_epochs: 10
Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Unsafe Precision | Unsafe Recall | Unsafe F1 | Unsafe Fpr | Unsafe Aucpr | Safe Precision | Safe Recall | Safe F1 | Safe Fpr | Safe Aucpr |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 0.3259 | 1.0 | 8454 | 0.3147 | 0.8613 | 0.8933 | 0.8527 | 0.8725 | 0.1278 | 0.9538 | 0.8251 | 0.8722 | 0.8480 | 0.1473 | 0.9174 |
| 0.2732 | 2.0 | 16908 | 0.3122 | 0.8623 | 0.8712 | 0.8831 | 0.8772 | 0.1637 | 0.9560 | 0.8508 | 0.8363 | 0.8435 | 0.1169 | 0.9221 |
| 0.2762 | 3.0 | 25362 | 0.3147 | 0.8623 | 0.8855 | 0.8643 | 0.8748 | 0.1402 | 0.9561 | 0.8348 | 0.8598 | 0.8471 | 0.1357 | 0.9235 |
| 0.268 | 4.0 | 33816 | 0.3376 | 0.8601 | 0.8726 | 0.8766 | 0.8746 | 0.1606 | 0.9551 | 0.8443 | 0.8394 | 0.8418 | 0.1234 | 0.9220 |
Framework versions
- Transformers 4.57.1
- Pytorch 2.7.1+cu118
- Datasets 4.4.1
- Tokenizers 0.22.1
- Downloads last month
- 4
Model tree for saiteki-kai/QA-DeBERTa-v3-large-binary-2
Base model
microsoft/deberta-v3-largeEvaluation results
- Accuracy on saiteki-kai/Beavertails-itself-reported0.862