Skin disease classification using two path deep transfer learning models

Ram Chandna,Aarav Bansal,Aryan Kumar,Shrestha Hardia,Omogbolaga Daramola,Anirudh Sahu,Kritika Verma,Karan Dhingra,Saloni Verma
DOI: https://doi.org/10.60087/jklst.v3.n4.p169
2024-12-25
Abstract:Skin diseases are among the most common diseases that affect millions of lives per year, yet diagnosing these has several complexities even for trained dermatologists due to overlapping symptoms and features in several diseases. A myriad of deep learning models have been proposed as a solution for diagnosing but a clinically useful model with high accuracy multi disease classification and lower computational complexity is still unavailable. This study focuses on comparing different image pre-processing techniques, transfer learning models and ensemble learning techniques to build a computationally cheap model for 8-class identification of skin diseases. A two path model with EfficientNet and MobileNetV2 transfer learning models as base feature extractors and a final model that stacks the two model results and classifies the images into one of the eight classes is used. The model is trained and tested on ISIC-2019 dataset for 8 class image classification that involves the three types of skin cancers as well. The dataset has an extreme class imbalance problem which leads to favored prediction of the classes with more image, for this first we run simple image augmentation. Secondly, two distinctly processed images are created from each initial image. The two path model takes the two images, gives each to a base model and combines the two outputs, enabling the classifier to consider different features that become prominent due to dissimilar preprocessing techniques. The model is tested with new images on multiple standard metrics to get a final overview of its performance, it gives a diagnosing accuracy of 70% which is close to some state of the art models that consume higher computational power. The classification results imply that by further improving the data gathering and preprocessing techniques along with exploring other base transfer learning models the results of the final model can be reliable while maintaining a low computational requirement, making the diagnoses accessible. This also highlights that such two path algorithms that employ simpler models could be useful for multi class classification tasks where differently processed images might be required to extract features of distinct diseases.
What problem does this paper attempt to address?