Non-small cell lung cancer (NSCLC) classification using convolutional neural network (CNN)

Abstract

Small Cell Lung Cancer (SCLC) and Non-Small Cell Lung Cancer (NSCLC) are types of Lung cancer. SCLC usually occurs in patients who have a history of heavy smoking and spreads more rapidly than NSCLC. However, about 80−85% of all lung cancer cases are NSCLC types that mostly attack men and women. This study aimed to classify NSCLC into squamous cell carcinoma, adenocarcinoma, large cell carcinoma, and normal lung and to compare the architecture VGG19 and ResNet50 for classification NSCLC. 1000 images data for each class was used in this study from CT Scan images. Three processes were used for classification processes, such as preprocessing, classification, and validation. The resizing and grayscale process was conducted for preprocessing step to ensure all input images are uniform. The significant result was achieved in the ResNet50 architecture, with an accuracy of 98.35% on testing data, 99.87% on training data, and 96% on validation data. Meanwhile, the best performance on the validation data in the normal class was that the results of precision, sensitivity, F1-score, and specificity were 100%, 100%, 100%, and 100%, respectively.