A semi-supervised learning approach for COVID-19 detection from chest CT scans

被引:14
作者
Zhang, Yong [1 ,2 ]
Su, Li [1 ,2 ]
Liu, Zhenxing [1 ,2 ]
Tan, Wei [4 ]
Jiang, Yinuo [3 ]
Cheng, Cheng [3 ]
机构
[1] Wuhan Univ Sci & Technol, Sch Informat Sci & Engn, Wuhan 430081, Peoples R China
[2] Minist Educ, Engn Res Ctr Met Automat & Measurement Technol, Wuhan 430081, Peoples R China
[3] Huazhong Univ Sci & Technol, Sch Artificial Intelligence & Automat, Wuhan 430074, Peoples R China
[4] Wuhan Univ Sci & Technol, Hosp WUST, Wuhan 430081, Peoples R China
基金
中国国家自然科学基金;
关键词
COVID-19; Computed tomography; Semi-supervised learning; Deep learning; Attention mechanisms;
D O I
10.1016/j.neucom.2022.06.076
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
COVID-19 has spread rapidly all over the world and has infected more than 200 countries and regions. Early screening of suspected infected patients is essential for preventing and combating COVID-19. Computed Tomography (CT) is a fast and efficient tool which can quickly provide chest scan results. To reduce the burden on doctors of reading CTs, in this article, a high precision diagnosis algorithm of COVID-19 from chest CTs is designed for intelligent diagnosis. A semi-supervised learning approach is developed to solve the problem when only small amount of labelled data is available. While following the MixMatch rules to conduct sophisticated data augmentation, we introduce a model training technique to reduce the risk of model over-fitting. At the same time, a new data enhancement method is proposed to modify the regularization term in MixMatch. To further enhance the generalization of the model, a convolutional neural network based on an attention mechanism is then developed that enables to extract multi-scale features on CT scans. The proposed algorithm is evaluated on an independent CT data-set of the chest from COVID-19 and achieves the area under the receiver operating characteristic curve (AUC) value of 0.932, accuracy of 90.1%, sensitivity of 91.4%, specificity of 88.9%, and F1-score of 89.9%. The results show that the proposed algorithm can accurately diagnose whether a chest CT belongs to a positive or negative indication of COVID-19, and can help doctors to diagnose rapidly in the early stages of a COVID-19 outbreak. (C) 2022 Published by Elsevier B.V.
引用
收藏
页码:314 / 324
页数:11
相关论文
共 38 条
[1]   Epidemiology, causes, clinical manifestation and diagnosis, prevention and control of coronavirus disease (COVID-19) during the early outbreak period: a scoping review [J].
Adhikari, Sasmita Poudel ;
Meng, Sha ;
Wu, Yu-Ju ;
Mao, Yu-Ping ;
Ye, Rui-Xue ;
Wang, Qing-Zhi ;
Sun, Chang ;
Sylvia, Sean ;
Rozelle, Scott ;
Raat, Hein ;
Zhou, Huan .
INFECTIOUS DISEASES OF POVERTY, 2020, 9 (01)
[2]   Correlation of Chest CT and RT-PCR Testing for Coronavirus Disease 2019 (COVID-19) in China: A Report of 1014 Cases [J].
Ai, Tao ;
Yang, Zhenlu ;
Hou, Hongyan ;
Zhan, Chenao ;
Chen, Chong ;
Lv, Wenzhi ;
Tao, Qian ;
Sun, Ziyong ;
Xia, Liming .
RADIOLOGY, 2020, 296 (02) :E32-E40
[3]  
[Anonymous], 2014, STRIVING SIMPLICITY
[4]   Representation learning for mammography mass lesion classification with convolutional neural networks [J].
Arevalo, John ;
Gonzalez, Fabio A. ;
Ramos-Pollan, Raul ;
Oliveira, Jose L. ;
Guevara Lopez, Miguel Angel .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2016, 127 :248-257
[5]  
Berthelot D, 2019, ADV NEUR IN, V32
[6]   A familial cluster of pneumonia associated with the 2019 novel coronavirus indicating person-to-person transmission: a study of a family cluster [J].
Chan, Jasper Fuk-Woo ;
Yuan, Shuofeng ;
Kok, Kin-Hang ;
To, Kelvin Kai-Wang ;
Chu, Hin ;
Yang, Jin ;
Xing, Fanfan ;
Liu, Jieling ;
Yip, Cyril Chik-Yan ;
Poon, Rosana Wing-Shan ;
Tsoi, Hoi-Wah ;
Lo, Simon Kam-Fai ;
Chan, Kwok-Hung ;
Poon, Vincent Kwok-Man ;
Chan, Wan-Mui ;
Ip, Jonathan Daniel ;
Cai, Jian-Piao ;
Cheng, Vincent Chi-Chung ;
Chen, Honglin ;
Hui, Christopher Kim-Ming ;
Yuen, Kwok-Yung .
LANCET, 2020, 395 (10223) :514-523
[7]  
Chung MS, 2020, EUR RADIOL, V30, P2182, DOI [10.1148/radiol.2020200230, 10.1007/s00330-019-06574-1]
[8]  
DeVries T, 2017, Arxiv, DOI arXiv:1708.04552
[9]   Sensitivity of Chest CT for COVID-19: Comparison to RT-PCR [J].
Fang, Yicheng ;
Zhang, Huangqi ;
Xie, Jicheng ;
Lin, Minjie ;
Ying, Lingjun ;
Pang, Peipei ;
Ji, Wenbin .
RADIOLOGY, 2020, 296 (02) :E115-E117
[10]   Multi-sensor filtering fusion meets censored measurements under a constrained network environment: advances, challenges and prospects [J].
Geng, Hang ;
Liu, Hongjian ;
Ma, Lifeng ;
Yi, Xiaojian .
INTERNATIONAL JOURNAL OF SYSTEMS SCIENCE, 2021, 52 (16) :3410-3436