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 条
[61]  
King G., Zeng L., Logistic regression in rare events data, Political Anal, 9, 2, pp. 137-163, (2001)
[62]  
COVID-19 Database, (2020)
[63]  
Cohen J.P., Morrison P., Dao L., Covid-19 image data collection. arXiv, 2003, (2020)
[64]  
Kermany D., Zhang K., Goldbaum M.U., Labeled optical coherence tomography (oct) and chest x-ray images for classification. Mendeley data, 2, (2018)
[65]  
Kingma D.P., Ba J., Adam: A method for stochastic optimization, (2014)
[66]  
Selvaraju R.R., Cogswell M., Das A., Vedantam R., Parikh D., Batra D., Grad-cam: Visual explanations from deep networks via gradient-based localization, Proceedings of the IEEE international conference on computer vision
[67]  
, pp. 618-626, (2017)
[68]  
Narin A., Kaya C., Pamuk Z., Automatic detection of coronavirus disease (covid-19) using x-ray images and deep convolutional neural networks. arXiv preprint arXiv, 2003, (2020)