CLASSIFICATION OF SKIN LESION USING CONVOLUTIONAL NEURAL NETWORK: FINDING THE OPTIMAL EPOCH, HIDDEN LAYERS, AND HIDDEN UNITS

Main Article Content

TAEKWON KANG
ERIK LEE

Abstract

The prevalence of AI usage has exploded in the last decade. One aspect, image recognition software, has been heavily used in different fields of study, ranging from physics all the way to computational biology. In this study, the efficacy of image recognition software from TensorFlow and its parameters were tested and explored. After the specific parameters were divulged, a convolutional neural network (CNN) was used to classify images of skin lesions into seven different categories. The deep neural network implemented in TensorFlow achieved the average accuracy of 38%, which was significantly higher than a chance level of 14%. It was found that one hidden layer with 1000 hidden units at 15 epochs produced the highest performance compared to other simulation parameters.

Keywords:
Artificial Intelligence, convolutional neural network, deep neural network, skin cancer

Article Details

How to Cite
KANG, T., & LEE, E. (2021). CLASSIFICATION OF SKIN LESION USING CONVOLUTIONAL NEURAL NETWORK: FINDING THE OPTIMAL EPOCH, HIDDEN LAYERS, AND HIDDEN UNITS. Journal of Medicine and Health Research, 6(2), 35-41. Retrieved from https://www.ikprress.org/index.php/JOMAHR/article/view/7034
Section
Original Research Article

References

Esteva A, Kuprel B, Novoa RA, Ko J, Swetter SM, Blau HM, Thrun S. Dermatologist-level classification of skin cancer with deep neural networks. Nature. 2017;542(7639):115– 118.

Bowen disease - NORD (national organization for rare disorders). Rarediseases.org. 2015 Feb 11 [accessed 2021 Feb 28].
Available:https://rarediseases.org/rare-diseases/bowen-disease/

Basal cell carcinoma. Mayoclinic.org. [accessed 2021 Feb 28].
Available:https://www.mayoclinic.org/diseases-conditions/basal-cell-carcinoma/symptoms-causes/syc-2 0354187

Seborrheic keratosis. Mayoclinic.org. [accessed 2021 Feb 28].
Available:https://www.mayoclinic.org/diseases-conditions/seborrheic-keratosis/symptoms-causes/syc-20 353878

Dermatofibroma. Webmd.com. [accessed 2021 Feb 28].
Available:https://www.webmd.com/skin-problems-and-treatments/picture-of-dermatofibroma

Melanocytic naevus. Dermnetnz.org. [accessed 2021 Feb 28]. Available:https://www.dermnetnz.org/topics/melanocytic-naevus/

Vascular lesions: Ectasias, tumors, and malformations. In: Yamada’s Handbook of Gastroenterology. Oxford: John Wiley & Sons. 2013;471–478.

Melanoma. Mayoclinic.org. [accessed 2021 Feb 28].
Available:https://www.mayoclinic.org/diseases-conditions/melanoma/symptoms-causes/syc-20374884

Cancer Facts & Figures 2016. Cancer.org. [accessed 2021 Jan 31]. Available:https://www.cancer.org/research/cancer-facts-statistics/all-cancer-facts-figures/cancer-facts-fig ures-2016.html

Scott Mader K. Skin Cancer MNIST: HAM10000. [Accessed 2021 Feb 14].