Image classification and auxiliary diagnosis system for hyperpigmented skin diseases based on deep learning

BACKGROUND AND AIM: Melasma (ML), naevus fusco-caeruleus zygomaticus (NZ), freckles (FC), cafe-au-lait spots (CS), nevus of ota (NO), and lentigo simplex (LS), are common skin diseases causing hyperpigmentation. Deep learning algorithms learn the inherent laws and representation levels of sample data and can analyze the internal details of the image and classify it objectively to be used for image diagnosis. However, deep learning algorithms that can assist clinicians in diagnosing skin hyperpigmentation conditions are lacking. METHODS: The optimal deep-learning image recognition algorithm was explored for the auxiliary diagnosis of hyperpigmented skin disease. Pretrained models, such as VGG-19, GoogLeNet, InceptionV3, ResNet50V2, ResNet101V2, ResNet152V2, InceptionResNetV2, DesseNet201, MobileNet, and NASNetMobile were used to classify images of six common hyperpigmented skin diseases. The best deep learning algorithm for developing an online clinical diagnosis system was selected by using accuracy and area under curve (AUC) as evaluation indicators. RESULTS: In this research, the parameters of the above-mentioned ten deep learning algorithms were 18333510, 5979702, 21815078, 23577094, 42638854, 58343942, 54345958, 18333510, 3235014, and 4276058, respectively, and their training time was 380, 162, 199, 188, 315, 511, 471, 697, 101, and 144 min respectively. The respective accuracies of the training set were 85.94%, 99.72%, 99.61%, 99.52%, 99.52%, 98.84%, 99.61%, 99.13%, 99.52%, and 99.61%. The accuracy rates of the test set data were 73.28%, 57.40%, 70.04%, 71.48%, 68.23%, 71.11%, 71.84%, 73.28%, 70.39%, and 43.68%, respectively. Finally, the areas of AUC curves were 0.93, 0.86, 0.93, 0.91, 0.91, 0.92, 0.93, 0.92, 0.93, and 0.82, respectively. CONCLUSIONS: The experimental parameters, training time, accuracy, and AUC of the above models suggest that MobileNet provides a good clinical application prospect in the auxiliary diagnosis of hyperpigmented skin.