Affiliation:
1. College of Computer Science and Engineering Changchun University of Technology Changchun China
2. Department of Dermatology The Second Hospital of Jilin University Changchun China
Abstract
AbstractMalignant skin lesions pose a great threat to patients' health, and the use of computer algorithms for automatic skin medical image classification can effectively improve the efficiency of clinical diagnosis. However, the existing methods for skin classification have complex models and are greatly affected by the imbalance of the dataset. In this work, we propose a two‐stage framework called G‐DMN, it uses CycleGAN to expand the dataset and Dense‐MobileNetV2 (DMN) to achieve the automatic classification of skin lesion images. In the first stage, we use CycleGAN for data augmentation and propose a new image pairing strategy for training. Image pairs are formed from majority class images and minority class images, generators are trained for majority to minority class image conversion, and then minority class images are generated to balance the dataset. In the second stage, we propose a lightweight model called DMN by improving MobileNetV2, it enhances feature reuse by increasing the width of the network and allows the network to focus on focal areas from different scales. The original training set combined with the generated images is used to train DMN for skin lesion classification. We tested the proposed model on the HAM10000 dataset, and the G‐DMN achieved 87.07% classification accuracy, 80.13% precision, 75.28% sensitivity, 96.19% specificity, 77.26% F1‐Score and 0.952 AUC, which has a good classification effect, while the number of parameters of the model is only 5.33 M, which is much lower than other classical classification models. We demonstrate that the proposed method is lighter and more effective than classical classification methods, achieving significant performance improvements.
Subject
Electrical and Electronic Engineering,Computer Vision and Pattern Recognition,Software,Electronic, Optical and Magnetic Materials
Cited by
3 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献