Skip to main content
medRxiv
  • Home
  • About
  • Submit
  • ALERTS / RSS
Advanced Search

Explainable machine learning models to understand determinants of COVID-19 mortality in the United States

Piyush Mathur, Tavpritesh Sethi, Anya Mathur, Kamal Maheshwari, Jacek B Cywinski, Ashish K Khanna, Simran Dua, Frank Papay
doi: https://doi.org/10.1101/2020.05.23.20110189
Piyush Mathur
1Department of General Anesthesiology, Anesthesiology Institute, Cleveland Clinic, Cleveland
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
  • For correspondence: pmathurmd@gmail.com
Tavpritesh Sethi
2Indraprastha Institute of Information Technology, Delhi, India
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
Anya Mathur
3Brecksville Broadview Heights Middle School, Ohio
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
Kamal Maheshwari
1Department of General Anesthesiology, Anesthesiology Institute, Cleveland Clinic, Cleveland
5Outcomes Research Consortium, Cleveland, OH
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
Jacek B Cywinski
1Department of General Anesthesiology, Anesthesiology Institute, Cleveland Clinic, Cleveland
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
Ashish K Khanna
4Department of Anesthesiology, Section on Critical Care Medicine, Wake Forest School of Medicine
5Outcomes Research Consortium, Cleveland, OH
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
Simran Dua
6Lambert High School, Suwanee, Georgia
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
Frank Papay
7Dermatology and Plastic Surgery Institute, Cleveland Clinic, Cleveland
  • Find this author on Google Scholar
  • Find this author on PubMed
  • Search for this author on this site
  • Abstract
  • Full Text
  • Info/History
  • Metrics
  • Supplementary material
  • Data/Code
  • Preview PDF
Loading

Abstract

Background COVID-19 is now one of the leading causes of mortality amongst adults in the United States for the year 2020. Multiple epidemiological models have been built, often based on limited data, to understand the spread and impact of the pandemic. However, many geographic and local factors may have played an important role in higher morbidity and mortality in certain populations.

Objective The goal of this study was to develop machine learning models to understand the relative association of socioeconomic, demographic, travel, and health care characteristics of different states across the United States and COVID-19 mortality.

Methods Using multiple public data sets, 24 variables linked to COVID-19 disease were chosen to build the models. Two independent machine learning models using CatBoost regression and random forest were developed. SHAP feature importance and a Boruta algorithm were used to elucidate the relative importance of features on COVID-19 mortality in the United States.

Results Feature importances from both the categorical models, i.e., CatBoost and random forest consistently showed that a high population density, number of nursing homes, number of nursing home beds and foreign travel were strongest predictors of COVID-19 mortality. Percentage of African American amongst the population was also found to be of high importance in prediction of COVID-19 mortality whereas racial majority (primarily, Caucasian) was not. Both models fitted the data well with a training R2 of 0.99 and 0.88 respectively. The effect of median age,median income, climate and disease mitigation measures on COVID-19 related mortality remained unclear.

Conclusions COVID-19 policy making will need to take population density, pre-existing medical care and state travel policies into account. Our models identified and quantified the relative importance of each of these for mortality predictions using machine learning.

Competing Interest Statement

The authors have declared no competing interest.

Funding Statement

No external funding was received.

Author Declarations

I confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.

Yes

The details of the IRB/oversight body that provided approval or exemption for the research described are given below:

IRB approval not required.

All necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived.

Yes

I understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).

Yes

I have followed all appropriate research reporting guidelines and uploaded the relevant EQUATOR Network research reporting checklist(s) and other pertinent material as supplementary files, if applicable.

Yes

Data Availability

Data sources listed in supplemental content

Copyright 
The copyright holder for this preprint is the author/funder, who has granted medRxiv a license to display the preprint in perpetuity. It is made available under a CC-BY-NC-ND 4.0 International license.
Back to top
PreviousNext
Posted June 21, 2020.
Download PDF

Supplementary Material

Data/Code
Email

Thank you for your interest in spreading the word about medRxiv.

NOTE: Your email address is requested solely to identify you as the sender of this article.

Enter multiple addresses on separate lines or separate them with commas.
Explainable machine learning models to understand determinants of COVID-19 mortality in the United States
(Your Name) has forwarded a page to you from medRxiv
(Your Name) thought you would like to see this page from the medRxiv website.
CAPTCHA
This question is for testing whether or not you are a human visitor and to prevent automated spam submissions.
Share
Explainable machine learning models to understand determinants of COVID-19 mortality in the United States
Piyush Mathur, Tavpritesh Sethi, Anya Mathur, Kamal Maheshwari, Jacek B Cywinski, Ashish K Khanna, Simran Dua, Frank Papay
medRxiv 2020.05.23.20110189; doi: https://doi.org/10.1101/2020.05.23.20110189
Digg logo Reddit logo Twitter logo CiteULike logo Facebook logo Google logo Mendeley logo
Citation Tools
Explainable machine learning models to understand determinants of COVID-19 mortality in the United States
Piyush Mathur, Tavpritesh Sethi, Anya Mathur, Kamal Maheshwari, Jacek B Cywinski, Ashish K Khanna, Simran Dua, Frank Papay
medRxiv 2020.05.23.20110189; doi: https://doi.org/10.1101/2020.05.23.20110189

Citation Manager Formats

  • BibTeX
  • Bookends
  • EasyBib
  • EndNote (tagged)
  • EndNote 8 (xml)
  • Medlars
  • Mendeley
  • Papers
  • RefWorks Tagged
  • Ref Manager
  • RIS
  • Zotero
  • Tweet Widget
  • Facebook Like
  • Google Plus One

Subject Area

  • Health Informatics
Subject Areas
All Articles
  • Addiction Medicine (62)
  • Allergy and Immunology (142)
  • Anesthesia (46)
  • Cardiovascular Medicine (415)
  • Dentistry and Oral Medicine (70)
  • Dermatology (48)
  • Emergency Medicine (144)
  • Endocrinology (including Diabetes Mellitus and Metabolic Disease) (171)
  • Epidemiology (4861)
  • Forensic Medicine (3)
  • Gastroenterology (183)
  • Genetic and Genomic Medicine (676)
  • Geriatric Medicine (70)
  • Health Economics (192)
  • Health Informatics (630)
  • Health Policy (321)
  • Health Systems and Quality Improvement (206)
  • Hematology (85)
  • HIV/AIDS (156)
  • Infectious Diseases (except HIV/AIDS) (5343)
  • Intensive Care and Critical Care Medicine (331)
  • Medical Education (93)
  • Medical Ethics (24)
  • Nephrology (75)
  • Neurology (686)
  • Nursing (42)
  • Nutrition (115)
  • Obstetrics and Gynecology (126)
  • Occupational and Environmental Health (208)
  • Oncology (440)
  • Ophthalmology (140)
  • Orthopedics (36)
  • Otolaryngology (90)
  • Pain Medicine (35)
  • Palliative Medicine (16)
  • Pathology (129)
  • Pediatrics (194)
  • Pharmacology and Therapeutics (131)
  • Primary Care Research (84)
  • Psychiatry and Clinical Psychology (781)
  • Public and Global Health (1817)
  • Radiology and Imaging (325)
  • Rehabilitation Medicine and Physical Therapy (138)
  • Respiratory Medicine (255)
  • Rheumatology (86)
  • Sexual and Reproductive Health (69)
  • Sports Medicine (62)
  • Surgery (100)
  • Toxicology (23)
  • Transplantation (29)
  • Urology (37)