Skip to content

Skin cancer detection using ensembling of DenseNet, EfficientNet, and ConvNet

Notifications You must be signed in to change notification settings

pssingh1434131/Skin-Cancer-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Skin-Cancer-Classification

Abstract

Diagnosis of skin cancer in early stage plays vital role in increasing its survival rate. Advancement in Artificial intelligence and deep learning technologies provides the path for easing the detection of the disease using medical image analysis particularly dermoscopic images. This repository studies several deep neural network (DNN) models and their comparative performance. The study was performed on three dataset HAM10000, ISIC 2020 and ISIC 2019. Due to variations in the structure of the datasets utilized, such as differing numbers of class labels and types of skin cancer, a modified dataset was curated. This modified dataset ensured uniformity by providing common class labels across all datasets, thereby enabling consistent training for the models across the entire dataset. The resultant dataset consists of 68,472 dermoscopic images. Transfer learning approach was applied on three modified models: DenseNet201, EfficientNetV2M and ConvNeXtBase. Fine-tuning methods were employed to increase the performance of models on modified dataset, while data augmentation methods were applied to improve class balancing. The model's layered architecture classified the images into seven classes which were aggregated into binary classes. The final classification was done using ensembling of the three models used. The results favored ensembling of the models along with the superior individual model performance of DenseNet201. Results show that ensembling the models achieved overall accuracy of 96.5% in multiclass and 98.7% in binary classification.

Research paper

https://drive.google.com/file/d/19AIlfcN7f6odJ8OESzV7HkNQqrxRKJP0/view?usp=sharing

Dataset:

Model Architecture

image

Flow Diagram

image

Results

Accuracy of individual model on the dataset

image

Performance of the Ensembled Model on the Dataset in Multiclass Classification

image

Performance of the Ensembled Model on the Dataset in Binary Classification

image

About

Skin cancer detection using ensembling of DenseNet, EfficientNet, and ConvNet

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published