COVID-19 Image Data Collection: Prospective Predictions Are the Future
Joseph Paul Cohen, Paul Morrison, Lan Dao, Karsten Roth, Tim Q Duong, Marzyeh Ghassemi
TL;DR
This work provides the first public COVID-19 chest X-ray dataset with rich prospective metadata collected from 26 countries, enabling prognostic and management-oriented ML tasks. It emphasizes domain shift through a LOCO evaluation and offers baseline results using a DenseNet-derived feature space across diagnosis, severity, survival, and trajectory tasks. The authors highlight the potential and limitations of imaging alone for COVID-19 prognosis, and propose concrete future directions (saliency, OoD detection, and domain adaptation) to accelerate multicenter validation. Overall, the dataset and task definitions aim to catalyze rapid, cross-country ML development for COVID-19 imaging-based prognosis and care planning.
Abstract
Across the world's coronavirus disease 2019 (COVID-19) hot spots, the need to streamline patient diagnosis and management has become more pressing than ever. As one of the main imaging tools, chest X-rays (CXRs) are common, fast, non-invasive, relatively cheap, and potentially bedside to monitor the progression of the disease. This paper describes the first public COVID-19 image data collection as well as a preliminary exploration of possible use cases for the data. This dataset currently contains hundreds of frontal view X-rays and is the largest public resource for COVID-19 image and prognostic data, making it a necessary resource to develop and evaluate tools to aid in the treatment of COVID-19. It was manually aggregated from publication figures as well as various web based repositories into a machine learning (ML) friendly format with accompanying dataloader code. We collected frontal and lateral view imagery and metadata such as the time since first symptoms, intensive care unit (ICU) status, survival status, intubation status, or hospital location. We present multiple possible use cases for the data such as predicting the need for the ICU, predicting patient survival, and understanding a patient's trajectory during treatment. Data can be accessed here: https://github.com/ieee8023/covid-chestxray-dataset
