Texture feature-based machine learning classifier could assist in the diagnosis of COVID-19

Abstract

Purpose: Differentiating COVID-19 from other acute infectious pneumonias rapidly is challenging at present. This study aims to improve the diagnosis of COVID-19 using computed tomography (CT).

Method: COVID-19 was confirmed mainly by virus nucleic acid testing and epidemiological history according to WHO interim guidance, while other infectious pneumonias were diagnosed by antigen testing. The texture features were extracted from CT images by two radiologists with 5 years of work experience using modified wavelet transform and matrix computation analyses. The random forest (RF) classifier was applied to identify COVID-19 patients and images.

Results: We retrospectively analysed the data of 95 individuals (291 images) with COVID-19 and 96 individuals (279 images) with other acute infectious pneumonias, including 50 individuals (160 images) with influenza A/B. In total, 6 texture features showed a positive association with COVID-19, while 4 features were negatively associated. The mean AUROC, accuracy, sensitivity, and specificity values of the 5-fold test sets were 0.800, 0.722, 0.770, and 0.680 for image classification and 0.858, 0.826, 0.809, and 0.842 for individual classification, respectively. The feature 'Correlation' contributed most both at the image level and individual level, even compared with the clinical factors. In addition, the texture features could discriminate COVID-19 from influenza A/B, with an AUROC of 0.883 for images and 0.957 for individuals.

Conclusions: The developed texture feature-based RF classifier could assist in the diagnosis of COVID-19, which could be a rapid screening tool in the era of pandemic.

Keywords: Computed tomography; Coronavirus disease 2019; Machine learning; Texture analysis.