Prediction of COVID-19 Status Using Baseline Demographics and Lab Data

Abstract

The global surge in COVID-19 cases underscores the need for fast, scalable, and reliable testing. Current COVID-19 diagnostic tests are limited by turnaround time, limited availability, or occasional false ndings. Here, we developed a machine learning-based framework for predicting COVID-19 positive test status relying only on readily-available baseline data, including patient demographics, comorbidities, and common lab values. Leveraging a cohort of 31,739 adults within an academic health system, we trained and tested multiple types of machine learning models, achieving an area under the curve of 0.75. Feature importance analyses highlighted serum calcium levels, aspartate aminotransferase levels, and oxygen saturation as key predictors. Additionally, we developed a single decision tree model that provided an operable method for stratifying sub-populations. Overall, this study provides a proof-of-concept that COVID-19 status prediction models can be developed using only baseline data. The resulting prediction can complement existing tests to enhance screening and pandemic containment work ows.