Eklavya16 commited on
Commit
70fd054
·
verified ·
1 Parent(s): c6eecda

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +98 -3
README.md CHANGED
@@ -1,3 +1,98 @@
1
- ---
2
- license: apache-2.0
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.