Adaptive Feature Selection Guided Deep Forest for COVID-19 Classification With Chest CT

IEEE J Biomed Health Inform. 2020 Oct;24(10):2798-2805. doi: 10.1109/JBHI.2020.3019505. Epub 2020 Aug 26.

Abstract

Chest computed tomography (CT) becomes an effective tool to assist the diagnosis of coronavirus disease-19 (COVID-19). Due to the outbreak of COVID-19 worldwide, using the computed-aided diagnosis technique for COVID-19 classification based on CT images could largely alleviate the burden of clinicians. In this paper, we propose an Adaptive Feature Selection guided Deep Forest (AFS-DF) for COVID-19 classification based on chest CT images. Specifically, we first extract location-specific features from CT images. Then, in order to capture the high-level representation of these features with the relatively small-scale data, we leverage a deep forest model to learn high-level representation of the features. Moreover, we propose a feature selection method based on the trained deep forest model to reduce the redundancy of features, where the feature selection could be adaptively incorporated with the COVID-19 classification model. We evaluated our proposed AFS-DF on COVID-19 dataset with 1495 patients of COVID-19 and 1027 patients of community acquired pneumonia (CAP). The accuracy (ACC), sensitivity (SEN), specificity (SPE), AUC, precision and F1-score achieved by our method are 91.79%, 93.05%, 89.95%, 96.35%, 93.10% and 93.07%, respectively. Experimental results on the COVID-19 dataset suggest that the proposed AFS-DF achieves superior performance in COVID-19 vs. CAP classification, compared with 4 widely used machine learning methods.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Betacoronavirus*
  • COVID-19
  • COVID-19 Testing
  • Clinical Laboratory Techniques / statistics & numerical data*
  • Computational Biology
  • Coronavirus Infections / classification
  • Coronavirus Infections / diagnosis*
  • Coronavirus Infections / diagnostic imaging*
  • Databases, Factual / statistics & numerical data
  • Deep Learning
  • Humans
  • Neural Networks, Computer
  • Pandemics / classification
  • Pneumonia, Viral / classification
  • Pneumonia, Viral / diagnosis*
  • Pneumonia, Viral / diagnostic imaging*
  • Radiographic Image Interpretation, Computer-Assisted / statistics & numerical data
  • Radiography, Thoracic / statistics & numerical data
  • SARS-CoV-2
  • Tomography, X-Ray Computed / statistics & numerical data*