Lung Images Segmentation and Classification Based on Deep Learning: A New Automated CNN Approach

Wessam M. Salama,Moustafa H. Aly,Azza M. Elbagoury
DOI: https://doi.org/10.1088/1742-6596/2128/1/012011
2021-12-01
Journal of Physics: Conference Series
Abstract:Abstract Lung cancer became a significant health problem worldwide over the past decades. This paper introduces a new generalized framework for lung cancer detection where many different strategies are explored for the classification. The ResNet50 model is applied to classify CT lung images into benign or malignant. Also, the U-Net, which is one of the most used architectures in deep learning for image segmentation, is employed to segment CT images before classification to increase system performance. Moreover, Image Size Dependent Normalization Technique (ISDNT) and Wiener filter are utilized as the preprocessing phase to enhance the images and suppress the noise. Our proposed framework which comprises preprocessing, segmentation and classification phases, is applied on two databases: Lung Nodule Analysis 2016 (Luna 16) and National Lung Screening Trial (NLST). Data augmentation technique is applied to solve the problem of lung CT images deficiency, and consequently, the overfitting of deep models will be avoided. The classification results show that the preprocessing for the CT lung image as the input for ResNet50-U-Net hybrid model achieves the best performance. The proposed model achieves 98.98% accuracy ( ACC ), 98.65% area under the ROC curve ( AUC ), 98.99% sensitivity ( Se ), 98.43% precision ( Pr ), 98.86% F1 - score and 1.9876 s computational time.
What problem does this paper attempt to address?