COVIDXception-Net: A Bayesian Optimization-Based Deep Learning Approach to Diagnose COVID-19 from X-Ray Images

被引:10
作者
Shifat E. Arman
Sejuti Rahman
Shamim Ahmed Deowan
机构
[1] Department of Robotics and Mechatronics Engineering, University of Dhaka, Dhaka
关键词
Bayesian optimization; Chest X-ray; COVID-19; Deep learning; SARS-CoV-2;
D O I
10.1007/s42979-021-00980-3
中图分类号
学科分类号
摘要
COVID-19 is spreading around the world like wildfire. Chest X-rays are used as one of the primary tools for diagnosing COVID-19. However, about two-thirds of the world population do not have access to sufficient radiological services. In this work, we propose a deep learning-driven automated system, COVIDXception-Net, for diagnosing COVID-19 from chest X-rays. A primary challenge in any data-driven COVID-19 detection is the scarcity of COVID-19 data, which heavily deteriorates a deep learning model’s performance. To address this issue, we incorporate a weighted-loss function that ensures the COVID-19 cases are given more importance during the training process. We also propose using Bayesian Optimization to find the best architecture for detecting COVID-19. Extensive experimentation on four publicly available COVID-19 datasets shows that our proposed model achieves an accuracy of 0.94, precision 0.95, recall 0.94, specificity 0.997, F1-score 0.94, and Matthews correlation coefficient 0.992 outperforming three widely used architectures—VGG16, MobileNetV2, and InceptionV3. It also surpasses the performance of several state-of-the-art COVID-19 detection methods. We also performed two ablation studies that show our model’s accuracy degrades from 0.994 to 0.950 when a random search is used and to 0.983 when a regular loss function is employed instead of the Bayesian and weighted loss, respectively. © The Author(s), under exclusive licence to Springer Nature Singapore Pte Ltd 2021.
引用
收藏
相关论文
共 68 条
[1]  
Weiss S.R., Leibowitz J.L., Coronavirus pathogenesis, Advances in virus research, 81, pp. 85-164, (2011)
[2]  
Kooraki S., Hosseiny M., Myers L., Gholamrezanezhad A., Coronavirus (covid-19) outbreak: what the department of radiology should know, Journal Am College Radiol, (2020)
[3]  
Baud D., Qi X., Nielsen- S.K., Pomar Leo M.D., Guillaume F., Real estimates of mortality following covid-19 infection, Lancet Infectious Dis, (2020)
[4]  
Cucinotta D., Vanelli M., Who declares covid-19 a pandemic, Acta bio-medica, 91, 1, pp. 157-160, (2020)
[5]  
Jandric P., Postdigital research in the time of covid-19, Postdigital Sci Educ, pp. 1-6, (2020)
[6]  
Nuno F., Economic effects of coronavirus outbreak (covid-19) on the world economy, SSRN, 2020
[7]  
Oran D.P., Topol E.J., Prevalence of asymptomatic sars-cov-2 infection: a narrative review, Ann Internal Med, (2020)
[8]  
Sheikh A., Sheikh A., Sheikh Z., Dhami S., Sridhar D., What’s the way out? potential exit strategies from the covid-19 lockdown, J Global Health, 10, 1, (2020)
[9]  
Wang W., Yanli X., Gao R., Roujian L., Han K., Guizhen W., Tan W., Detection of sars-cov-2 in different types of clinical specimens, Jama, 323, 18, pp. 1843-1844, (2020)
[10]  
Fang Y., Zhang H., Xie J., Lin M., Ying L., Pang P., Ji W., Sensitivity of chest ct for covid-19: comparison to rt-pcr, Radiology, (2020)