Deep Ensemble Learning-Based Models for Diagnosis of COVID-19 from Chest CT Images

被引:13
作者
Mouhafid, Mohamed [1 ]
Salah, Mokhtar [1 ]
Yue, Chi [1 ]
Xia, Kewen [1 ]
机构
[1] Hebei Univ Technol, Sch Elect & Informat Engn, Tianjin 300401, Peoples R China
基金
中国国家自然科学基金;
关键词
coronavirus detection; deep learning; convolutional neural network; transfer learning; stacking; weighted average ensemble; NEURAL-NETWORK; NET;
D O I
10.3390/healthcare10010166
中图分类号
R19 [保健组织与事业(卫生事业管理)];
学科分类号
摘要
Novel coronavirus (COVID-19) has been endangering human health and life since 2019. The timely quarantine, diagnosis, and treatment of infected people are the most necessary and important work. The most widely used method of detecting COVID-19 is real-time polymerase chain reaction (RT-PCR). Along with RT-PCR, computed tomography (CT) has become a vital technique in diagnosing and managing COVID-19 patients. COVID-19 reveals a number of radiological signatures that can be easily recognized through chest CT. These signatures must be analyzed by radiologists. It is, however, an error-prone and time-consuming process. Deep Learning-based methods can be used to perform automatic chest CT analysis, which may shorten the analysis time. The aim of this study is to design a robust and rapid medical recognition system to identify positive cases in chest CT images using three Ensemble Learning-based models. There are several techniques in Deep Learning for developing a detection system. In this paper, we employed Transfer Learning. With this technique, we can apply the knowledge obtained from a pre-trained Convolutional Neural Network (CNN) to a different but related task. In order to ensure the robustness of the proposed system for identifying positive cases in chest CT images, we used two Ensemble Learning methods namely Stacking and Weighted Average Ensemble (WAE) to combine the performances of three fine-tuned Base-Learners (VGG19, ResNet50, and DenseNet201). For Stacking, we explored 2-Levels and 3-Levels Stacking. The three generated Ensemble Learning-based models were trained on two chest CT datasets. A variety of common evaluation measures (accuracy, recall, precision, and F1-score) are used to perform a comparative analysis of each method. The experimental results show that the WAE method provides the most reliable performance, achieving a high recall value which is a desirable outcome in medical applications as it poses a greater risk if a true infected patient is not identified.
引用
收藏
页数:26
相关论文
共 67 条
[31]   Evaluation of deep learning-based approaches for COVID-19 classification based on chest X-ray images [J].
KC, Kamal ;
Yin, Zhendong ;
Wu, Mingyang ;
Wu, Zhilu .
SIGNAL IMAGE AND VIDEO PROCESSING, 2021, 15 (05) :959-966
[32]  
Keskar NS, 2017, IMPROVING GEN PERFOR, DOI 10.1109/ICNN.1994.374153
[33]  
King DB, 2015, ACS SYM SER, V1214, P1, DOI 10.1021/bk-2015-1214.ch001
[34]   A deep transfer learning model with classical data augmentation and CGAN to detect COVID-19 from chest CT radiography digital images [J].
Loey, Mohamed ;
Manogaran, Gunasekaran ;
Khalifa, Nour Eldeen M. .
NEURAL COMPUTING & APPLICATIONS, 2020,
[35]  
Lokwani R., 2020, ARXIV, DOI DOI 10.48550/ARXIV.2020.13212
[36]  
Manning CD., 2008, Introduction to Information Retrieval, V1st, DOI 10.1017CBO9780511809071
[37]   Lung Nodule Classification via Deep Transfer Learning in CT Lung Images [J].
Medeiros da Nobrega, Raul Victor ;
Peixoto, Solon Alves ;
da Silva, Suane Pires P. ;
Reboucas Filho, Pedro Pedrosa .
2018 31ST IEEE INTERNATIONAL SYMPOSIUM ON COMPUTER-BASED MEDICAL SYSTEMS (CBMS 2018), 2018, :244-249
[38]  
Mehra R., 2018, ICT EXPRESS, V4, P247, DOI [DOI 10.1016/j.icte.2018.10.007, 10.1016/j.icte.2018.10.007]
[39]  
Missinglink.ai, CONVOLUTIONAL NEURAL
[40]   Methods for interpreting and understanding deep neural networks [J].
Montavon, Gregoire ;
Samek, Wojciech ;
Mueller, Klaus-Robert .
DIGITAL SIGNAL PROCESSING, 2018, 73 :1-15