FIT calculator: a multi-risk prediction framework for medical outcomes using cardiorespiratory fitness data

被引:1
作者
Elshawi, Radwa [1 ]
Sakr, Sherif [1 ]
Al-Mallah, Mouaz H. [2 ]
Keteyian, Steven J. [3 ]
Brawner, Clinton A. [3 ]
Ehrman, Jonathan K. [3 ]
机构
[1] Univ Tartu, Inst Comp Sci, Tartu, Estonia
[2] Houston Methodist DeBakey Heart & Vasc Ctr, Houston, TX USA
[3] Henry Ford Hosp, Div Cardiovasc Med, 6525 Second Ave, Detroit, MI 48202 USA
关键词
Prediction model; Classification techniques; Interpretability; Automatic algorithm selection; Hyperparameter optimization; ELECTRONIC HEALTH RECORDS; EXPLAINABLE AI; MODELS;
D O I
10.1038/s41598-024-59401-z
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Accurately predicting patients' risk for specific medical outcomes is paramount for effective healthcare management and personalized medicine. While a substantial body of literature addresses the prediction of diverse medical conditions, existing models predominantly focus on singular outcomes, limiting their scope to one disease at a time. However, clinical reality often entails patients concurrently facing multiple health risks across various medical domains. In response to this gap, our study proposes a novel multi-risk framework adept at simultaneous risk prediction for multiple clinical outcomes, including diabetes, mortality, and hypertension. Leveraging a concise set of features extracted from patients' cardiorespiratory fitness data, our framework minimizes computational complexity while maximizing predictive accuracy. Moreover, we integrate a state-of-the-art instance-based interpretability technique into our framework, providing users with comprehensive explanations for each prediction. These explanations afford medical practitioners invaluable insights into the primary health factors influencing individual predictions, fostering greater trust and utility in the underlying prediction models. Our approach thus stands to significantly enhance healthcare decision-making processes, facilitating more targeted interventions and improving patient outcomes in clinical practice. Our prediction framework utilizes an automated machine learning framework, Auto-Weka, to optimize machine learning models and hyper-parameter configurations for the simultaneous prediction of three medical outcomes: diabetes, mortality, and hypertension. Additionally, we employ a local interpretability technique to elucidate predictions generated by our framework. These explanations manifest visually, highlighting key attributes contributing to each instance's prediction for enhanced interpretability. Using automated machine learning techniques, the models simultaneously predict hypertension, mortality, and diabetes risks, utilizing only nine patient features. They achieved an average AUC of 0.90 +/- 0.001 on the hypertension dataset, 0.90 +/- 0.002 on the mortality dataset, and 0.89 +/- 0.001 on the diabetes dataset through tenfold cross-validation. Additionally, the models demonstrated strong performance with an average AUC of 0.89 +/- 0.001 on the hypertension dataset, 0.90 +/- 0.001 on the mortality dataset, and 0.89 +/- 0.001 on the diabetes dataset using bootstrap evaluation with 1000 resamples.
引用
收藏
页数:10
相关论文
共 27 条
  • [1] Rationale and Design of the Henry Ford ExercIse Testing Project (The FIT Project)
    Al-Mallah, Mouaz H.
    Keteyian, Steven J.
    Brawner, Clinton A.
    Whelton, Seamus
    Blaha, Michael J.
    [J]. CLINICAL CARDIOLOGY, 2014, 37 (08) : 456 - 461
  • [2] Predicting diabetes mellitus using SMOTE and ensemble machine learning approach: The Henry Ford ExercIse Testing (FIT) project
    Alghamdi, Manal
    Al-Mallah, Mouaz
    Keteylan, Steven
    Brawner, Clinton
    Ehrman, Jonathan
    Sakr, Sherif
    [J]. PLOS ONE, 2017, 12 (07):
  • [3] Random forests
    Breiman, L
    [J]. MACHINE LEARNING, 2001, 45 (01) : 5 - 32
  • [4] Predicting Chronic Disease Hospitalizations from Electronic Health Records: An Interpretable Classification Approach
    Brisimi, Theodora S.
    Xu, Tingting
    Wang, Taiyao
    Dai, Wuyang
    Adams, William G.
    Paschalidis, Ioannis Ch
    [J]. PROCEEDINGS OF THE IEEE, 2018, 106 (04) : 690 - 707
  • [5] Use of a Machine Learning Algorithm Improves Prediction of Progression to Diabetes
    Cahn, Avivit
    Shoshan, Avi
    Sagiv, Tal
    Yesharim, Rachel
    Raz, Itamar
    Goshen, Ran
    [J]. DIABETES, 2018, 67
  • [6] Chang W., 2023, shiny: Web Application Framework for R
  • [7] Interpretable Local Concept-based Explanation with Human Feedback to Predict All-cause Mortality
    Elshawi, Radwa
    Al-Mallah, Mouaz
    [J]. JOURNAL OF ARTIFICIAL INTELLIGENCE RESEARCH, 2022, 75 : 833 - 855
  • [8] Elshawi R, 2020, LECT NOTES BUS INF P, V385, P627
  • [9] On the interpretability of machine learning-based model for predicting hypertension
    Elshawi, Radwa
    Al-Mallah, Mouaz H.
    Sakr, Sherif
    [J]. BMC MEDICAL INFORMATICS AND DECISION MAKING, 2019, 19 (1)
  • [10] Goodman B, 2016, Arxiv, DOI arXiv:1606.08813