Momentum contrastive learning for few-shot COVID-19 diagnosis from chest CT images

PATTERN RECOGNITION

卷:113

Article

Chen, Xiaocong¹  Yao, Lina¹  Zhou, Tao²  Dong, Jinming¹  Zhang, Yu³ 

[1] Univ New South Wales, Sch Comp Sci & Engn, Sydney, NSW 2052, Australia

[2] Incept Inst Artificial Intelligence, Abu Dhabi, U Arab Emirates

[3] Lehigh Univ, Dept Bioengn, Bethlehem, PA 18015 USA

关键词: COVID-19 diagnosis;    Few-shot learning;    Contrastive learning;    Chest CT images;   

DOI  :  10.1016/j.patcog.2021.107826

来源: Elsevier

【 摘 要 】

The current pandemic, caused by the outbreak of a novel coronavirus (COVID-19) in December 2019, has led to a global emergency that has significantly impacted economies, healthcare systems and personal wellbeing all around the world. Controlling the rapidly evolving disease requires highly sensitive and specific diagnostics. While RT-PCR is the most commonly used, it can take up to eight hours, and requires significant effort from healthcare professionals. As such, there is a critical need for a quick and automatic diagnostic system. Diagnosis from chest CT images is a promising direction. However, current studies are limited by the lack of sufficient training samples, as acquiring annotated CT images is time-consuming. To this end, we propose a new deep learning algorithm for the automated diagnosis of COVID-19, which only requires a few samples for training. Specifically, we use contrastive learning to train an encoder which can capture expressive feature representations on large and publicly available lung datasets and adopt the prototypical network for classification. We validate the efficacy of the proposed model in comparison with other competing methods on two publicly available and annotated COVID-19 CT datasets. Our results demonstrate the superior performance of our model for the accurate diagnosis of COVID-19 based on chest CT images. (c) 2021 Elsevier Ltd. All rights reserved.

【 授权许可】

Free