Detecting COVID-19 in chest CT images based on several pre-trained models

Esraa Hassan,Mahmoud Y. Shams,Noha A. Hikal,Samir Elmougy
DOI: https://doi.org/10.1007/s11042-023-17990-3
IF: 2.577
2024-01-16
Multimedia Tools and Applications
Abstract:This paper explores the use of chest CT scans for early detection of COVID-19 and improved patient outcomes. The proposed method employs advanced techniques, including binary cross-entropy, transfer learning, and deep convolutional neural networks, to achieve accurate results. The COVIDx dataset, which contains 104,009 chest CT images from 1,489 patients, is used for a comprehensive analysis of the virus. A sample of 13,413 images from this dataset is categorised into two groups: 7,395 CT scans of individuals with confirmed COVID-19 and 6,018 images of normal cases. The study presents pre-trained transfer learning models such as ResNet (50), VGG (19), VGG (16), and Inception V3 to enhance the DCNN for classifying the input CT images. The binary cross-entropy metric is used to compare COVID-19 cases with normal cases based on predicted probabilities for each class. Stochastic Gradient Descent and Adam optimizers are employed to address overfitting issues. The study shows that the proposed pre-trained transfer learning models achieve accuracies of 99.07%, 98.70%, 98.55%, and 96.23%, respectively, in the validation set using the Adam optimizer. Therefore, the proposed work demonstrates the effectiveness of pre-trained transfer learning models in enhancing the accuracy of DCNNs for image classification. Furthermore, this paper provides valuable insights for the development of more accurate and efficient diagnostic tools for COVID-19.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the early detection of COVID - 19 in chest CT images in order to improve the treatment outcomes of patients. Specifically, by using advanced techniques such as Binary Cross - Entropy (BCE), Transfer Learning (TL) and Deep Convolutional Neural Networks (DCNN), the paper aims to improve the accuracy of COVID - 19 detection. The paper uses the COVIDx dataset containing 104,009 chest CT images from 1,489 patients, among which 13,413 images are classified into two categories: 7,395 CT scan images confirmed as COVID - 19 and 6,018 normal case images. ### Main objectives of the paper: 1. **Improve the accuracy of COVID - 19 detection**: By using pre - trained transfer learning models (such as ResNet (50), VGG (19), VGG (16) and Inception V3), enhance the performance of DCNN in classifying input CT images. 2. **Reduce the over - fitting problem**: Use Stochastic Gradient Descent (SGD) and Adaptive Moment Estimation (Adam) optimizers to solve the over - fitting problem. 3. **Provide valuable insights**: Provide references for the development of more accurate and efficient COVID - 19 diagnostic tools. ### Main methods: - **Dataset**: Use 13,413 chest CT images in the COVIDx dataset, divided into COVID - 19 cases and normal cases. - **Pre - processing**: Perform pre - processing steps such as normalization and enhancement on the images to increase the amount of training data and reduce the risk of over - fitting. - **Feature extraction**: Utilize pre - trained transfer learning models (such as ResNet (50), VGG (19), VGG (16) and Inception V3) to extract features. - **Classifier**: Use the Binary Cross - Entropy loss function (BCE) for classification and use Adam and SGD optimizers for training. ### Experimental results: - **Accuracy**: When using the Adam optimizer, ResNet (50), VGG (19), VGG (16) and Inception V3 achieved accuracies of 99.07%, 98.70%, 98.55% and 96.23% respectively on the validation set. - **Performance comparison**: Compared with other existing methods, this study performs excellently in terms of accuracy, especially when using the Adam optimizer. ### Conclusion: This paper demonstrates the effectiveness of using transfer learning and deep convolutional neural networks in detecting COVID - 19 in chest CT images. By using pre - trained models and the Binary Cross - Entropy loss function, the study not only improves the accuracy of detection, but also provides valuable methods and insights for further research in the field of medical image analysis.