Sample-Efficient Deep Learning for COVID-19 Diagnosis Based on CT Scans
He, X.; Yang, X.; Zhang, S.; Zhao, J.; Zhang, Y.; Xing, E.; Xie, P.
Show abstract
Coronavirus disease 2019 (COVID-19) has infected more than 1.3 million individuals all over the world and caused more than 106,000 deaths. One major hurdle in controlling the spreading of this disease is the inefficiency and shortage of medical tests. There have been increasing efforts on developing deep learning methods to diagnose COVID-19 based on CT scans. However, these works are difficult to reproduce and adopt since the CT data used in their studies are not publicly available. Besides, these works require a large number of CTs to train accurate diagnosis models, which are difficult to obtain. In this paper, we aim to address these two problems. We build a publicly-available dataset containing hundreds of CT scans positive for COVID-19 and develop sample-efficient deep learning methods that can achieve high diagnosis accuracy of COVID-19 from CT scans even when the number of training CT images are limited. Specifically, we propose a Self-Trans approach, which synergistically integrates contrastive self-supervised learning with transfer learning to learn powerful and unbiased feature representations for reducing the risk of overfitting. Extensive experiments demonstrate the superior performance of our proposed Self-Trans approach compared with several state-of-the-art baselines. Our approach achieves an F1 of 0.85 and an AUC of 0.94 in diagnosing COVID-19 from CT scans, even though the number of training CTs is just a few hundred.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- pathCLIP: Detection of Genes and Gene Relations from Biological Pathway Figures through Image-Text Contrastive Learning 93%
- SimSearch: A Human-in-the-Loop Learning Framework for Fast Detection of Regions of Interest in Microscopy Images 93%
- Hierarchical Coarse-to-Fine cGAN for Subtype-Specific Freezing of Gait Signal Generation 92%
Similar papers in this journal
- Generalized Radiograph Representation Learning via Cross-supervision between Images and Free-text Radiology Reports 96%
- Improving protein function prediction with synthetic feature samples created by generative adversarial networks 91%
- COSIME: Cooperative multi-view integration with Scalable and Interpretable Model Explainer 91%
Similar papers in this journal
- A novel interpretable deep transfer learning combining diverse learnable parameters for improved T2D prediction based on single-cell gene regulatory networks 95%
- Incremental Learning Approach for Semantic Segmentation of Skin Histology Images 94%
- Assisting Scalable Diagnosis Automatically via CT Images in the Combat against COVID-19 94%
Similar papers in this journal
- Inferring global-scale temporal latent topics from news reports to predict public health interventions for COVID-19 91%
- An Interpretable Deep Learning Model for Early Prediction of Sepsis in the Emergency Department 91%
- Single-Cell Multi-Modal GAN (scMMGAN) reveals spatial patterns in single-cell data from triple negative breast cancer 91%
Similar papers in this journal
- Transformer with Convolution and Graph-Node co-embedding: An accurate and interpretable vision backbone for predicting gene expressions from local histopathological image 94%
- Deep learning for predicting COVID-19 malignant progression 94%
- Spatial Transcriptomics Expression Prediction from Histopathology Based on Cross-Modal Mask Reconstruction and Contrastive Learning 94%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.