sowmya333 commited on
Commit
d396031
·
verified ·
1 Parent(s): 4fafde0

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +14 -14
README.md CHANGED
@@ -9,34 +9,34 @@ tags:
9
  - convnext
10
  - explainable-ai
11
  metrics:
12
- - auc: 0.9362
13
- - accuracy: 0.8681
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 7,058 image dataset.
21
 
22
  | Fold | Accuracy | AUC | Sensitivity | Specificity |
23
  |-------:|-----------:|---------:|--------------:|--------------:|
24
- | 1 | 0.884561 | 0.94365 | 0.912134 | 0.826754 |
25
- | 2 | 0.86898 | 0.936338 | 0.881799 | 0.842105 |
26
- | 3 | 0.875354 | 0.944329 | 0.912134 | 0.798246 |
27
- | 4 | 0.860383 | 0.932138 | 0.876569 | 0.826374 |
28
- | 5 | 0.851169 | 0.924778 | 0.872385 | 0.806593 |
29
 
30
  ### Summary Statistics:
31
- - **Mean AUC:** 0.9362 ± 0.0082
32
- - **Mean Sensitivity:** 0.8910 ± 0.0196
33
- - **Mean Specificity:** 0.8200 ± 0.0175
34
 
35
  ## 🚀 Clinical Application
36
- This model is designed for clinical decision support. The weights hosted here (pytorch_model.bin) correspond to **Fold 3**, which achieved the highest individual AUC of 0.9443.
37
 
38
  ## 🛠 Methodology
39
  - **Backbone:** ConvNeXt-Tiny (Spatial Stream)
40
  - **Texture Analysis:** FFT Magnitude Spectrum (Frequency Stream)
41
- - **Preprocessing:** CLAHE (Contrast Limited Adaptive Histogram Equalization)
42
- - **Loss Function:** Focal Loss (α=1, γ=2) to handle class imbalance.
 
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)