Covid-19 Detection For CT-scan Images Using Transfer Learning Models
Article Sidebar
Main Article Content
Abstract
COVID-19 is a respiratory illness caused by a virus called SARS-CoV-2 which affected around 455 million people around the world. CT-scan is a medical imaging technique that uses X-rays to create detailed images of the body and which can be used to detect many respiratory diseases. Transfer learning models are a type of machine learning model that are trained on a large dataset of images and which can be used for their already trained ability to extract features from image in other tasks. They can then be used to classify new images with similar features.This paper presents a study of different transfer learning models for the task of classifying chest X-ray images into three classes: COVID-19, pneumonia, and normal. The study was implemented using Python and the dataset used was the COVID-19 Chest X-ray Dataset. The train-test split used was 0.2–0.8. The parameters used to test the models were the precision, recall, accuracy, F1 score, and Matthew’s correlation score. Other than these, different optimizers were also compared such as ADAM, SGD with different learning rates of 0.01, 0.001, and 0.0001.The models used in this study are EfficientNetB0, EfficientNetB7, VGG16, and InceptionV3. Out of these models, the most effective model was the EfficientNetB0 model, which achieved an accuracy of 98.6%. This study provides valuable insights into the use of transfer learning for medical image analysis. The results suggest that transfer learning can be used to develop accurate and efficient models that can be used as a secondary option for the diagnosis of COVID-19 using chest X-ray images.
Article Details
References
Anas M. Tahir , Muhammad E.H. Chowdhury , Amith Khandakar ,Tawsifur Rahman ,Yazan Qiblawey , Uzair Khurshid , Serkan Kiranyaz , Nabil Ibtehaz , M. Sohel Rahman ,Somaya Al-Maadeed , Sakib Mahmud , Maymouna Ezeddin , Khaled Hameed ,Tahir Hamid,“COVID-19 infection localization and severity grading from chest X-ray images”,https://doi.org/10.1016/j.compbiomed.2021.105002.
Elmehdi Benmalek, Jamal Elmhamdi, Abdelilah JilbabElmehdi Benmalek, Jamal Elmhamdi, Abdelilah Jilbab, “Comparing CT scan and chest X-ray imaging for COVID-19 diagnosis” https://doi.org/10.1016/j.bea.2021.100003.
Alaa S. Al-Waisy, Shumoos Al-Fahdawi, Mazin Abed Mohammed, Karrar Hameed Abdulkareem, Salama A. Mostafa, Mashael S. Maashi, Muhammad Arif, Begonya Garcia-Zapirain, “COVID-CheXNet: hybrid deep learning framework for identifying COVID-19 virus in chest X-rays images,” https://doi.org/10.1007/s00500-020-05424-3(0123456789().,-volV)(0123456789().
Arman Haghanifar, Mahdiyar Molahasani Majdabadi,Younhee Choi, S. Deivalakshmi, Seokbum Ko, “COVID-CXNet: Detecting COVID-19 in frontal chest X-ray images using deep learning,”Published online: 7 April 2022,https://doi.org/10.1007/s11042-022-12156-z.
Keno K. Bressem, Lisa C.Adams, Christoph Erxleben, Bernd Hamm, Stefan M. Niehues & Janis L.Vahldiek, “Comparing diferent deep learning architectures for classification of chest radiographs,”www.nature.com/scientificreports,https://doi.org/10.1038/s41598-020-70479-z
LindaWang, Zhong Qiu Lin & Alexander Wong, “COVID?Net: a tailored deep convolutional neural network design for detection of COVID?19 cases from chest X?ray images,”https://doi.org/10.1038/s41598-020-76550-z.
Pedro R. A. S. Bassi & Romis Attux, “A deep convolutional neural network for COVID-19 detection using chest X-rays,” 2 April 2021,https://doi.org/10.1007/s42600-021-00132-9.
Prottoy Saha, Muhammad Sheikh Sadi, Md. Milon Islam, “EMCNet: Automated COVID-19 diagnosis from X-ray images using convolutional neural network and ensemble of machine learning classifiers,” http://www.elsevier.com/locate/imu, https://doi.org/10.1016/j.imu.2020.100505.
Tahmina Zebin, Shahadate Rezvy, “COVID-19 detection and disease progression visualization: Deep learning on chest X-rays for classification and coarse localization,” https://doi.org/10.1007/s10489-020-01867-1.
Kolhe, J., Deshpande, G., Patel, G., Rajani, P.K.,” Crop Decision Using Various Machine Learning Classification Algorithms”, In: Choudrie, J., Mahalle, P., Perumal, T., Joshi, A. (eds) IOT with Smart Systems. Smart Innovation, Systems and Technologies, vol 312. Springer, Singapore,2023. https://doi.org/10.1007/978-981-19-3575-6_49 .
Nair, K., Motagi, N., Narayankar, R., Rajani, P.K,” Error Detection and Error Concealment of Medical Images Using Frequency Selective Extrapolation (FSE) Algorithm”, In: Kaiser, M.S., Xie, J., Rathore, V.S. (eds) Information and Communication Technology for Competitive Strategies (ICTCS 2021). Lecture Notes in Networks and Systems, vol 401. Springer, Singapore,2023. https://doi.org/10.1007/978-981-19-0098-3_48
Salman, Fatima M, Abu-Naser, Samy S, Alajrami, Eman, Abu-Nasser, Bassem S, Alashqar, Belal A. M., “COVID-19 Detection using Artificial Intelligence”, http://dspace.alazhar.edu.ps/xmlui/handle/123456789/587.
Soufiane Hamida,1Oussama El Gannour,1Bouchaib Cherradi,1,2 Abdelhadi Raihani,1 Hicham Moujahid,1 and Hassan Ouajji1, “A Novel COVID-19 Diagnosis Support System Using the Stacking Approach and Transfer Learning Technique on Chest X-Ray Images”
O. El Gannour, S. Hamida, B. Cherradi, A. Raihani, and H. Moujahid, “Performance evaluation of transfer learning technique for automatic detection of patients with COVID-19 on X-Ray images,” in Proceedings of the 2020 IEEE 2nd International Conference on Electronics, Control, Optimization and Computer Science (ICECOCS), pp. 1–6, Kenitra, Morocco, December 2020.
H. Moujahid, B. Cherradi, M. Al-Sarem, and L. Bahatti, “Diagnosis of COVID-19 disease using convolutional neural network models based transfer learning,” in Innovative Systems for Intelligent Health Informatics, F. Saeed, F. Mohammed, and A. Al-Nahari, Eds., vol. 72, pp. 148–159, Springer International Publishing, Cham, 2021,
B. K. Umri, M. Wafa Akhyari, and K. Kusrini, “Detection of Covid-19 in Chest X-ray image using CLAHE and convolutional neural network,” in Proceedings of the 2020 2nd International Conference on Cybernetics and Intelligent System (ICORIS), pp. 1–5, Manado, Indonesia, October 2020.
M. D. K. Hasan, S. Ahmed, Z. M. E. Abdullah et al., “Deep learning approaches for detecting pneumonia in COVID-19 patients by analyzing chest X-ray images,” Mathematical Problems in Engineering, vol. 2021, Article ID 9929274, 8 pages, 2021.
T. Li, W. Wei, L. Cheng et al., “Computer-aided diagnosis of COVID-19 CT scans based on spatiotemporal information fusion,” Journal of Healthcare Engineering, vol. 2021, Article ID 6649591, 11 pages, 2021.
X. Li, W. Tan, P. Liu, Q. Zhou, and J. Yang, “Classification of COVID-19 chest CT images based on ensemble deep learning,” Journal of Healthcare Engineering, vol. 2021, Article ID 5528441, 7 pages, 2021.
M. To?açar, B. Ergen, and Z. Cömert, “COVID-19 detection using deep learning models to exploit social mimic optimization and structured chest X-ray images using fuzzy color and stacking approaches,” Computers in Biology and Medicine, vol. 121, Article ID 103805, 2020.
Y. Jiang, H. Chen, M. Loew, and H. Ko, “COVID-19 CT image synthesis with a conditional generative adversarial network,” IEEE Journal of Biomedical and Health Informatics, vol. 25, no. 2, pp. 441–452, 2021.
S. Hamida, O. E. Gannour, B. Cherradi, H. Ouajji, and A. Raihani, “Optimization of machine learning algorithms hyper-parameters for improving the prediction of patients infected with COVID-19,” in Proceedings of the 2020 IEEE 2nd International Conference on Electronics, Control, Optimization and Computer Science (ICECOCS), pp. 1–6, Kenitra, Morocco, December 2020.
C. C. John, V. Ponnusamy, S. Krishnan Chandrasekaran, and N. R, “A survey on mathematical, machine learning and deep learning models for COVID-19 transmission and diagnosis,” IEEE Reviews in Biomedical Engineering, vol. 2021, Article ID 3069213, 1 page, 2021.
E. F. Ohata, G. M. Bezerra, J. V. S. d. Chagas et al., “Automatic detection of COVID-19 infection using chest X-ray images through transfer learning,” IEEE/CAA Journal of Automatica Sinica, vol. 2020, Article ID 1003393, 10 pages, 2020.
Y. Pathak, P. K. Shukla, A. Tiwari, S. Stalin, S. Singh, and P. K. Shukla, “Deep transfer learning based classification model for COVID-19 disease,” IRBM, vol. 2020, Article ID S1959031820300993, 2020.
T. Mahmud, M. A. Rahman, and S. A. Fattah, “CovXNet: a multi-dilation convolutional neural network for automatic COVID-19 and other pneumonia detection from chest X-ray images with transferable multi-receptive feature optimization,” Computers in Biology and Medicine, vol. 122, Article ID 103869, 2020.
Shi H., Han X., Jiang N., Cao Y., Alwalid O., Gu J., et al., “Radiological findings from 81 patients with covid-19 pneumonia in wuhan, china: a descriptive study”, The Lancet Infectious Diseases, 20 (4) (2020), pp. 425-434
Singh A.K., Kumar A., Mahmud M., Kaiser M.S., Kishore A., “Covid-19 infection detection from chest X-ray images using hybrid social group optimization and support vector classifier”, Cognitive Computation (2021), pp. 1-13
Wang W., Xu Y., Gao R., Lu R., Han K., Wu G., et al., “Detection of SARS-CoV-2 in different types of clinical specimens”, Journal of the American Medical Association, 323 (18) (2020), pp. 1843-1844
Yang W., Sirajuddin A., Zhang X., Liu G., Teng Z., Zhao S., et al., “The role of imaging in 2019 novel coronavirus pneumonia (covid-19)”, * Radiology, 30 (9) (2020), pp. 4874-4882