Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,98 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# DermAssist – ResNet50 Ensemble
|
| 2 |
+
|
| 3 |
+
## Model Overview
|
| 4 |
+
|
| 5 |
+
DermAssist is a dermoscopic skin lesion malignancy risk estimation model built using a 3-model ResNet50 ensemble.
|
| 6 |
+
|
| 7 |
+
The system outputs:
|
| 8 |
+
|
| 9 |
+
- Malignancy probability
|
| 10 |
+
- Confidence score
|
| 11 |
+
- Uncertainty estimation (ensemble disagreement)
|
| 12 |
+
|
| 13 |
+
This model is designed for dermoscopic image triage workflows.
|
| 14 |
+
|
| 15 |
+
---
|
| 16 |
+
|
| 17 |
+
## Architecture
|
| 18 |
+
|
| 19 |
+
Backbone:
|
| 20 |
+
ResNet50 (ImageNet pretrained)
|
| 21 |
+
|
| 22 |
+
Classifier Head:
|
| 23 |
+
Linear → ReLU → Dropout (0.5) → Linear → 1 output logit
|
| 24 |
+
|
| 25 |
+
Training Details:
|
| 26 |
+
- Optimizer: AdamW (lr = 1e-4, weight_decay = 1e-4)
|
| 27 |
+
- Loss: BCEWithLogitsLoss (class-weighted)
|
| 28 |
+
- Early Stopping: Validation AUC
|
| 29 |
+
- Full fine-tuning applied
|
| 30 |
+
|
| 31 |
+
Ensemble Strategy:
|
| 32 |
+
Three independently trained models.
|
| 33 |
+
Final probability = mean prediction.
|
| 34 |
+
Uncertainty = standard deviation across models.
|
| 35 |
+
|
| 36 |
+
---
|
| 37 |
+
|
| 38 |
+
## Performance
|
| 39 |
+
|
| 40 |
+
### Internal Validation (HAM10000)
|
| 41 |
+
|
| 42 |
+
| Metric | Value |
|
| 43 |
+
|--------|-------|
|
| 44 |
+
| AUC-ROC | 0.937 |
|
| 45 |
+
| Accuracy | 0.86 |
|
| 46 |
+
| Malignant Recall | > 0.90 |
|
| 47 |
+
| Malignant Precision | ~0.59 |
|
| 48 |
+
| Benign Precision | ~0.96 |
|
| 49 |
+
|
| 50 |
+
---
|
| 51 |
+
|
| 52 |
+
### External Validation (ISIC 2019 – Filtered Classes)
|
| 53 |
+
|
| 54 |
+
| Metric | Value |
|
| 55 |
+
|--------|-------|
|
| 56 |
+
| AUC-ROC | ~0.72 |
|
| 57 |
+
| Accuracy | ~0.57 |
|
| 58 |
+
|
| 59 |
+
Observed performance variation reflects dataset distribution differences.
|
| 60 |
+
|
| 61 |
+
---
|
| 62 |
+
|
| 63 |
+
## Intended Use
|
| 64 |
+
|
| 65 |
+
This model is intended for:
|
| 66 |
+
|
| 67 |
+
- Dermoscopic image triage
|
| 68 |
+
- Educational experimentation
|
| 69 |
+
- Ensemble uncertainty analysis
|
| 70 |
+
- Explainability demonstration
|
| 71 |
+
|
| 72 |
+
This model is not intended for autonomous medical diagnosis.
|
| 73 |
+
|
| 74 |
+
---
|
| 75 |
+
|
| 76 |
+
## Limitations
|
| 77 |
+
|
| 78 |
+
- Trained on dermoscopic images only
|
| 79 |
+
- Binary mapping reduces lesion taxonomy complexity
|
| 80 |
+
- Sensitive to dataset distribution shift
|
| 81 |
+
- Requires calibrated threshold tuning
|
| 82 |
+
|
| 83 |
+
---
|
| 84 |
+
|
| 85 |
+
## Files in Repository
|
| 86 |
+
|
| 87 |
+
- resnet50_model_1.pth
|
| 88 |
+
- resnet50_model_2.pth
|
| 89 |
+
- resnet50_model_3.pth
|
| 90 |
+
|
| 91 |
+
All three models are required for ensemble inference.
|
| 92 |
+
|
| 93 |
+
---
|
| 94 |
+
|
| 95 |
+
## Ethical Considerations
|
| 96 |
+
|
| 97 |
+
AI-based medical systems must be used responsibly.
|
| 98 |
+
This model does not replace dermatologist evaluation.
|