CNN Ensemble Approach to Detect COVID-19 from Computed Tomography Chest Images
Source abstract
Abstract
In January 2020, the World Health Organization declared a global health emergency concerning the spread of a new coronavirus disease, which was later named COVID-19. Early and fast diagnosis and isolation of COVID-19 patients have proven to be instrumental in limiting the spread of the disease. Computed tomography (CT) is a promising imaging method for fast diagnosis of COVID-19. In this study, we develop a unique preprocessing step to resize CT chest images to a fixed size (256 × 256 pixels) that preserves the aspect ratio and reduces image loss. Then, we present a deep learning (DL) method to classify CT chest images based on the light-weight pre-trained EfficientNet-B3 CNN model and ensemble techniques. The proposed method, which we refer to as EfficientNet-B3-GAP-Ensemble, comprises an ensemble of a modified version of the EfficientNet-B3. We build the ensemble using multiple runs and multiple training epochs. We test the EfficientNet-B3-GAP-Ensemble on two common benchmark datasets, i.e., the COVID19-CT and SARS-CoV-2-CT datasets. The proposed method has outperformed state-of-the-art methods for both datasets. For the COVID19-CT dataset, it achieved 88.18% sensitivity, 88.29% precision, 88.18% accuracy, an F1-score of 88.15%, and AUC of 92.10%. With the SARS-CoV-2-CT dataset, we tested the proposed method under different train-test splits, i.e., 20%–80%, 50%–50%, and 80%–20%. For the latter split, the proposed method achieved 99.72% accuracy, 99.80% sensitivity, precision, and F1-scores, and an AUC score of 99.99%.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: COVID-19 diagnosis using AI · Radiomics and Machine Learning in Medical Imaging · Artificial Intelligence in Healthcare and Education
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (33)
Show source-provided references
1. Weekly operational update on COVID-19 September 27, 2020
2. Multi-tiered screening and diagnosis strategy for COVID-19: A model for sustainable testing capacity in response to pandemic10.1080/07853890.2020.1763449 ↗
3. Correlation of chest CT and RT-PCR testing for coronavirus disease 2019 (COVID-19) in China: A report of 1014 cases10.1148/radiol.2020200642 ↗
4. Chest CT for detecting COVID-19: A systematic review and meta-analysis of diagnostic accuracy10.1007/s00330-020-06934-2 ↗
5. Diagnosing COVID-19 from chest X-ray in resource limited environment-case report
6. A state-of-the-art survey on deep learning theory and architectures
7. D. Chen, S. Ji, F. Liu1, Z. Li and X. Zhou, “A review of automated diagnosis of COVID-19 based on scanning images,” arXiv., arXiv:2006.05245 [eess.IV], June 09, 2020.10.1145/3449301.3449778 ↗
8. J. Zhao, X. He, X. Yang, Y. Zhang, S. Zhang et al., “COVID-CT-dataset: A CT scan dataset about COVID-19,” arXiv., arXiv: 2003.13865 [eess.IV], March 30, 2020.
9. E. Soares, P. Angelov, S. Biaso, M. H. Froes and D. K. Abe, “SARS-CoV-2 CT-scan dataset: A large dataset of real patients CT scans for SARS-CoV-2 identification,” medRxiv, no. 2020.04.24.20078584, 2020. https://doi.org/10.1101/2020.04.24.20078584.10.1101/2020.04.24.20078584 ↗
10. E. Soares, P. Angelov, S. Biaso, M. H. Froes and D. K. Abe, “SARS-COV-2 Ct-Scan dataset|Kaggle,” 2020. [Online]. Available: https://www.kaggle.com/plameneduardo/sarscov2-ctscan-dataset.
11. S. Wang, B. Kang, J. Ma, X. Zeng, M. Xiao et al., “A deep learning algorithm using CT images to screen for Corona Virus Disease (COVID-19),” medRxiv, no. 2020.02.14.20023028, 2020. https://doi.org/10.1101/2020.02.14.20023028.10.1101/2020.02.14.20023028 ↗
12. S. Ying, S. Zheng, L. Li, X. Zhang, X. Zhang et al., “Deep learning enables accurate diagnosis of novel coronavirus (COVID-19) with CT images,” medRxiv, no. 2020.02.23.20026930, February 25, 2020. https://doi.org/10.1101/2020.02.23.20026930.10.1101/2020.02.23.20026930 ↗
13. X. He, X. Yang, S. Zhang, J. Zhao, Y. Zhang et al., “Sample-efficient deep learning for COVID-19 diagnosis based on CT scans,” medRxiv, no. 2020.04.13.20063941, 2020. https://doi.org/10.1101/2020.04.13.20063941.10.1101/2020.04.13.20063941 ↗
14. A. Mobiny, P. A. Cicalese, S. Zare, P. Yuan, M. Abavisani et al., “Radiologist-Level COVID-19 detection using CT scans with detail-oriented capsule networks,” arXiv., arXiv:2004.07407 [eess.IV], April 16, 2020.
15. A light CNN for detecting COVID-19 from CT scans of the chest10.1016/j.patrec.2020.10.001 ↗
16. P. Angelov and E. Soares, “Explainable-by-design approach for COVID-19 classification via CT-scan,” medRxiv, no. 2020.04.24.20078584, 2020. https://doi.org/10.1101/2020.04.24.20078584.10.1101/2020.04.24.20078584 ↗
17. Classification of the COVID-19 infected patients using DenseNet201 based deep transfer learning
18. COVID-19 detection in CT images with deep learning: A voting-based scheme and cross-datasets analysis10.1016/j.imu.2020.100427 ↗
19. Weakly supervised deep learning for COVID-19 infection detection and classification from CT images10.1109/ACCESS.2020.3005510 ↗
20. Artificial intelligence for the detection of COVID-19 pneumonia on chest CT using multinational datasets10.1038/s41467-020-17971-2 ↗
21. A deep learning system to screen novel coronavirus disease 2019 pneumonia10.1016/j.eng.2020.04.010 ↗
22. EfficientNet: Rethinking model scaling for convolutional neural networks
23. ImageNet large scale visual recognition challenge10.1007/s11263-015-0816-y ↗
24. Going deeper with convolutions
25. Variability and reproducibility in deep learning for medical image segmentation10.1038/s41598-020-69920-0 ↗
26. Stacked regressions10.1007/BF00117832 ↗
27. Stacked generalization10.1016/S0893-6080(05)80023-1 ↗
28. Stacked generalization: An introduction to super learning10.1007/s10654-018-0390-z ↗
29. A review of ensemble methods in bioinformatics10.2174/157489310794072508 ↗
30. A hybrid knowledge and ensemble classification approach for prediction of venous thromboembolism10.1111/exsy.12388 ↗
31. Enhanced skin condition prediction through machine learning using dynamic training and testing tugmentation10.1109/ACCESS.2020.2976045 ↗
32. J. Zhao, X. He, X. Yang, Y. Zhang, S. Zhang et al., “COVID-CT-Dataset: A CT scan dataset about COVID-19,” 2020. [Online]. Available: https://github.com/UCSD-AI4H/COVID-CT.
33. COVID-19 CT image datasets