Interpretable prediction of heart disease based on random forest and SHAP

Lin Wu

doi:10.1117/12.2682322

20 June 2023 Interpretable prediction of heart disease based on random forest and SHAP

Lin Wu

Proceedings Volume 12715, Eighth International Conference on Electronic Technology and Information Science (ICETIS 2023); 127151F (2023) https://doi.org/10.1117/12.2682322
Event: Eighth International Conference on Electronic Technology and Information Science (ICETIS 2023), 2023, Dalian, China

Abstract

In order to improve the accuracy of heart disease prediction models and address the lack of interpretability in traditional machine learning models, this paper proposes a heart disease prediction method based on random forests and SHAP value. This method first preprocesses the dataset by encoding the data, filling in missing values, and removing outliers. It then uses recursive feature elimination and cross-validation to remove irrelevant features and select relevant features for further model training. The results, compared with other methods using accuracy, precision, recall, and F1 score, show that the proposed method outperforms other models. The interpretable model constructed based on SHAP value reflects the effect of feature values on prediction model results and provides a ranking of feature importance. The experimental results show that the method can effectively improve the accuracy of heart disease prediction, and provide a clear interpretation of the model prediction results. It can be an aid in the treatment and prevention of heart disease.

Citation Download Citation

Lin Wu "Interpretable prediction of heart disease based on random forest and SHAP", Proc. SPIE 12715, Eighth International Conference on Electronic Technology and Information Science (ICETIS 2023), 127151F (20 June 2023); https://doi.org/10.1117/12.2682322

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
5 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Cardiovascular disorders

Random forests

Data modeling

Feature selection

Machine learning

RELATED CONTENT

Heart attack prediction using machine learning
Proceedings of SPIE (June 14 2023)

Detecting heart diseases through machine learning
Proceedings of SPIE (June 13 2023)

Data-centric optimization method to imbalanced datasets
Proceedings of SPIE (April 10 2023)

Exploring the key factors related to the risk of heart...
Proceedings of SPIE (March 24 2023)

Machine learning based prediction of diabetes risk by combining exposome...
Proceedings of SPIE (March 06 2023)

Prediction and analysis of in-vehicle coupon acceptance behavior
Proceedings of SPIE (October 11 2023)

Network traffic anomaly detection based on attention mechanism with LSTM
Proceedings of SPIE (April 03 2024)

Subscribe to Digital Library

Receive Erratum Email Alert

Keywords/Phrases

Search In:

Publication Years