Framework for COVID-19 segmentation and classification based on deep learning of computed tomography lung images

ABSTRACT

Corona Virus Disease 2019 (COVID-19) has affected millions of people worldwide and caused more than 3.9 million deaths [1]. Increased attempts have been made to develop deep learning methods to diagnosis COVID-19 based on computed tomography (CT) lung images. It is a challenge to reproduce and obtain the CT lung data because it is not publicly available. This paper introduces a new generalized framework to segment and classify CT images and determine whether a patient is tested positive or negative for COVID-19 based on lung CT images. In this work, many different strategies are explored for the classification task. ResNet50 and VGG16 models are applied to classify CT lung images into COVID-19 positive or negative. Also, VGG16 and ReNet50 combined with U-Net, which is one of the most used architectures in deep learning for image segmentation, is employed to segment CT lung images before the classifying process to increase system performance. Moreover, the image size dependent normalization technique (ISDNT) and Wiener filter are utilized as the preprocessing techniques to enhance images and noise suppression. Additionally, transfer learning and data augmentation techniques are performed to solve the problem of COVID-19 CT lung images deficiency, therefore the over-fitting of deep models can be avoided. The proposed frameworks which compromises of, end-to-end, VGG16, ResNet50, and U-Net with VGG16 or ResNet50 are applied on the dataset which is sourced from COVID-19 lung CT images in Kaggle. The classification results show that using the preprocessed CT lung images as the input for U-Net hybrid with ResNet50 achieve the best performance. The proposed classification model achieves 98.98% accuracy(ACC), 98.87% area under the ROC curve (AUC), 98.89% sensitivity (Se), 97.99% precision (Pr), 97.88% F1- score, and 1.8974-seconds computational time.