Patient is NOT all you need : Enhancing Machine Learning-based COVID-19 Screening Models with Epidemiological and Mobility Features
ABSTRACT
Background:
Despite the significant post-coronavirus disease 2019 (COVID-19) pandemic surge in research using symptom data and machine learning for patient screening, data on patient trajectories and epidemiological conditions, although crucial, have remained underutilized.
Objective:
This study aimed to improve the screening performance of machine learning models by incorporating mobility and epidemic information, to patient symptom data.
Methods:
Data, including daily self-reported symptoms, location information, and test results, were collected from 48,798 individuals using a smartphone application. These data were then combined with Our World in Data (OWID) and national government epidemic information to train five machine-learning-based screening models to classify patient infection status. The models were logistic regression, XGBoost, LGBM, TabNet, and Google AutoML.
Results:
The addition of mobility and epidemic data significantly improved the performance of all the five models. The highest AUROC score increased from 0·8712 without mobility and epidemic data to 0·9104 with mobility and epidemic data. This highlights the considerable impact of external information on enhancing the performance of machine learning models.
Conclusions:
This study demonstrated the potential of using mobility and epidemic data, such as location information and epidemic data, in combination with patient symptom data to improve the accuracy of machine learning models for diagnosing COVID-19. Considering additional contextual information can enhance the ability to screen COVID-19.
Citation
Request queued. Please wait while the file is being generated. It may take some time.