On the explainability of hospitalization prediction on a large COVID-19 patient dataset

10/28/2021

∙

We develop various AI models to predict hospitalization on a large (over 110k) cohort of COVID-19 positive-tested US patients, sourced from March 2020 to February 2021. Models range from Random Forest to Neural Network (NN) and Time Convolutional NN, where combination of the data modalities (tabular and time dependent) are performed at different stages (early vs. model fusion). Despite high data unbalance, the models reach average precision 0.96-0.98 (0.75-0.85), recall 0.96-0.98 (0.74-0.85), and F_1-score 0.97-0.98 (0.79-0.83) on the non-hospitalized (or hospitalized) class. Performances do not significantly drop even when selected lists of features are removed to study model adaptability to different scenarios. However, a systematic study of the SHAP feature importance values for the developed models in the different scenarios shows a large variability across models and use cases. This calls for even more complete studies on several explainability methods before their adoption in high-stakes scenarios.

READ FULL TEXT

On the explainability of hospitalization prediction on a large COVID-19 patient dataset

COVID-Net Biochem: An Explainability-driven Framework to Building Machine Learning Models for Predicting Survival and Kidney Injury of COVID-19 Patients from Clinical and Bioch

Hybrid quantum convolutional neural networks model for COVID-19 prediction using chest X-Ray images

Improving Explainability of Image Classification in Scenarios with Class Overlap: Application to COVID-19 and Pneumonia

Using natural language processing and structured medical data to phenotype patients hospitalized due to COVID-19

Deep survival analysis with longitudinal X-rays for COVID-19

CNN AE: Convolution Neural Network combined with Autoencoder approach to detect survival chance of COVID 19 patients

Explainability Guided Multi-Site COVID-19 CT Classification

On the explainability of hospitalization prediction on a large COVID-19 patient dataset

Related Research

COVID-Net Biochem: An Explainability-driven Framework to Building Machine Learning Models for Predicting Survival and Kidney Injury of COVID-19 Patients from Clinical and Bioch

Hybrid quantum convolutional neural networks model for COVID-19 prediction using chest X-Ray images

Improving Explainability of Image Classification in Scenarios with Class Overlap: Application to COVID-19 and Pneumonia

Using natural language processing and structured medical data to phenotype patients hospitalized due to COVID-19

Deep survival analysis with longitudinal X-rays for COVID-19

CNN AE: Convolution Neural Network combined with Autoencoder approach to detect survival chance of COVID 19 patients

Explainability Guided Multi-Site COVID-19 CT Classification