期刊
IEEE TRANSACTIONS ON MEDICAL IMAGING
卷 40, 期 3, 页码 879-890出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/TMI.2020.3040950
关键词
Diseases; Lung; COVID-19; X-rays; Anomaly detection; Viruses (medical); Task analysis; Viral pneumonia screening; deep anomaly detection; confidence prediction; chest X-ray
类别
资金
- National Natural Science Foundation of China [61771397]
- Science and Technology Innovation Committee of Shenzhen Municipality, China [JCYJ20180306171334997]
- Innovation Foundation for Doctor Dissertation of Northwestern Polytechnical University [CX202010]
Clusters of viral pneumonia cases over a short period may indicate an outbreak. The study proposes a method to detect viral pneumonia using chest X-rays, which performs well on two different datasets without fine-tuning.
Clusters of viral pneumonia occurrences over a short period may be a harbinger of an outbreak or pandemic. Rapid and accurate detection of viral pneumonia using chest X-rays can be of significant value for large-scale screening and epidemic prevention, particularly when other more sophisticated imaging modalities are not readily accessible. However, the emergence of novel mutated viruses causes a substantial dataset shift, which can greatly limit the performance of classification-based approaches. In this paper, we formulate the task of differentiating viral pneumonia from non-viral pneumonia and healthy controls into a one-class classification-based anomaly detection problem. We therefore propose the confidence-aware anomaly detection (CAAD) model, which consists of a shared feature extractor, an anomaly detection module, and a confidence prediction module. If the anomaly score produced by the anomaly detection module is large enough, or the confidence score estimated by the confidence prediction module is small enough, the input will be accepted as an anomaly case (i.e., viral pneumonia). The major advantage of our approach over binary classification is that we avoid modeling individual viral pneumonia classes explicitly and treat all known viral pneumonia cases as anomalies to improve the one-class model. The proposed model outperforms binary classification models on the clinical X-VIRAL dataset that contains 5,977 viral pneumonia (no COVID-19) cases, 37,393 non-viral pneumonia or healthy cases. Moreover, when directly testing on the X-COVID dataset that contains 106 COVID-19 cases and 107 normal controls without any fine-tuning, our model achieves an AUC of 83.61% and sensitivity of 71.70%, which is comparable to the performance of radiologists reported in the literature.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据