Automated COVID-19 detection with convolutional neural networks

被引:12
作者
Dumakude, Aphelele [1 ]
Ezugwu, Absalom E. [2 ]
机构
[1] Univ KwaZulu Natal, Sch Math Stat & Comp Sci, King Edward Ave, Pietermaritzburg Campus, ZA-3201 Pietermaritzburg, Kwazulu Natal, South Africa
[2] North West Univ, Unit Data Sci & Comp, 11 Hoffman St, ZA-2520 Potchefstroom, South Africa
关键词
CNN;
D O I
10.1038/s41598-023-37743-4
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
This paper focuses on addressing the urgent need for efficient and accurate automated screening tools for COVID-19 detection. Inspired by existing research efforts, we propose two framework models to tackle this challenge. The first model combines a conventional CNN architecture as a feature extractor with XGBoost as the classifier. The second model utilizes a classical CNN architecture with a Feedforward Neural Network for classification. The key distinction between the two models lies in their classification layers. Bayesian optimization techniques are employed to optimize the hyperparameters of both models, enabling a "cheat-start" to the training process with optimal configurations. To mitigate overfitting, transfer learning techniques such as Dropout and Batch normalization are incorporated. The CovidxCT-2A dataset is used for training, validation, and testing purposes. To establish a benchmark, we compare the performance of our models with state-of-the-art methods reported in the literature. Evaluation metrics including Precision, Recall, Specificity, Accuracy, and F1-score are employed to assess the efficacy of the models. The hybrid model demonstrates impressive results, achieving high precision (98.43%), recall (98.41%), specificity (99.26%), accuracy (99.04%), and F1-score (98.42%). The standalone CNN model exhibits slightly lower but still commendable performance, with precision (98.25%), recall (98.44%), specificity (99.27%), accuracy (98.97%), and F1-score (98.34%). Importantly, both models outperform five other state-of-the-art models in terms of classification accuracy, as demonstrated by the results of this study.
引用
收藏
页数:30
相关论文
共 46 条
[1]   COVID-19 Detection Using Deep Learning Algorithm on Chest X-ray Images [J].
Akter, Shamima ;
Shamrat, F. M. Javed Mehedi ;
Chakraborty, Sovon ;
Karim, Asif ;
Azam, Sami .
BIOLOGY-BASEL, 2021, 10 (11)
[2]  
Alazab Moutaz, 2020, International Journal of Computer Information Systems and Industrial Management Applications, P168
[3]   COVID-19 diagnosis using state-of-the-art CNN architecture features and Bayesian Optimization [J].
Aslan, Muhammet Fatih ;
Sabanci, Kadir ;
Durdu, Akif ;
Unlersen, Muhammed Fahri .
COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 142
[4]   CNN-based transfer learning-BiLSTM network: A novel approach for COVID-19 infection detection [J].
Aslan, Muhammet Fatih ;
Unlersen, Muhammed Fahri ;
Sabanci, Kadir ;
Durdu, Akif .
APPLIED SOFT COMPUTING, 2021, 98
[5]   Variation of the hydrological regime of Bele-Shira closed basin in Southern Siberia and its reflection in the radial growth of Larix sibirica [J].
Babushkina, Elena A. ;
Belokopytova, Liliana V. ;
Grachev, Alexi M. ;
Meko, David M. ;
Vaganov, Eugene A. .
REGIONAL ENVIRONMENTAL CHANGE, 2017, 17 (06) :1725-1737
[6]   Prediction of COVID-19 from Chest CT Images Using an Ensemble of Deep Learning Models [J].
Biswas, Shreya ;
Chatterjee, Somnath ;
Majee, Arindam ;
Sen, Shibaprasad ;
Schwenker, Friedhelm ;
Sarkar, Ram .
APPLIED SCIENCES-BASEL, 2021, 11 (15)
[7]  
Chen T., 2015, R PACKAGE VERSION 04, V1, P1
[8]   Accurate seat belt detection in road surveillance images based on CNN and SVM [J].
Chen, Yanxiang ;
Tao, Gang ;
Ren, Hongmei ;
Lin, Xinyu ;
Zhang, Luming .
NEUROCOMPUTING, 2018, 274 :80-87
[9]   Can AI Help in Screening Viral and COVID-19 Pneumonia? [J].
Chowdhury, Muhammad E. H. ;
Rahman, Tawsifur ;
Khandakar, Amith ;
Mazhar, Rashid ;
Kadir, Muhammad Abdul ;
Bin Mahbub, Zaid ;
Islam, Khandakar Reajul ;
Khan, Muhammad Salman ;
Iqbal, Atif ;
Al Emadi, Nasser ;
Reaz, Mamun Bin Ibne ;
Islam, Mohammad Tariqul .
IEEE ACCESS, 2020, 8 :132665-132676
[10]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848