Task COBRA test set | Sub-class | Balanced accuracy (%) | Sensitivity (%) | Specificity (%) | ||||||
---|---|---|---|---|---|---|---|---|---|---|
Pretrained | Finetuned | Ablation study | Pretrained | Finetuned | Ablation study | Pretrained | Finetuned | Ablation study | ||
2 Classes | 0 - No Tumor | 74.4 | 84.9 | 55.3 | 55.1 | 82.2 | 76.1 | 93.6 | 87.5 | 34.7 |
1 - Tumor | 93.6 | 87.5 | 34.7 | 55.1 | 82.2 | 76.1 | ||||
3 Classes | 0 - No tumor | 65.2 | 70.5 | 51.3 | 57.1 | 85.0 | 94.8 | 93.7 | 83.8 | 12.8 |
1 - Low risk | 54.5 | 51.4 | 10.7 | 73.4 | 89.1 | 93.4 | ||||
2 - High Risk | 84.0 | 75.2 | 2.3 | 77.8 | 85.0 | 99.2 |