International Journal of Web Research

International Journal of Web Research

A Two-Stage Method for Diagnosing COVID-19, Leveraging CNN, and Transfer Learning on CT Scan Images

Document Type : Original Article

Authors
1 Department of Computer, Yazd Branch, Islamic Azad University, Yazd, Iran;
2 Department of Computer, Ershad Damavand University, Tehran, Iran
Abstract
Lung infection represents one of the most perilous indicators of Covid-19. The most efficient diagnostic approach entails the analysis of CT scan images. Utilizing deep learning algorithms and machine vision, computer scientists have devised a method for automated detection of this disease. This study proposes a two-stage approach to identifying lung infection. In the initial stage, image features are extracted through a transfer learning framework employing ResNet50, with the last two layers being fixed. Subsequently, a CNN neural network is constructed for image detection and categorization in the second stage. By employing superior image feature selection and minimizing non-informative features, this proposed method achieves impressive accuracy metrics: 98.99% accuracy, 98.91% sensitivity, and 99.10% specificity. Furthermore, a comparative analysis is conducted between this method and six other architectures (Inception, InceptionResNetV2, ResNet101, ResNet152, VGG16, VGG19), with and without transfer learning. The findings demonstrate that the proposed method attains 98% accuracy on test data, without succumbing to overfitting.
Keywords

Subjects


[1] C. Wang, P. W. Horby, F. G. Hayden and G. F. Gao, “A novel coronavirus outbreak of global health concern,” The lancet, vol. 395, no. 10223, pp. 470-473, 2020. https://doi.org/10.1016/S0140-6736(20)30185-9.
[2] T. Ozturk, M. Talo, E. A. Yildirim, U. B. Baloglu, O. Yildirim and U. R. Acharya, “Automated detection of COVID-19 cases using deep neural networks with X-ray images,” Computers in biology and medicine, vol. 121, p. 103792, 2020. https://doi.org/10.1016/j.compbiomed.2020.103792.
[3] V. Shah, R. Keniya, A. Shridharani, M. Punjabi, J. Shah and N. Mehendale, “Diagnosis of COVID-19 using CT scan images and deep learning techniques,” Emergency radiology, vol. 28, pp. 497-505, 2021. https://doi.org/10.1007/s10140-020-01886-y.
[4] J. Bullock, A. Luccioni, K. H. Pham, C. S. N. Lam and M. Luengo-Oroz, “Mapping the landscape of artificial intelligence applications against COVID-19,” Journal of Artificial Intelligence Research, vol. 69, pp. 807-845, 2020. https://doi.org/10.1613/jair.1.12162.
[5] T. Ai et al., “Correlation of chest CT and RT-PCR testing for coronavirus disease 2019 (COVID-19) in China: a report of 1014 cases,” Radiology, vol. 296, no. 2, pp. E32-E40, 2020. https://doi.org/10.1148/radiol.2020200642.
[6] K. M. Yee, “X-ray may be missing COVID cases found with CT,” Korean Journal of Radiology, pp. 1-7, 2020.
[7] Y. Fang et al., “Sensitivity of chest CT for COVID-19: comparison to RT-PCR,” Radiology, vol. 296, no. 2, pp. E115-E117, 2020. https://doi.org/10.1148/radiol.2020200432.
[8] K. He, X. Zhang, S. Ren and J. Sun, “Deep residual learning for image recognition,” in Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA, 2016, pp. 770-778. https://doi.org/10.1109/CVPR.2016.90.
[9] X. Yang, X. He, J. Zhao, Y. Zhang, S. Zhang and P. Xie, “COVID-CT-dataset: a CT scan dataset about COVID-19,” arXiv preprint arXiv:2003.13865, 2020. https://doi.org/10.48550/arXiv.2003.13865.
[10] M. Loey, G. Manogaran and N. E. M. Khalifa, “A deep transfer learning model with classical data augmentation and CGAN to detect COVID-19 from chest CT radiography digital images,” Neural Computing and Applications, pp. 1-13, 2020. https://doi.org/10.1007/s00521-020-05437-x.
[11] O. Gozes et al., “Rapid AI development cycle for the coronavirus (covid-19) pandemic: Initial results for automated detection & patient monitoring using deep learning ct image analysis,” arXiv preprint arXiv:2003.05037, 2020. https://doi.org/10.48550/arXiv.2003.05037.
[12] Y. H. Wu et al., “Jaccs: An explainable covid-19 diagnosis system by joint classification and segmentation,” IEEE Transactions on Image Processing, vol. 30, pp. 3113-3126, 2021. https://doi.org/10.1109/TIP.2021.3058783.
[13] S. Ahuja, B. K. Panigrahi, N. Dey, V. Rajinikanth and T. K. Gandhi, “Deep transfer learning-based automated detection of COVID-19 from lung CT scan slices,” Applied Intelligence, vol. 51, pp. 571-585, 2021. https://doi.org/10.1007/s10489-020-01826-w.
[14] S. H. Khan, A. Sohail, A. Khan and Y.-S. Lee, “COVID-19 detection in chest X-ray images using a new channel boosted CNN,” Diagnostics, vol. 12, no. 2, p. 267, 2022. https://doi.org/10.3390/diagnostics12020267.
[15] A. Banerjee, A. Sarkar, S. Roy, P. K. Singh and R. Sarkar, “COVID-19 chest X-ray detection through blending ensemble of CNN snapshots,” Biomedical Signal Processing and Control, vol. 78, p. 104000, 2022. https://doi.org/10.1016/j.bspc.2022.104000.
[16] A. H. Alamoodi et al., “Sentiment analysis and its applications in fighting COVID-19 and infectious diseases: A systematic review,” Expert systems with applications, vol. 167, p. 114155, 2021. https://doi.org/10.1016/j.eswa.2020.114155.
[17] A. Heidari and H. Erfanian, “Online COVID-19 Infection Diagnoses via Chest X-Ray Images using Alexnet Deep Learning Model,” International Journal of Web Research, vol. 5, no. 1, pp. 50-55, 2022. https://doi.org/10.22133/IJWR.2022.342423.1113.
[18] M. Behnia, T. Torabipour and S. Siadat, “Detection of COVID-19 Using a Pre-trained CNN Model Over Chest X-ray Images,” International Journal of Web Research, vol. 5, no. 2, pp. 103-112, 2022. https://doi.org/10.22133/IJWR.2023.375263.1145.
[19] M. J. Shayegan, “A brief review and scientometric analysis on ensemble learning methods for handling COVID-19,” Heliyon, vol. 10, no. 4, p. e26694, 2024. https://doi.org/10.1016/j.heliyon.2024.e26694.
[20] M. Hajiarbabi, “Skin cancer detection using multi-scale deep learning and transfer learning,” Journal of Medical Artificial Intelligence, vol. 6, 2023. https://doi.org/10.21037/jmai-23-67.
[21] A. S. Albahri et al., “Role of biological data mining and machine learning techniques in detecting and diagnosing the novel coronavirus (COVID-19): a systematic review,” Journal of medical systems, vol. 44, pp. 1-11, 2020. https://doi.org/10.1007/s10916-020-01582-x.
[22] T. Torabipour, Y. Jahangirigolshavari and S. Siadat, “A Deep Learning Approach for Diagnosis Chest Diseases,” International Journal of Web Research, vol. 4, no. 1, pp. 10-17, 2021. 10.22133/IJWR.2021.286134.1094.
[23] Z. Chen, E. A. Shamsabadi, S. Jiang, L. Shen and D. Dias-da-Costa, “An average pooling designed Transformer for robust crack segmentation,” Automation in Construction, vol. 162, p. 105367, 2024. https://doi.org/10.1016/j.autcon.2024.105367.
[24] SARS-COV-2Ct-Scan Dataset. https://www.kaggle.com/datasets/plameneduardo/sarscov2-ctscan-dataset. Accessed on: 19.12.2022.
[25] J. Han, M. Kamber and J. Pei, Data mining concepts and techniques third edition, University of Illinois at Urbana-Champaign Micheline Kamber Jian Pei Simon Fraser University, 2012.