COVID-AL: The diagnosis of COVID-19 with deep active learning

被引:82
作者
Wu, Xing [1 ,2 ]
Chen, Cheng [1 ]
Zhong, Mingyu [1 ]
Wang, Jianjia [1 ,2 ]
Shi, Jun [2 ,3 ]
机构
[1] Shanghai Univ, Sch Comp Engn & Sci, Shanghai, Peoples R China
[2] Shanghai Univ, Shanghai Inst Adv Commun & Data Sci, Shanghai, Peoples R China
[3] Shanghai Univ, Sch Commun & Informat Engn, Key Lab Specialty Fiber Opt & Opt Access Networks, Joint Int Res Lab Specialty Fiber Opt & Adv Commu, Shanghai, Peoples R China
基金
中国国家自然科学基金; 国家重点研发计划;
关键词
COVID-19; Deep active learning; Computer-aided diagnosis; Sample diversity; Predicted loss;
D O I
10.1016/j.media.2020.101913
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The efficient diagnosis of COVID-19 plays a key role in preventing the spread of this disease. The computer-aided diagnosis with deep learning methods can perform automatic detection of COVID-19 using CT scans. However, large scale annotation of CT scans is impossible because of limited time and heavy burden on the healthcare system. To meet the challenge, we propose a weakly-supervised deep active learning framework called COVID-AL to diagnose COVID-19 with CT scans and patient-level labels. The COVID-AL consists of the lung region segmentation with a 2D U-Net and the diagnosis of COVID-19 with a novel hybrid active learning strategy, which simultaneously considers sample diversity and predicted loss. With a tailor-designed 3D residual network, the proposed COVID-AL can diagnose COVID-19 efficiently and it is validated on a large CT scan dataset collected from the CC-CCII. The experimental results demonstrate that the proposed COVID-AL outperforms the state-of-the-art active learning approaches in the diagnosis of COVID-19. With only 30% of the labeled data, the COVID-AL achieves over 95% accuracy of the deep learning method using the whole dataset. The qualitative and quantitative analysis proves the effectiveness and efficiency of the proposed COVID-AL framework. (c) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页数:8
相关论文
共 29 条
[1]   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
[2]   End-to-end lung cancer screening with three-dimensional deep learning on low-dose chest computed tomography [J].
Ardila, Diego ;
Kiraly, Atilla P. ;
Bharadwaj, Sujeeth ;
Choi, Bokyung ;
Reicher, Joshua J. ;
Peng, Lily ;
Tse, Daniel ;
Etemadi, Mozziyar ;
Ye, Wenxing ;
Corrado, Greg ;
Naidich, David P. ;
Shetty, Shravya .
NATURE MEDICINE, 2019, 25 (06) :954-+
[3]   Quantitative CT Analysis of Diffuse Lung Disease [J].
Chen, Alicia ;
Karwoski, Ronald A. ;
Gierada, David S. ;
Bartholmai, Brian J. ;
Koo, Chi Wan .
RADIOGRAPHICS, 2020, 40 (01) :28-43
[4]   Deep learning algorithms for detection of critical findings in head CT scans: a retrospective study [J].
Chilamkurthy, Sasank ;
Ghosh, Rohit ;
Tanamala, Swetha ;
Biviji, Mustafa ;
Campeau, Norbert G. ;
Venugopal, Vasantha Kumar ;
Mahajan, Vidur ;
Rao, Pooja ;
Warier, Prashant .
LANCET, 2018, 392 (10162) :2388-2396
[5]   CT Imaging Features of 2019 Novel Coronavirus (2019-nCoV) [J].
Chung, Michael ;
Bernheim, Adam ;
Mei, Xueyan ;
Zhang, Ning ;
Huang, Mingqian ;
Zeng, Xianjun ;
Cui, Jiufa ;
Xu, Wenjian ;
Yang, Yang ;
Fayad, Zahi A. ;
Jacobi, Adam ;
Li, Kunwei ;
Li, Shaolin ;
Shan, Hong .
RADIOLOGY, 2020, 295 (01) :202-207
[6]   A guide to deep learning in healthcare [J].
Esteva, Andre ;
Robicquet, Alexandre ;
Ramsundar, Bharath ;
Kuleshov, Volodymyr ;
DePristo, Mark ;
Chou, Katherine ;
Cui, Claire ;
Corrado, Greg ;
Thrun, Sebastian ;
Dean, Jeff .
NATURE MEDICINE, 2019, 25 (01) :24-29
[7]   Clinical Characteristics of Coronavirus Disease 2019 in China [J].
Guan, W. ;
Ni, Z. ;
Hu, Yu ;
Liang, W. ;
Ou, C. ;
He, J. ;
Liu, L. ;
Shan, H. ;
Lei, C. ;
Hui, D. S. C. ;
Du, B. ;
Li, L. ;
Zeng, G. ;
Yuen, K. -Y. ;
Chen, R. ;
Tang, C. ;
Wang, T. ;
Chen, P. ;
Xiang, J. ;
Li, S. ;
Wang, Jin-lin ;
Liang, Z. ;
Peng, Y. ;
Wei, L. ;
Liu, Y. ;
Hu, Ya-hua ;
Peng, P. ;
Wang, Jian-ming ;
Liu, J. ;
Chen, Z. ;
Li, G. ;
Zheng, Z. ;
Qiu, S. ;
Luo, J. ;
Ye, C. ;
Zhu, S. ;
Zhong, N. .
NEW ENGLAND JOURNAL OF MEDICINE, 2020, 382 (18) :1708-1720
[8]   Learning Spatio-Temporal Features with 3D Residual Networks for Action Recognition [J].
Hara, Kensho ;
Kataoka, Hirokatsu ;
Satoh, Yutaka .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW 2017), 2017, :3154-3160
[9]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[10]  
Kiser K., 2020, DATA THORACIC VOLUME