A Low-Cost High-Performance Data Augmentation for Deep Learning-Based Skin Lesion Classification

Shuwei Shen; Mengjuan Xu; Fan Zhang; Pengfei Shao; Honghong Liu; Liang Xu; Chi Zhang; Peng Liu; Peng Yao; Ronald X Xu

doi:10.34133/2022/9765307

A Low-Cost High-Performance Data Augmentation for Deep Learning-Based Skin Lesion Classification

BME Front. 2022 Apr 26:2022:9765307. doi: 10.34133/2022/9765307. eCollection 2022.

Authors

Shuwei Shen^{1

2}, Mengjuan Xu³, Fan Zhang³, Pengfei Shao³, Honghong Liu¹, Liang Xu³, Chi Zhang¹, Peng Liu³, Peng Yao³, Ronald X Xu^{2

3}

Affiliations

¹ First Affiliated Hospital, University of Science and Technology of China, Hefei 230031, China.
² Suzhou Institute for Advanced Research, University of Science and Technology of China, Suzhou 215000, China.
³ Department of Precision Machinery and Precision Instrumentation, University of Science and Technology of China, Hefei 230026, China.

Abstract

Objective and Impact Statement. There is a need to develop high-performance and low-cost data augmentation strategies for intelligent skin cancer screening devices that can be deployed in rural or underdeveloped communities. The proposed strategy can not only improve the classification performance of skin lesions but also highlight the potential regions of interest for clinicians' attention. This strategy can also be implemented in a broad range of clinical disciplines for early screening and automatic diagnosis of many other diseases in low resource settings. Methods. We propose a high-performance data augmentation strategy of search space 10¹, which can be combined with any model through a plug-and-play mode and search for the best argumentation method for a medical database with low resource cost. Results. With EfficientNets as a baseline, the best BACC of HAM10000 is 0.853, outperforming the other published models of "single-model and no-external-database" for ISIC 2018 Lesion Diagnosis Challenge (Task 3). The best average AUC performance on ISIC 2017 achieves 0.909 (±0.015), exceeding most of the ensembling models and those using external datasets. Performance on Derm7pt archives the best BACC of 0.735 (±0.018) ahead of all other related studies. Moreover, the model-based heatmaps generated by Grad-CAM++ verify the accurate selection of lesion features in model judgment, further proving the scientific rationality of model-based diagnosis. Conclusion. The proposed data augmentation strategy greatly reduces the computational cost for clinically intelligent diagnosis of skin lesions. It may also facilitate further research in low-cost, portable, and AI-based mobile devices for skin cancer screening and therapeutic guidance.