A model-agnostic approach for understanding heart failure risk factors

BMC Res Notes. 2021 May 17;14(1):184. doi: 10.1186/s13104-021-05596-7.

Abstract

Objective: Understanding the risk factors for developing heart failure among patients with type 2 diabetes can contribute to preventing deterioration of quality of life for those persons. Electronic health records (EHR) provide an opportunity to use sophisticated machine learning models to understand and compare the effect of different risk factors for developing HF. As the complexity of the model increases, however, the transparency of the model often decreases. To interpret the results, we aimed to develop a model-agnostic approach to shed light on complex models and interpret the effect of features on developing heart failure. Using the HealthFacts EHR database of the Cerner EHR, we extracted the records of 723 patients with at least 6 yeas of follow up of type 2 diabetes, of whom 134 developed heart failure. Using age and comorbidities as features and heart failure as the outcome, we trained logistic regression, random forest, XGBoost, neural network, and then applied our proposed approach to rank the effect of each factor on developing heart failure.

Results: Compared to the "importance score" built-in function of XGBoost, our proposed approach was more accurate in ranking the effect of the different risk factors on developing heart failure.

Keywords: Explainable AI; Heart failure; Model-agnostic approach.

MeSH terms

  • Diabetes Mellitus, Type 2* / complications
  • Diabetes Mellitus, Type 2* / epidemiology
  • Electronic Health Records
  • Heart Failure* / epidemiology
  • Humans
  • Machine Learning
  • Quality of Life
  • Risk Factors