Improvement of a Prediction Model for Heart Failure Survival through Explainable Artificial Intelligence

by   Pedro A. Moreno-Sanchez, et al.

Cardiovascular diseases and their associated disorder of heart failure are one of the major death causes globally, being a priority for doctors to detect and predict its onset and medical consequences. Artificial Intelligence (AI) allows doctors to discover clinical indicators and enhance their diagnosis and treatments. Specifically, explainable AI offers tools to improve the clinical prediction models that experience poor interpretability of their results. This work presents an explainability analysis and evaluation of a prediction model for heart failure survival by using a dataset that comprises 299 patients who suffered heart failure. The model employs a data workflow pipeline able to select the best ensemble tree algorithm as well as the best feature selection technique. Moreover, different post-hoc techniques have been used for the explainability analysis of the model. The paper's main contribution is an explainability-driven approach to select the best prediction model for HF survival based on an accuracy-explainability balance. Therefore, the most balanced explainable prediction model implements an Extra Trees classifier over 5 selected features (follow-up time, serum creatinine, ejection fraction, age and diabetes) out of 12, achieving a balanced-accuracy of 85.1 cross-validation and new unseen data respectively. The follow-up time is the most influencing feature followed by serum-creatinine and ejection-fraction. The explainable prediction model for HF survival presented in this paper would improve a further adoption of clinical prediction models by providing doctors with intuitions to better understand the reasoning of, usually, black-box AI clinical solutions, and make more reasonable and data-driven decisions.


Deciphering knee osteoarthritis diagnostic features with explainable artificial intelligence: A systematic review

Existing artificial intelligence (AI) models for diagnosing knee osteoar...

An Explainable Classification Model for Chronic Kidney Disease Patients

Currently, Chronic Kidney Disease (CKD) is experiencing a globally incre...

An explainable XGBoost-based approach towards assessing the risk of cardiovascular disease in patients with Type 2 Diabetes Mellitus

Cardiovascular Disease (CVD) is an important cause of disability and dea...

QXAI: Explainable AI Framework for Quantitative Analysis in Patient Monitoring Systems

Artificial Intelligence techniques can be used to classify a patient's p...

Challenges facing the explainability of age prediction models: case study for two modalities

The prediction of age is a challenging task with various practical appli...

Please sign up or login with your details

Forgot password? Click here to reset