International Journal of Advanced Technology and Engineering Exploration (IJATEE) ISSN (P): 2394-5443 ISSN (O): 2394-7454 Vol - 10, Issue - 103, June 2023
  1. 1
    Google Scholar
Transfer learning with fine-tuned deep CNN model for COVID-19 diagnosis from chest X-ray images

Mamta Patel and Mehul Shah

Abstract

The COVID-19 pandemic has had a significant impact on people's lives, necessitating accurate detection and early diagnosis to control the dissemination of virus. Reverse transcription-polymerase chain reaction (RT-PCR) is the most prevalent diagnostic strategy, but its accuracy is influenced by various factors such as sample collection, timing, and processing. Deep Convolutional Neural Networks (DCNNs) have shown great promise in medical image analysis and are consequently being utilized for the diagnosis of COVID-19 from radiographic images. This study evaluates the effectiveness of different convolutional neural network (CNN) architectures with optimum hyperparameters for COVID-19 diagnosis using publicly available chest radiography datasets. The evaluated models included both CNN architectures built from scratch and pre-trained CNN architectures, such as residual network (ResNet-50), visual geometry group (VGG-16), VGG-19, Inception-V3, and MobileNet-V2. The experimental results demonstrate that MobileNet-V2 achieved 96% accuracy, precision, recall, F1 score, and area under the curve (AUC), making it a prospective and acceptable model for COVID-19 diagnosis. In contrast to existing models, the proposed model's evaluation also includes an assessment of network training time and memory consumption. The study also describes the web deployment of a deep CNN-based computer-aided diagnosis (CAD) system that can assist doctors in diagnosing COVID-19 faster, more accurately, and more consistently. This advancement leads to better patient outcomes and improved efficiency within the healthcare system.

Keyword

COVID-19, Chest radiography, Deep learning (DL), Convolutional neural network (CNN), Transfer learning (TL).

Cite this article

Patel M, Shah M

Refference

[1][1]WHO. Coronavirus overview, prevention and symptoms. https://www.who.int/emergencies/diseases/novel-coronavirus-2019. Accessed 20 November 2020.

[2][2]Okolo GI, Katsigiannis S, Althobaiti T, Ramzan N. On the use of deep learning for imaging-based COVID-19 detection using chest X-rays. Sensors. 2021; 21(17):5702.

[3][3]Sharma A, Tiwari S, Deb MK, Marty JL. Severe acute respiratory syndrome coronavirus-2 (SARS-CoV-2): a global pandemic and treatment strategies. International Journal of Antimicrobial Agents. 2020; 56(2):1-13.

[4][4]Abbasi-oshaghi E, Mirzaei F, Farahani F, Khodadadi I, Tayebinia H. Diagnosis and treatment of coronavirus disease 2019 (COVID-19): laboratory, PCR, and chest CT imaging findings. International Journal of Surgery. 2020; 79:143-53.

[5][5]Wu YC, Chen CS, Chan YJ. The outbreak of COVID-19: an overview. Journal of the Chinese Medical Association. 2020; 83(3):217-20.

[6][6]Himoto Y, Sakata A, Kirita M, Hiroi T, Kobayashi KI, Kubo K, et al. Diagnostic performance of chest CT to differentiate COVID-19 pneumonia in non-high-epidemic area in Japan. Japanese Journal of Radiology. 2020; 38:400-6.

[7][7]Woloshin S, Patel N, Kesselheim AS. False negative tests for SARS-CoV-2 infection—challenges and implications. New England Journal of Medicine. 2020; 383(6):1-3.

[8][8]Rubin GD, Ryerson CJ, Haramati LB, Sverzellati N, Kanne JP, Raoof S, et al. The role of chest imaging in patient management during the COVID-19 pandemic: a multinational consensus statement from the Fleischner society. Radiology. 2020; 296(1):172-80.

[9][9]Moghadam SO. A review on an ongoing pandemic caused by the severe acute respiratory syndrome coronavirus 2: the pathogenesis, epidemiology, immunological features, and currently available diagnostic tests. Reviews in Medical Microbiology. 2022; 33(1):e212-23.

[10][10]Chowdhury ME, Rahman T, Khandakar A, Mazhar R, Kadir MA, Mahbub ZB, et al. Can AI help in screening viral and COVID-19 pneumonia? IEEE Access. 2020; 8:132665-76.

[11][11]Khan A, Sohail A, Zahoora U, Qureshi AS. A survey of the recent architectures of deep convolutional neural networks. Artificial Intelligence Review. 2020; 53:5455-516.

[12][12]Apostolopoulos ID, Mpesiana TA. Covid-19: automatic detection from x-ray images utilizing transfer learning with convolutional neural networks. Physical and Engineering Sciences in Medicine. 2020; 43:635-40.

[13][13]Bohr A, Memarzadeh K. The rise of artificial intelligence in healthcare applications. In artificial intelligence in healthcare 2020 (pp. 25-60). Academic Press.

[14][14]Alzubaidi L, Zhang J, Humaidi AJ, Al-dujaili A, Duan Y, Al-shamma O, et al. Review of deep learning: concepts, CNN architectures, challenges, applications, future directions. Journal of Big Data. 2021; 8:1-74.

[15][15]Alafif T, Tehame AM, Bajaba S, Barnawi A, Zia S. Machine and deep learning towards COVID-19 diagnosis and treatment: survey, challenges, and future directions. International Journal of Environmental Research and Public Health. 2021; 18(3):1117.

[16][16]Nayak SR, Nayak DR, Sinha U, Arora V, Pachori RB. Application of deep learning techniques for detection of COVID-19 cases using chest X-ray images: a comprehensive study. Biomedical Signal Processing and Control. 2021; 64:102365.

[17][17]Gatto A, Accarino G, Aloisi V, Immorlano F, Donato F, Aloisio G. Limits of compartmental models and new opportunities for machine learning: a case study to forecast the second wave of COVID-19 hospitalizations in Lombardy, Italy. Informatics. 2021; 8(3): 57.

[18][18]Paul SG, Saha A, Biswas AA, Zulfiker MS, Arefin MS, Rahman MM, et al. Combating Covid-19 using machine learning and deep learning: Applications, challenges, and future perspectives. Array. 2022:100271.

[19][19]Cai L, Gao J, Zhao D. A review of the application of deep learning in medical image classification and segmentation. Annals of Translational Medicine. 2020; 8(11):1-15.

[20][20]Esteva A, Chou K, Yeung S, Naik N, Madani A, Mottaghi A, et al. Deep learning-enabled medical computer vision. NPJ Digital Medicine. 2021; 4(1):1-9.

[21][21]Kim M, Yun J, Cho Y, Shin K, Jang R, Bae HJ, et al. Deep learning in medical imaging. Neurospine. 2019; 16(4):657-68.

[22][22]Arias-londoño JD, Gomez-garcia JA, Moro-velazquez L, Godino-llorente JI. Artificial intelligence applied to chest X-ray images for the automatic detection of COVID-19. A thoughtful evaluation approach. IEEE Access. 2020; 8:226811-27.

[23][23]Yang D, Martinez C, Visuña L, Khandhar H, Bhatt C, Carretero J. Detection and analysis of COVID-19 in medical images using deep learning techniques. Scientific Reports. 2021; 11(1):1-13.

[24][24]Toraman S, Alakus TB, Turkoglu I. Convolutional capsnet: a novel artificial neural network approach to detect COVID-19 disease from X-ray images using capsule networks. Chaos, Solitons & Fractals. 2020; 140:1-11.

[25][25]He X, Yang X, Zhang S, Zhao J, Zhang Y, Xing E, et al. Sample-efficient deep learning for COVID-19 diagnosis based on CT scans. IEEE Transactions on Medical Imaging. 2020:1-10.

[26][26]Kassania SH, Kassanib PH, Wesolowskic MJ, Schneidera KA, Detersa R. Automatic detection of coronavirus disease (COVID-19) in X-ray and CT images: a machine learning based approach. Biocybernetics and Biomedical Engineering. 2021; 41(3):867-79.

[27][27]Basha SS, Dubey SR, Pulabaigari V, Mukherjee S. Impact of fully connected layers on performance of convolutional neural networks for image classification. Neurocomputing. 2020; 378:112-9.

[28][28]Haque KF, Haque FF, Gandy L, Abdelgawad A. Automatic detection of COVID-19 from chest X-ray images with convolutional neural networks. In international conference on computing, electronics & communications engineering 2020 (pp. 125-30). IEEE.

[29][29]Wang L, Lin ZQ, Wong A. Covid-net: a tailored deep convolutional neural network design for detection of covid-19 cases from chest x-ray images. Scientific Reports. 2020; 10(1):1-2.

[30][30]Diaz-escobar J, Ordóñez-guillén NE, Villarreal-reyes S, Galaviz-mosqueda A, Kober V, Rivera-rodriguez R, et al. Deep-learning based detection of COVID-19 using lung ultrasound imagery. Plos one. 2021; 16(8):e0255886.

[31][31]Wang N, Liu H, Xu C. Deep learning for the detection of COVID-19 using transfer learning and model integration. In 10th international conference on electronics information and emergency communication 2020 (pp. 281-4). IEEE.

[32][32]Wang X, Peng Y, Lu L, Lu Z, Bagheri M, Summers RM. Chestx-ray8: hospital-scale chest x-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases. In proceedings of the IEEE conference on computer vision and pattern recognition 2017 (pp. 2097-106).

[33][33]Kaur P, Harnal S, Tiwari R, Alharithi FS, Almulihi AH, Noya ID, et al. A hybrid convolutional neural network model for diagnosis of COVID-19 using chest X-ray images. International Journal of Environmental Research and Public Health. 2021; 18(22):1-17.

[34][34]Bahuguna A, Yadav D, Senapati A, Nath Saha B. k NN-SVM with deep features for COVID-19 pneumonia detection from chest X-ray. In international conference on mathematics and computing 2022 (pp. 103-15). Singapore: Springer Nature Singapore.

[35][35]Biswas S, Chatterjee S, Majee A, Sen S, Schwenker F, Sarkar R. Prediction of covid-19 from chest CT images using an ensemble of deep learning models. Applied Sciences. 2021; 11(15):1-16.

[36][36]Hossain MB, Iqbal SH, Islam MM, Akhtar MN, Sarker IH. Transfer learning with fine-tuned deep CNN ResNet50 model for classifying COVID-19 from chest X-ray images. Informatics in Medicine Unlocked. 2022; 30:1-10.

[37][37]Mercaldo F, Belfiore MP, Reginelli A, Brunese L, Santone A. Coronavirus covid-19 detection by means of explainable deep learning. Scientific Reports. 2023; 13(1):462.

[38][38]Duong LT, Nguyen PT, Iovino L, Flammini M. Automatic detection of Covid-19 from chest X-ray and lung computed tomography images using deep neural networks and transfer learning. Applied Soft Computing. 2023; 132:109851.

[39][39]https://www.sciencedirect.com/topics/computer-science/imaging-modality. Accessed 10 May 2023.

[40][40]https://www.kaggle.com/tawsifurrahman/covid19-radiography-database. Accessed 10 May 2023.

[41][41]Yamashita R, Nishio M, Do RK, Togashi K. Convolutional neural networks: an overview and application in radiology. Insights into Imaging. 2018; 9:611-29.

[42][42]Sarker IH. Deep learning: a comprehensive overview on techniques, taxonomy, applications and research directions. SN Computer Science. 2021; 2(6):1-20.

[43][43]He K, Zhang X, Ren S, Sun J. Deep residual learning for image recognition. In proceedings of the IEEE conference on computer vision and pattern recognition 2016 (pp. 770-8).

[44][44]Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556. 2014.

[45][45]Szegedy C, Liu W, Jia Y, Sermanet P, Reed S, Anguelov D, et al. Going deeper with convolutions. In proceedings of the IEEE conference on computer vision and pattern recognition 2015 (pp. 1-9).

[46][46]Krizhevsky A, Sutskever I, Hinton GE. Imagenet classification with deep convolutional neural networks. Communications of the ACM. 2017; 60(6):84-90.

[47][47]Howard AG, Zhu M, Chen B, Kalenichenko D, Wang W, Weyand T, et al. Mobilenets: efficient convolutional neural networks for mobile vision applications. arXiv preprint arXiv:1704.04861. 2017.

[48][48]Choudhary K, Decost B, Chen C, Jain A, Tavazza F, Cohn R, et al. Recent advances and applications of deep learning methods in materials science. NPJ Computational Materials. 2022; 8(1):59.

[49][49]Nikolaou V, Massaro S, Fakhimi M, Stergioulas L, Garn W. COVID-19 diagnosis from chest x-rays: developing a simple, fast, and accurate neural network. Health Information Science and Systems. 2021; 9:1-11.

[50][50]Hemdan EE, Shouman MA, Karar ME. Covidx-net: a framework of deep learning classifiers to diagnose covid-19 in x-ray images. Computer Vision and Pattern Recognition. 2020:1-14.

[51][51]Medhi K, Jamil M, Hussain MI. Automatic detection of COVID-19 infection from chest X-ray using deep learning. Medrxiv. 2020: 2020-05.

[52][52]Waheed A, Goyal M, Gupta D, Khanna A, Al-turjman F, Pinheiro PR. Covidgan: data augmentation using auxiliary classifier gan for improved covid-19 detection. IEEE Access. 2020; 8:91916-23.

[53][53]Tammina S. Covidsort: detection of novel covid-19 in chest x-ray images by leveraging deep transfer learning models. In ICDSMLA: proceedings of the 2nd international conference on data science, machine learning and applications 2022 (pp. 431-47). Springer Singapore.

[54][54]Constantinou M, Exarchos T, Vrahatis AG, Vlamos P. COVID-19 classification on chest X-ray images using deep learning methods. International Journal of Environmental Research and Public Health. 2023; 20(3):2035.

[55][55]Sait U, Lal KG, Prajapati S, Bhaumik R, Kumar T, Sanjana S, et al. Curated dataset for COVID-19 posterior-anterior chest radiography images (X-Rays). Mendeley Data. 2020;1.

[56][56]https://data.mendeley.com/datasets/8h65ywd2jr. Accessed 10 May 2023.

[57][57]Cohen JP, Morrison P, Dao L, Roth K, Duong TQ, Ghassemi M. Covid-19 image data collection: Prospective predictions are the future. arXiv preprint arXiv:2006.11988. 2020.