The primary goal of this project is to classify dermoscopic images of skin lesions into one of eight diagnostic categories:
- Melanoma
- Melanocytic Nevus
- Basal Cell Carcinoma
- Actinic Keratosis
- Benign Keratosis
- Dermatofibroma
- Vascular Lesion
- Squamous Cell Carcinoma
Some images in the dataset are provided with corresponding masks. To utilize this information, I trained a U-Net model from scratch to segment the lesions. The training was performed on a dataset comprising 1800 images. The U-Net model achieved a Dice score of 0.87, indicating a high level of accuracy in segmenting the lesions.
For the classification task, I applied several classical machine learning models, including Random Forest (RF), Support Vector Machine (SVM), and XGBoost. These models were used to classify the segmented lesions into the eight diagnostic classes. The combined approach yielded a weighted accuracy of 0.5.
- Segmentation: U-Net model achieved a Dice score of 0.87.
- Classification: Classical ML models achieved a weighted accuracy of 0.5.
The combination of U-Net for lesion segmentation and classical machine learning models for classification provides a solid foundation for improving the accuracy of skin lesion diagnosis. Future work will involve enhancing the classification accuracy by experimenting with pretrained model like ResNet.