论文标题

自适应特征选择引导深森林,用于与胸部CT分类的COVID-19分类

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

论文作者

Sun, Liang, Mo, Zhanhao, Yan, Fuhua, Xia, Liming, Shan, Fei, Ding, Zhongxiang, Shao, Wei, Shi, Feng, Yuan, Huan, Jiang, Huiting, Wu, Dijia, Wei, Ying, Gao, Yaozong, Gao, Wanchun, Sui, He, Zhang, Daoqiang, Shen, Dinggang

论文摘要

胸部计算机断层扫描(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.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源