PT  - JOURNAL ARTICLE
AU  - Qiu, Wei
AU  - Chen, Hugh
AU  - Dincer, Ayse Berceste
AU  - Lundberg, Scott
AU  - Kaeberlein, Matt
AU  - Lee, Su-In
TI  - Interpretable machine learning prediction of all-cause mortality
AID  - 10.1101/2021.01.20.21250135
DP  - 2022 Jan 01
TA  - medRxiv
PG  - 2021.01.20.21250135
4099  - http://medrxiv.org/content/early/2022/01/12/2021.01.20.21250135.short
4100  - http://medrxiv.org/content/early/2022/01/12/2021.01.20.21250135.full
AB  - Prior studies on all-cause mortality traditionally use linear models; however, growing field of explainable artificial intelligence (XAI) can improve prediction accuracy over traditional linear models using complex machine learning (ML) models while still revealing novel insights. We propose the IMPACT (Interpretable Machine learning Prediction of All-Cause morTality) framework that implements and explains complex, non-linear ML models by combining a tree ensemble mortality prediction model and a principled XAI technique. We apply IMPACT to the NHANES (1999-2014) dataset, which enables us to understand different subpopulations according to shorter or longer term mortality and younger and older individuals. Our IMPACT models have higher predictive accuracy than popular pre-existing mortality risk scores and biological ages. Using individualized feature importance scores, we discover novel risk predictors (e.g., arm circumference) and interactions between risk predictors (e.g., serum chloride with age and/or gender). Furthermore, IMPACT provides a novel perspective of reference intervals and may suggest that the widely accepted reference intervals for serum albumin, mean cell volume and platelet count may in fact be sub-optimal for health. Finally, in order to ensure that our models are useful to as broad of a community as possible, we develop and publish a variety of explainable risk scores usable by individuals with and without medical expertise. The predictive accuracy of IMPACT combined with the capability of discovering mortality risk predictors and complex relationships demonstrates the value and utility of XAI in epidemiologic study design.Competing Interest StatementThe authors have declared no competing interest.Funding StatementThis work is funded by National Science Foundation [DBI-1552309] and National Institutes of Health [R35 GM 128638].Author DeclarationsI confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.YesThe details of the IRB/oversight body that provided approval or exemption for the research described are given below:This work does not need approval of the IRB/oversight body.I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.YesI 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).YesI 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.YesData is available from The National Health and Nutrition Examination Survey (NHANES - https://www.cdc.gov/nchs/nhanes/). https://github.com/qiuweipku/IMPACT