Clinical outcome prediction in COVID-19 using self-supervised vision transformer representations

Aishik Konwer; Prateek Prasanna

doi:10.1117/12.2612957

4 April 2022 Clinical outcome prediction in COVID-19 using self-supervised vision transformer representations

Aishik Konwer, Prateek Prasanna

Proceedings Volume 12033, Medical Imaging 2022: Computer-Aided Diagnosis; 1203307 (2022) https://doi.org/10.1117/12.2612957
Event: SPIE Medical Imaging, 2022, San Diego, California, United States

Abstract

Automated analysis of chest imaging in coronavirus disease (COVID-19) has mostly been performed on smaller datasets leading to overfitting and poor generalizability. Training of deep neural networks on large datasets requires data labels. This is not always available and can be expensive to obtain. Self-supervision is being increasingly used in various medical imaging tasks to leverage large amount of unlabeled data during pretraining. Our proposed approach pretrains a vision transformer to perform two self-supervision tasks - image reconstruction and contrastive learning on a Chest Xray (CXR) dataset. In the process, we generate more robust image embeddings. The reconstruction module models visual semantics within the lung fields by reconstructing the input image through a mechanism which mimics denoising and autoencoding. On the other hand, the constrastive learning module learns the concept of similarity between two texture representations. After pretraining, the vision transformer is used as a feature extractor towards a clinical outcome prediction task on our target dataset. The pretraining multi-kaggle dataset comprises 27499 CXR scans while our target dataset contains 530 images. Specifically, our framework predicts ventilation and mortality outcomes for COVID-19 positive patients using baseline CXR. We compare our method against a baseline approach using pretrained ResNet50 features. Experimental results demonstrate that our proposed approach outperforms the supervised method.

Conference Presentation

Citation Download Citation

Aishik Konwer and Prateek Prasanna "Clinical outcome prediction in COVID-19 using self-supervised vision transformer representations", Proc. SPIE 12033, Medical Imaging 2022: Computer-Aided Diagnosis, 1203307 (4 April 2022); https://doi.org/10.1117/12.2612957

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available