Diagnosis of COVID-19 Cases from Chest X-ray Images Using Deep Neural Network and LightGBM

Ezzoddin, Mobina and Nasiri, Hamid and Dorrigiv, Morteza (2022) Diagnosis of COVID-19 Cases from Chest X-ray Images Using Deep Neural Network and LightGBM. In: Proceedings of 2022 12th Iranian/2nd International Conference on Machine Vision and Image Processing, MVIP 2022 :. Iranian Conference on Machine Vision and Image Processing, MVIP . IEEE Computer Society Press, IRN. ISBN 9781665412162

Abstract

The Coronavirus was detected in Wuhan, China in late 2019 and then led to a pandemic with a rapid worldwide outbreak. The number of infected people has been swiftly increasing since then. Therefore, in this study, an attempt was made to propose a new and efficient method for automatic diagnosis of Corona disease from X-ray images using Deep Neural Networks (DNNs). In the proposed method, the DensNet169 was used to extract the features of the patients' Chest X-Ray (CXR) images. The extracted features were given to a feature selection algorithm (i.e., ANOVA) to select a number of them. Finally, the selected features were classified by LightGBM algorithm. The proposed approach was evaluated on the ChestX-ray8 dataset and reached 99.20% and 94.22% accuracies in the two-class (i.e., COVID-19 and No-findings) and multi-class (i.e., COVID-19, Pneumonia, and No-findings) classification problems, respectively.