EDL-COVID: Ensemble Deep Learning for COVID-19 Case Detection From Chest X-Ray Images

IEEE Trans Industr Inform. 2021 Feb 8;17(9):6539-6549. doi: 10.1109/TII.2021.3057683. eCollection 2021 Sep.

Abstract

Effective screening of COVID-19 cases has been becoming extremely important to mitigate and stop the quick spread of the disease during the current period of COVID-19 pandemic worldwide. In this article, we consider radiology examination of using chest X-ray images, which is among the effective screening approaches for COVID-19 case detection. Given deep learning is an effective tool and framework for image analysis, there have been lots of studies for COVID-19 case detection by training deep learning models with X-ray images. Although some of them report good prediction results, their proposed deep learning models might suffer from overfitting, high variance, and generalization errors caused by noise and a limited number of datasets. Considering ensemble learning can overcome the shortcomings of deep learning by making predictions with multiple models instead of a single model, we propose EDL-COVID, an ensemble deep learning model employing deep learning and ensemble learning. The EDL-COVID model is generated by combining multiple snapshot models of COVID-Net, which has pioneered in an open-sourced COVID-19 case detection method with deep neural network processed chest X-ray images, by employing a proposed weighted averaging ensembling method that is aware of different sensitivities of deep learning models on different classes types. Experimental results show that EDL-COVID offers promising results for COVID-19 case detection with an accuracy of 95%, better than COVID-Net of 93.3%.

Keywords: Covid-19; EDL-COVID; chest X-ray images; deep learning; ensemble learning.

Grants and funding

This work was supported in part by the National the Natural Science Foundation of China under Grant 61972277, in part by Tianjin Natural Science Foundation under Grant 18JCZDJC30800, in part by National Natural Science Foundation of China under Grant 62071343 and Grant 51609195, and in part by the Open Research Project of the State Key Laboratory of Industrial Control Technology, Zhejiang University, China under Grant ICT20044.