COVID-DETECT: A DEEP LEARNING APPROACH FOR CLASSIFICATION OF COVID-19 PNEUMONIA FROM LUNG SEGMENTED CHEST X-RAYS
Abstract
The novel coronavirus (COVID-19) was first reported in the Wuhan City of China in 2019 and became a pandemic. The outbreak has caused shocking effects to the people across the globe. It is important to screen a majority of the population in every country and for the respective governments to take appropriate action. There is a need for a rapid screening system to triage and recommend the patients for appropriate treatment. Chest X-ray imaging is one of the potential modalities, which has ample advantages such as wide availability even in the villages, portability, fast data sharing option from the point of capturing to the point of investigation, etc. The aim of the proposed work is to develop a deep learning algorithm for screening COVID-19 cases by leveraging the widely available X-ray imaging. We have built a deep learning Convolutional Neural Network model utilizing a combination of the public domain (open-source COVID-19) and private data (pneumonia and normal cases). The dataset was used before and after the segmentation of the lung region for training and testing. The outcome of the classification after lung segmentation resulted in significant superiority. The average accuracy achieved by the proposed system was 96%. The heat maps incorporated in the system were helpful for our radiologists to cross-verify whether the appropriate features are identified. This system (COVID-Detect) can be used in remote places in the countries affected by COVID-19 for mass screening of suspected cases and suggesting appropriate actions, such as recommending confirmatory tests.
References
- 1. Wang L, Wong A, COVID-Net: A tailored deep convolutional neural network design for detection of COVID-19 cases from chest radiography images, preprint, arXiv:2003.09871 [eess.IV], 2020. Google Scholar
- 2. , Detection of SARS-CoV-2 in different types of clinical specimens, JAMA 323 :1843–1844, 2020. ISI, Google Scholar
- 3. , The potential dangers of artificial intelligence for radiology and radiologists, J Am Coll Radiol 17 :1309–1311, 2020. Crossref, ISI, Google Scholar
- 4. , Coronavirus (COVID-19) outbreak: What the Department of Radiology should know, J Am Coll Radiol 17 :447–451, 2020. Crossref, ISI, Google Scholar
- 5. , Two public chest X-ray datasets for computer-aided screening of pulmonary diseases, Quant Imaging Med Surg 4 :475–477, 2014. Google Scholar
- 6. , Augmenting the National Institutes of Health chest radiograph dataset with expert annotations of possible pneumonia, Radiol, Artif Intell 1 :e180041, 2019. Crossref, Google Scholar
- 7. , ChestX-Ray8: Hospital-scale chest X-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases, Proc IEEE Conf Computer Vision and Pattern Recognition, pp. 2097–2106, 2017. Crossref, Google Scholar
- 8. , Edge detection of images based on improved Sobel operator and genetic algorithms, Proc 2009 Int Conf Image Analysis and Signal Processing, pp. 31–35, 2009. Google Scholar
- 9. , Edge detection in noisy images using fuzzy reasoning, Proc IEEE Instrumentation and Measurement Technology Conf: Where Instrumentation is Going, pp. 369–372, 1998. Crossref, Google Scholar
- 10. , Edge detection in images using Marr-Hildreth filtering techniques, J Neurosci Methods 26 :75–81, 1988. Crossref, ISI, Google Scholar
- 11. , Multiscale corner detection of gray level images based on log-Gabor wavelet transform, IEEE Trans Circuits Syst Video Technol 17 :868–875, 2007. Crossref, ISI, Google Scholar
- 12. , Gray-level corner detection, Pattern Recognit Lett 1 :95–102, 1982. Crossref, ISI, Google Scholar
- 13. , A fuzzy-based histogram analysis technique for skin lesion discrimination in dermatology clinical images, Comput Med Imaging Graph 27 :387–396, 2003. Crossref, ISI, Google Scholar
- 14. , Feature extraction of kidney ultrasound images based on intensity histogram and gray level co-occurrence matrix, Proc 2012 Sixth Asia Modelling Symp, pp. 115–120, 2012. Crossref, Google Scholar
- 15. , PyTorch: An imperative style, high-performance deep learning library, in Advances in Neural Information Processing Systems, Vol. 32, Curran Associates, Inc., Red Hook, pp. 8026–8037, 2019. Google Scholar
- 16. , Toolkits and libraries for deep learning, J Digit Imaging 30 :400–405, 2017. Crossref, ISI, Google Scholar
- 17. , Hands-On Transfer Learning with Python: Implement Advanced Deep Learning and Neural Network Models using TensorFlow and Keras, Packt Publishing, Birmingham, 2018. Google Scholar
- 18. , Various frameworks and libraries of machine learning and deep learning: A survey, Arch Comput Methods Eng, 2019, https://doi.org/10.1007/s11831-018-09312-w. Google Scholar
- 19. , Traffic-sign detection and classification in the wild, Proc 2016 IEEE Conf Computer Vision and Pattern Recognition, pp. 2110–2118, 2016. Crossref, Google Scholar
- 20. , HCP: A flexible CNN framework for multi-label image classification, IEEE Trans Pattern Anal Mach Intell 38 :1901–1907, 2015. Crossref, ISI, Google Scholar
- 21. , Bounding box-based technique for pavement crack classification and measurement using 1 mm 3D laser data, J Comput Civ Eng 30 :04016011, 2016. Crossref, ISI, Google Scholar
- 22. , Deep residual learning for image recognition, Proc 2016 IEEE Conf Computer Vision and Pattern Recognition, pp. 770–778, 2016. Crossref, Google Scholar
- 23. Cohen JP, Morrison P, Dao L, COVID-19 image data collection, preprint, arXiv:2003.11597 [eess.IV], 2020. Google Scholar
- 24. Radiological Society of North America, RSNA pneumonia detection challenge, https://www.kaggle.com/c/rsnapneumonia-detection-challenge/data, 2019. Google Scholar
- 25. , Corona virus disease 2019 (COVID-19): A systematic review of imaging findings in 919 patients, Am J Roentgenol 215 :87–93, 2020. Crossref, ISI, Google Scholar
- 26. , Frequency and distribution of chest radiographic findings in COVID-19 positive patients, Radiology 296 :E72–E78, 2020. Crossref, ISI, Google Scholar
- 27. , Automated detection of COVID-19 cases using deep neural networks with X-ray images, Comput Biol Med 121 :103792, 2020. Crossref, ISI, Google Scholar
- 28. Hemdan EED, Shouman MA, Karar ME, COVIDX-Net: A framework of deep learning classifiers to diagnose COVID-19 in X-ray images, preprint, arXiv:2003. 11055 [eess.IV], 2020. Google Scholar
- 29. Sethy PK, Behera SK, Detection of coronavirus disease (COVID-19) based on deep features, Preprints, 2020030300, 2020. doi:10.20944/preprints202003.0300.v1. Google Scholar
- 30. Apostolopoulos ID, Bessiana T, COVID-19: Automatic detection from X-ray images utilizing transfer learning with convolutional neural networks, preprint, arXiv:2003.11617 [eess.IV], 2020. Google Scholar