论文标题
自适应特征选择引导深森林,用于与胸部CT分类的COVID-19分类
Adaptive Feature Selection Guided Deep Forest for COVID-19 Classification with Chest CT
论文作者
论文摘要
胸部计算机断层扫描(CT)成为有效的工具,可以帮助诊断冠状病毒199(Covid-19)。由于全球Covid-19的爆发,基于CT图像的COVID-19进行了计算的诊断技术,可以在很大程度上减轻临床医生的负担。在本文中,我们提出了一种自适应特征选择,以基于胸部CT图像为基于胸部CT图像的COVID-19分类。具体而言,我们首先从CT图像中提取特定于位置的特征。然后,为了通过相对较小的数据来捕获这些特征的高级表示,我们利用深森林模型来学习特征的高级表示。此外,我们提出了一种基于受过训练的深森林模型的特征选择方法,以减少特征的冗余,其中特征选择可以与COVID-19分类模型自适应合并。我们评估了1495例COVID-19患者和1027名社区后获得的肺炎患者(CAP),评估了1495例COVID-19数据集的AFS-DF。通过我们的方法实现的准确性(ACC),灵敏度(SEN),特异性(SPE)和AUC分别为91.79%,93.05%,89.95%和96.35%。与4种广泛使用的机器学习方法相比,COVID-19数据集的实验结果表明,所提出的AFS-DF与CAP分类相比,在Covid-19与CAP分类方面取得了出色的性能。
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) and AUC achieved by our method are 91.79%, 93.05%, 89.95% and 96.35%, 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.