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 条
[41]   STACKED GENERALIZATION [J].
WOLPERT, DH .
NEURAL NETWORKS, 1992, 5 (02) :241-259
[42]   Group Normalization [J].
Wu, Yuxin ;
He, Kaiming .
COMPUTER VISION - ECCV 2018, PT XIII, 2018, 11217 :3-19
[43]  
XGBoost, XGBOOST ML WINN SOL
[44]   CLASSIC: Consistent longitudinal alignment and segmentation for serial image computing [J].
Xue, Z ;
Shen, DG ;
Davatzikos, C .
NEUROIMAGE, 2006, 30 (02) :388-399
[45]   Deep learning for COVID-19 detection based on CT images [J].
Zhao, Wentao ;
Jiang, Wei ;
Qiu, Xinguo .
SCIENTIFIC REPORTS, 2021, 11 (01)
[46]   How to Represent Paintings: A Painting Classification Using Artistic Comments [J].
Zhao, Wentao ;
Zhou, Dalin ;
Qiu, Xinguo ;
Jiang, Wei .
SENSORS, 2021, 21 (06) :1-15