Skin cancer image classification optimization through transfer learning with Tensorflow and InceptionV3

Tianyu Cao

doi:10.1117/12.2672699

1 March 2023 Skin cancer image classification optimization through transfer learning with Tensorflow and InceptionV3

Tianyu Cao

Proceedings Volume 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022); 125961S (2023) https://doi.org/10.1117/12.2672699
Event: International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 2022, Changsha, China

Abstract

Skin cancer is a common illness that claims thousands of lives annually in the United States alone. Accurately identifying malignant tumors is crucial to survival but can be challenging as the visual distinctions between benign and life-threatening tumors are minimal. The purpose of this project is to explore deep learning algorithms that can be trained to systematically classify skin cancer images, create a program to execute the algorithm, and exemplify an optimization process for the program that can serve as a reference for future works. The project will adopt a transfer learning algorithm based on previous studies on the subject and select a pre-trained model given practical restraints. Then a program will be coded in Python to retrieve datasets, process images, train the model, and evaluate accuracy. Finally, the algorithm will be optimized by tuning model parameters and training restraints. The experiments revealed that the algorithm was able to perform the task with an accuracy of around 70%. Model parameters such as optimizer choice and learning rate and training restraints such as batch size and epoch count have significant impacts on the training results and require precise values for maxima accuracy and minimal overfitting.

Citation Download Citation

Tianyu Cao "Skin cancer image classification optimization through transfer learning with Tensorflow and InceptionV3", Proc. SPIE 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 125961S (1 March 2023); https://doi.org/10.1117/12.2672699

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available