TechRxiv
Convolutional Neural Network (CNN) in COVID-19 Detection_A Case Study with Chest CT Scan Images.pdf (5.72 MB)
Download file

Convolutional Neural Network (CNN) inCOVID-19 Detection: A Case Study with Chest CT Scan Images

Download (5.72 MB)
preprint
posted on 04.05.2022, 21:47 by Khandaker Nusaiba HafizKhandaker Nusaiba Hafiz, Khandaker Foysal HaqueKhandaker Foysal Haque

Deep Learning, especially Convolutional Neural Networks (CNN) have been performing very well for the last decade in medical image classification. CNN has already shown a great prospect in detecting COVID-19 from chest X-ray images. However, due to its three dimensional data, chest CT scan images can provide better understanding of the affected area through segmentation in comparison to the chest X-ray images. But the chest CT scan images have not been explored enough to achieve sufficiently good results in comparison to the X-ray images. However, with proper image pre-processing, fine tuning and optimization of the models better results can be achieved. This work aims in contributing to filling this void of the literature. On this aspect, this work explores and design both custom CNN model and three other models based on transfer learning: InceptionV3, ResNet50 and VGG19. The best performing model is VGG19 with an accuracy of 98.39% and F-1 score of 98.52%. The main contribution of this work includes: (i) modeling a custom CNN model and three pre-trained models based on InceptionV3, ResNet50, and VGG19 (ii) training and validating the models with a comparatively larger dataset of 1252 COVID19 and 1230 non-COVID CT images (iii) fine tune and optimize the designed models based on the parameters like number of dense layers, optimizer, learning rate, batch size, decay rate, and activation functions to achieve better results than the most of the state-of-the-art literature (iv) the designed models are made public in [1] for reproducibility by the research community for further developments and improvements.

History

Email Address of Submitting Author

nusaiba.hafiz@email.ucr.edu

Submitting Author's Institution

University of California, Riverside

Submitting Author's Country

Bangladesh