Classifying chest CT images as COVID-19 positive/negative using a convolutional neural network ensemble model and uniform experimental design method

BMC Bioinformatics

Jinn-Tsong Tsai¹  Wen-Hsien Ho²  Yenming J. Chen³  Yao-Mei Chen⁴ 

[1] Department of Healthcare Administration and Medical Informatics, Kaohsiung Medical University, 807, Kaohsiung, Taiwan;Department of Computer Science, National Pingtung University, 900, Pingtung, Taiwan;Department of Healthcare Administration and Medical Informatics, Kaohsiung Medical University, 807, Kaohsiung, Taiwan;Department of Medical Research, Kaohsiung Medical University Hospital, 807, Kaohsiung, Taiwan;Management School, National Kaohsiung University of Science and Technology, 824, Kaohsiung, Taiwan;School of Nursing, Kaohsiung Medical University, 807, Kaohsiung, Taiwan;Superintendent Office, Kaohsiung Medical University Hospital, 807, Kaohsiung, Taiwan;

关键词: COVID-19;    Chest computed tomography image;    Convolutional neural network;    Algorithm hyperparameter;    Ensemble model;   

DOI  :  10.1186/s12859-021-04083-x

来源: Springer

【 摘 要 】

BackgroundTo classify chest computed tomography (CT) images as positive or negative for coronavirus disease 2019 (COVID-19) quickly and accurately, researchers attempted to develop effective models by using medical images.ResultsA convolutional neural network (CNN) ensemble model was developed for classifying chest CT images as positive or negative for COVID-19. To classify chest CT images acquired from COVID-19 patients, the proposed COVID19-CNN ensemble model combines the use of multiple trained CNN models with a majority voting strategy. The CNN models were trained to classify chest CT images by transfer learning from well-known pre-trained CNN models and by applying their algorithm hyperparameters as appropriate. The combination of algorithm hyperparameters for a pre-trained CNN model was determined by uniform experimental design. The chest CT images (405 from COVID-19 patients and 397 from healthy patients) used for training and performance testing of the COVID19-CNN ensemble model were obtained from an earlier study by Hu in 2020. Experiments showed that, the COVID19-CNN ensemble model achieved 96.7% accuracy in classifying CT images as COVID-19 positive or negative, which was superior to the accuracies obtained by the individual trained CNN models. Other performance measures (i.e., precision, recall, specificity, and F1-score) obtained bythe COVID19-CNN ensemble model were higher than those obtained by individual trained CNN models.ConclusionsThe COVID19-CNN ensemble model had superior accuracy and excellent capability in classifying chest CT images as COVID-19 positive or negative.

【 授权许可】

CC BY