Classification of COVID-19 Chest CT Images Based on Ensemble Deep Learning.

Journal: Journal of healthcare engineering
Published Date:

Abstract

Novel coronavirus pneumonia (NCP) has become a global pandemic disease, and computed tomography-based (CT) image analysis and recognition are one of the important tools for clinical diagnosis. In order to assist medical personnel to achieve an efficient and fast diagnosis of patients with new coronavirus pneumonia, this paper proposes an assisted diagnosis algorithm based on ensemble deep learning. The method combines the Stacked Generalization ensemble learning with the VGG16 deep learning to form a cascade classifier, and the information constituting the cascade classifier comes from multiple subsets of the training set, each of which is used to collect deviant information about the generalization behavior of the data set, such that this deviant information fills the cascade classifier. The algorithm was experimentally validated for classifying patients with novel coronavirus pneumonia, patients with common pneumonia (CP), and normal controls, and the algorithm achieved a prediction accuracy of 93.57%, sensitivity of 94.21%, specificity of 93.93%, precision of 89.40%, and F1-score of 91.74% for the three categories. The results show that the method proposed in this paper has good classification performance and can significantly improve the performance of deep neural networks for multicategory prediction tasks.

Authors

  • Xiaoshuo Li
    Key Laboratory of Intelligent Computing in Medical Image, Ministry of Education, Shenyang, China.
  • Wenjun Tan
    Key Laboratory of Medical Image Computing, Ministry of Education, Shenyang, China.
  • Pan Liu
    Jiangsu Key Laboratory of Urban ITS, Southeast University, Si Pai Lou #2, Nanjing, 210096, China; Jiangsu Province Collaborative Innovation Center of Modern Urban Traffic Technologies, Si Pai Lou #2, Nanjing, 210096, China. Electronic address: pan_liu@hotmail.com.
  • Qinghua Zhou
    School of Informatics, University of Leicester, Leicester, LE1 7RH, UK.
  • Jinzhu Yang
    College of Information Science and Engineering, Northeastern University, 110819, Shenyang, China.