Early detection of coronary heart disease based on risk factors using interpretable machine learning

Abstract

Coronary heart disease (CHD) is the leading cause of death in the world. The risk of coronary heart disease can be reduced or even prevented by early detection. Early detection of CHD has been widely developed using machine learning, but the machine learning algorithms used sometimes have low interpretability. Low interpretability makes it difficult for users to understand the cause of the decision. Referring to this, this research aims to propose an early detection model using machine learning interpretability, which is implemented using the C5.0 algorithm and interpreted using Shapley additive explanations (SHAP). This research method is divided into 3 stages, namely preprocessing, interpretable machine learning, and performance evaluation. This study used 215 patient data from Dr. Moewardi Surakarta Hospital. Testing the resulting model using the k-folds cross-validation method. The test results show that the risk factors that make a high contribution to the output of the coronary heart disease detection model are systolic blood pressure, diastolic blood pressure, and employment level, with the resulting accuracy performance of 84.64%. The proposed model can be an alternative for early prediction of coronary heart disease which can explain the influence of each selected risk factor on the model output.