medsiglip-448-surgwound-v2
This model is a fine-tuned version of google/medsiglip-448 on the None dataset. It achieves the following results on the evaluation set:
- Loss: 1.0206
- Auc Healing Status: 0.8558
- Auc Erythema: 0.7087
- Auc Edema: 0.6892
- Auc Infection Risk: 0.8012
- Auc Urgency: 0.7679
- Auc Exudate: 0.9333
- Roc Auc Macro: 0.7927
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: 8e-05
- train_batch_size: 4
- eval_batch_size: 8
- seed: 42
- gradient_accumulation_steps: 4
- total_train_batch_size: 16
- optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: cosine
- lr_scheduler_warmup_steps: 15
- num_epochs: 10
- mixed_precision_training: Native AMP
Training results
| Training Loss | Epoch | Step | Validation Loss | Auc Healing Status | Auc Erythema | Auc Edema | Auc Infection Risk | Auc Urgency | Auc Exudate | Roc Auc Macro |
|---|---|---|---|---|---|---|---|---|---|---|
| 2.8822 | 1.0 | 30 | 1.1856 | 0.7341 | 0.6395 | 0.5863 | 0.6136 | 0.5082 | 0.7657 | 0.6412 |
| 3.8863 | 2.0 | 60 | 1.1678 | 0.7341 | 0.6498 | 0.6412 | 0.7512 | 0.6085 | 0.8515 | 0.7061 |
| 2.8724 | 3.0 | 90 | 1.1094 | 0.7743 | 0.6539 | 0.6353 | 0.7926 | 0.6662 | 0.8889 | 0.7352 |
| 3.6608 | 4.0 | 120 | 1.0557 | 0.8188 | 0.6674 | 0.6471 | 0.7877 | 0.7184 | 0.9212 | 0.7601 |
| 2.8843 | 5.0 | 150 | 0.9983 | 0.8554 | 0.6932 | 0.6784 | 0.7975 | 0.7473 | 0.9495 | 0.7869 |
| 2.5912 | 6.0 | 180 | 0.9958 | 0.8554 | 0.6952 | 0.6765 | 0.7877 | 0.7617 | 0.9394 | 0.7860 |
| 1.6655 | 7.0 | 210 | 1.0227 | 0.8519 | 0.7076 | 0.6882 | 0.8025 | 0.7734 | 0.9404 | 0.7940 |
| 1.7491 | 8.0 | 240 | 1.0211 | 0.8545 | 0.7066 | 0.6863 | 0.8012 | 0.7720 | 0.9354 | 0.7927 |
| 2.3213 | 9.0 | 270 | 1.0194 | 0.8554 | 0.7087 | 0.6882 | 0.8019 | 0.7692 | 0.9333 | 0.7928 |
| 2.8695 | 10.0 | 300 | 1.0206 | 0.8558 | 0.7087 | 0.6892 | 0.8012 | 0.7679 | 0.9333 | 0.7927 |
Framework versions
- Transformers 5.2.0
- Pytorch 2.9.0+cu126
- Datasets 4.0.0
- Tokenizers 0.22.2
- Downloads last month
- 1
Model tree for tyb343/medsiglip-448-surgwound-v2
Base model
google/medsiglip-448