An efficient transfer learning approach for prediction and classification of SARS - COVID-19

被引:3
作者
Joshi, Krishna Kumar [1 ]
Gupta, Kamlesh [2 ]
Agrawal, Jitendra [1 ]
机构
[1] Rajiv Gandhi Proudyogiki Vishwavidyalaya, Sch Informat Technol, Bhopal, Madhya Pradesh, India
[2] Rustam Ji Inst Technol, Dept Informat Technol, Gwalior, Madhya Pradesh, India
关键词
Transfer Learning; VGG; 16; Covid-19; Confusion Matrix; CLAHE; Weiner Filter; ReLU; ACUTE RESPIRATORY SYNDROME;
D O I
10.1007/s11042-023-17086-y
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
COVID-19 or Corona Virus Disease is a dangerous disease that spreads quickly and affects human life brutally, becoming the cause of death of millions of people. COVID mainly infects the lungs of human beings and that's why lung images are mostly used for Covid detection. In this research paper, an innovative efficient Transfer Learning approach for Covid -19 prediction is described. The dataset contains Computer Tomography images of the Chest and has been collected from two sources: First SARS-CoV-2 CT has 2482 images of Covid and healthy persons. Non-Covid CT scan images of both SARS-CoV-2 positive and negative patients and second UCSD-AI4H COVID CT scan dataset containing 745 CT scan images of both SARS-CoV-2 positive and negative patients. As datasets containing images are collected from various sources, machine learning pre-processing techniques such as CLAHE, and data augmentation, are applied to enhance contrast, quality, and quantity. Then, a popular pre-trained CNN, called VGG16, is used as the base model, as it has shown its ability on the image dataset in terms of very good accuracy in the prediction and classification of the images related to medical diagnosis. Then we build our sequential model by applying ReLU and Softmax functions and trained it with 21,137,986 parameters. The dataset is divided into three parts in the ratio of 80%, 10%, and 10%, for Training, Testing, and Validation, respectively. The model shows a very good classification of Covid-positive and Covid Negative images as it was able to provide an accuracy of 95%. The Precision was 95%, Recall was 95%, and the F-1 Score was 95%.
引用
收藏
页码:39435 / 39457
页数:23
相关论文
共 74 条
[1]   Review of deep learning: concepts, CNN architectures, challenges, applications, future directions [J].
Alzubaidi, Laith ;
Zhang, Jinglan ;
Humaidi, Amjad J. ;
Al-Dujaili, Ayad ;
Duan, Ye ;
Al-Shamma, Omran ;
Santamaria, J. ;
Fadhel, Mohammed A. ;
Al-Amidie, Muthana ;
Farhan, Laith .
JOURNAL OF BIG DATA, 2021, 8 (01)
[2]   Novel Transfer Learning Approach for Medical Imaging with Limited Labeled Data [J].
Alzubaidi, Laith ;
Al-Amidie, Muthana ;
Al-Asadi, Ahmed ;
Humaidi, Amjad J. ;
Al-Shamma, Omran ;
Fadhel, Mohammed A. ;
Zhang, Jinglan ;
Santamaria, J. ;
Duan, Ye .
CANCERS, 2021, 13 (07)
[3]   Machine Learning Approach for COVID-19 Detection on Twitter [J].
Amin, Samina ;
Uddin, M. Irfan ;
Al-Baity, Heyam H. ;
Zeb, M. Ali ;
Khan, M. Abrar .
CMC-COMPUTERS MATERIALS & CONTINUA, 2021, 68 (02) :2231-2247
[4]  
Arun JB., 2013, Int J Eng Res Technol, V2, P3219
[5]   Application of Deep Learning in Breast Cancer Imaging [J].
Balkenende, Luuk ;
Teuwen, Jonas ;
Mann, Ritse M. .
SEMINARS IN NUCLEAR MEDICINE, 2022, 52 (05) :584-596
[6]   Prediction of Heart Disease Using a Combination of Machine Learning and Deep Learning [J].
Bharti, Rohit ;
Khamparia, Aditya ;
Shabaz, Mohammad ;
Dhiman, Gaurav ;
Pande, Sagar ;
Singh, Parneet .
COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2021, 2021
[7]  
Brunese Luca, 2020, Procedia Comput Sci, V176, P2212, DOI 10.1016/j.procs.2020.09.258
[8]   Machine Learning Based Diabetes Classification and Prediction for Healthcare Applications [J].
Butt, Umair Muneer ;
Letchmunan, Sukumar ;
Ali, Mubashir ;
Hassan, Fadratul Hafinaz ;
Baqir, Anees ;
Sherazi, Hafiz Husnain Raza .
JOURNAL OF HEALTHCARE ENGINEERING, 2021, 2021
[9]   Estimating the efficacy of symptom-based screening for COVID-19 [J].
Callahan, Alison ;
Steinberg, Ethan ;
Fries, Jason A. ;
Gombar, Saurabh ;
Patel, Birju ;
Corbin, Conor K. ;
Shah, Nigam H. .
NPJ DIGITAL MEDICINE, 2020, 3 (01)
[10]  
Cardarilli GC, 2021, SCI REP-UK, V11, DOI 10.1038/s41598-021-94691-7