| Literature DB >> 33285482 |
Xing Wu1, Cheng Chen2, Mingyu Zhong2, Jianjia Wang3, Jun Shi4.
Abstract
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.Entities:
Keywords: COVID-19; Computer-aided diagnosis; Deep active learning; Predicted loss; Sample diversity
Year: 2020 PMID: 33285482 PMCID: PMC7689310 DOI: 10.1016/j.media.2020.101913
Source DB: PubMed Journal: Med Image Anal ISSN: 1361-8415 Impact factor: 8.545
Fig. 1The knowledge transfer for lung region segmentation.
Fig. 2The network architecture for lung region segmentation.
Fig. 3The COVID-AL framework for weakly-supervised active learning.
Algorithm 1COVID-AL: Active learning framework for COVID-19 diagnosis.
Fig. 4The network architecture of COVID-AL.
The performance of the active learning methods.
| Method | Accuracy | PR-AUC | ROC-AUC |
|---|---|---|---|
| RAND | 0.797 | 0.839 | 0.909 |
| ENT | 0.829 | 0.902 | 0.935 |
| AFT*-DIV | 0.850 | 0.929 | 0.955 |
| LP | 0.856 | 0.939 | 0.955 |
| CSET | 0.840 | 0.928 | 0.940 |
| OMEDAL | 0.840 | 0.939 | 0.951 |
| COVID-AL |
Fig. 5Comparison of the proposed COVID-AL with the state-of-the-art active learning methods using the accuracy.
Fig. 6Comparison of the proposed COVID-AL with the state-of-the-art active learning methods using the precision-recall curve.
Fig. 7Comparison of the proposed COVID-AL with the state-of-the-art active learning methods using the ROC curve.