Forecasting Heart Disease Risk with a Stacking-Based Ensemble Machine Learning Method

Abstract

As one of the main causes of sickness and mortality, heart disease, also known as cardiovascular disease, must be detected early in order to be prevented and treated. The rapid development of computer technology presents an opportunity for the cross-combination of medicine and informatics. A novel stacking model called SDKABL is presented in this work. It uses three classifiers, namely K-Nearest Neighbor (KNN), Decision Tree (DT), and Support Vector Machine (SVM) at the base layer and the Bidirectional Long Short-Term Memory based on Attention Mechanisms (ABiLSTM) model at the meta layer for the ultimate prediction. For lowering the temporal complexity and enhancing the model’s accuracy, the dimensionality reduction approach is seen to be crucial. Principal Component Analysis (PCA) was utilized in SDKABL to minimize dimensionality and facilitate feature fusion. Using several performance measures, including precision, F1-score, accuracy, recall, and Receiver Operating Characteristic (ROC) score, the performance of SDKABL was compared to that of other independent classifiers. The experimental findings demonstrate that our proposed model combining individual classifiers with the stacking method helps improve the prediction model’s accuracy.