Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -9,34 +9,34 @@ tags:
|
|
| 9 |
- convnext
|
| 10 |
- explainable-ai
|
| 11 |
metrics:
|
| 12 |
-
- auc: 0.
|
| 13 |
-
- accuracy: 0.
|
| 14 |
---
|
| 15 |
# Thyroid Nodule Malignancy Detector (5-Fold Validated)
|
| 16 |
|
| 17 |
This model uses a **Dual-Stream Spatial-Frequency Fusion architecture** (ConvNeXt-Tiny + FFT Magnitude Spectrum) to classify thyroid nodules in ultrasound images.
|
| 18 |
|
| 19 |
## 📊 5-Fold Cross-Validation Performance
|
| 20 |
-
The model was evaluated using a stratified 5-fold cross-validation on the consolidated
|
| 21 |
|
| 22 |
| Fold | Accuracy | AUC | Sensitivity | Specificity |
|
| 23 |
|-------:|-----------:|---------:|--------------:|--------------:|
|
| 24 |
-
|
|
| 25 |
-
|
|
| 26 |
-
|
|
| 27 |
-
|
|
| 28 |
-
|
|
| 29 |
|
| 30 |
### Summary Statistics:
|
| 31 |
-
- **Mean AUC:** 0.
|
| 32 |
-
- **Mean Sensitivity:** 0.
|
| 33 |
-
- **Mean Specificity:** 0.
|
| 34 |
|
| 35 |
## 🚀 Clinical Application
|
| 36 |
-
|
| 37 |
|
| 38 |
## 🛠 Methodology
|
| 39 |
- **Backbone:** ConvNeXt-Tiny (Spatial Stream)
|
| 40 |
- **Texture Analysis:** FFT Magnitude Spectrum (Frequency Stream)
|
| 41 |
-
- **Preprocessing:** CLAHE
|
| 42 |
-
- **Loss Function:** Focal Loss (α=1, γ=2)
|
|
|
|
| 9 |
- convnext
|
| 10 |
- explainable-ai
|
| 11 |
metrics:
|
| 12 |
+
- auc: 0.9414
|
| 13 |
+
- accuracy: 0.8747
|
| 14 |
---
|
| 15 |
# Thyroid Nodule Malignancy Detector (5-Fold Validated)
|
| 16 |
|
| 17 |
This model uses a **Dual-Stream Spatial-Frequency Fusion architecture** (ConvNeXt-Tiny + FFT Magnitude Spectrum) to classify thyroid nodules in ultrasound images.
|
| 18 |
|
| 19 |
## 📊 5-Fold Cross-Validation Performance
|
| 20 |
+
The model was evaluated using a stratified 5-fold cross-validation on the consolidated dataset.
|
| 21 |
|
| 22 |
| Fold | Accuracy | AUC | Sensitivity | Specificity |
|
| 23 |
|-------:|-----------:|---------:|--------------:|--------------:|
|
| 24 |
+
| 0 | 0.883144 | 0.9435 | 0.92364 | 0.798246 |
|
| 25 |
+
| 1 | 0.876062 | 0.941074 | 0.898536 | 0.828947 |
|
| 26 |
+
| 2 | 0.883853 | 0.943223 | 0.927824 | 0.791667 |
|
| 27 |
+
| 3 | 0.866761 | 0.938833 | 0.906904 | 0.782418 |
|
| 28 |
+
| 4 | 0.863926 | 0.940572 | 0.881799 | 0.826374 |
|
| 29 |
|
| 30 |
### Summary Statistics:
|
| 31 |
+
- **Mean AUC:** 0.9414 ± 0.0019
|
| 32 |
+
- **Mean Sensitivity:** 0.9077 ± 0.0188
|
| 33 |
+
- **Mean Specificity:** 0.8055 ± 0.0210
|
| 34 |
|
| 35 |
## 🚀 Clinical Application
|
| 36 |
+
The weights hosted here (pytorch_model.bin) correspond to **Fold 0**, which achieved the highest individual AUC of 0.9435.
|
| 37 |
|
| 38 |
## 🛠 Methodology
|
| 39 |
- **Backbone:** ConvNeXt-Tiny (Spatial Stream)
|
| 40 |
- **Texture Analysis:** FFT Magnitude Spectrum (Frequency Stream)
|
| 41 |
+
- **Preprocessing:** CLAHE
|
| 42 |
+
- **Loss Function:** Focal Loss (α=1, γ=2)
|