The Effectiveness of Data Augmentation for Melanoma Skin Cancer Prediction Using Convolutional Neural Networks

Kin, Wai Lee and Ka, Renee Yin Chin (2020) The Effectiveness of Data Augmentation for Melanoma Skin Cancer Prediction Using Convolutional Neural Networks. In: 2020 IEEE 2nd International Conference on Artificial Intelligence in Engineering and Technology (IICAIET), 20 November 2020, Kota Kinabalu, Malaysia.

[img] Text
The Effectiveness of Data Augmentation for Melanoma Skin Cancer Prediction Using Convolutional Neural Networks abstract.pdf

Download (38kB)
[img] Text
The Effectiveness of Data Augmentation for Melanoma Skin Cancer Prediction Using Convolutional Neural Networks.pdf
Restricted to Registered users only

Download (371kB)

Abstract

Melanoma skin cancer has been a serious threat due to its high fatality. For this reason, early detection and treatments are given more attention as countermeasures. In recent years, skin cancer detection has been utilizing artificial intelligence techniques, specifically deep convolutional neural network. However, the performance of the convolutional neural network is highly vulnerable to different data constraints, such as the quality and quantity of the data. Therefore, this study explores the synthetization of training data using different data augmentation methods. The work presented in this paper utilizes four different categories of data augmentation methods, which include geometrical transformation, noise addition, color transformation, and image mix. Multiple layers data augmentation approach is also explored. Dataset expansion strategies and optimized dataset expansion scale are determined to improve the performance of the skin cancer classification. The core findings in our study revealed that single-layer augmentation has better performance than multiple layers augmentation approaches, where region of interest (ROI) image mix has the highest effectiveness compared to other methods. In addition, the best dataset expansion strategy is random ROI image mix. Finally, the optimized dataset expansion is determined at 300%, which yielded the best overall test accuracy at 82.9%, 4.6% improvement compared to unprocessed raw dataset.

Item Type: Conference or Workshop Item (Paper)
Keyword: Skin cancer classification , Convolutional Neural network , Data augmentation
Subjects: ?? QA75 ??
R Medicine > RA Public aspects of medicine
Department: FACULTY > Faculty of Engineering
Depositing User: NORAINI LABUK -
Date Deposited: 29 Jul 2021 09:45
Last Modified: 29 Jul 2021 09:45
URI: https://eprints.ums.edu.my/id/eprint/30132

Actions (login required)

View Item View Item