Optimization Convolutional Neural Network for Automatic Skin Lesion Diagnosis Using a Genetic Algorithm

Abstract

Examining and predicting skin cancer from skin lesion images is challenging due to the complexity of the images. Early detection and treatment of skin lesion disease can prevent mortality as it can be curable. Computer-aided diagnosis (CAD) provides a second opinion for dermatologists as they can classify the type of skin lesion with high accuracy due to their ability to show various clinical identification features locally and globally. Convolutional neural networks (CNNs) have significantly improved the performance of CAD systems for medical image segmentation and classifications. However, tuning CNNs are challenging since the search space of all possible hyperparameter configurations is substantially vast. In this paper, we adopt a genetic algorithm to automatically configure a CNN model for an accurate, reliable, and robust automated skin lesion classification for early skin lesion diagnosis. The optimized CNN model uses four public datasets to train and be able to detect abnormalities based on skin lesion features in different orientations. The model achieves the best scores for each of the DICE coefficients, precision measure, and F-score. These scores compare better than other existing methods. Considering the success of this optimized model, it could be a valuable method to implement in clinical settings.

Keywords

• skin lesion classification
• CNN
• genetic algorithm
• medical image analysis