Machine Learning-Based Stacked Ensemble Models for Predicting Heart Disease
Heart disease is still the leading cause of mortality worldwide. Hence, advanced prediction models are necessary for early detection and care. By combining three base classifiersK-Nearest Neighbors (KNN), Support Vector Machines (SVMs), and Random Forests (RFs)with logistic regression as a meta-model, this work suggests a stacked ensemble model to increase the accuracy of heart disease prediction. Specifically, this paper utilizes KNN to capture local data patterns. Secondly, SVM deals with high-dimensional feature spaces, and RF is utilized to mitigate overfitting through ensemble voting. Logistic regression is used to weight its predictions optimally. The model was trained and tested on the UCI Heart Disease dataset (303 records, 14 clinical features). This paper's preprocessing stage deals with missing values, normalizes numerical data, and codes categorical variables. To optimize the performance, the study also performs hyperparameter tuning and cross-validation. According to the experimental data, the stacked ensemble outperforms a single model (KNN: 85.2%; SVM: 88.6%; RF: 89.3%) with an accuracy of 93.7%. Key risk factors identified included maximum heart rate, cholesterol level, and type of chest pain. Consistency with clinical knowledge validated the reliability of the model. This study highlights how effective ensemble learning is at increasing diagnostic precision, and it shows promise for use in clinical decision support systems. Future work will explore further integrating genetic and lifestyle data and deployment in real-time healthcare environments to improve predictive robustness and patient outcomes.
Authors
- Lu An (ORCID: https://orcid.org/0000-0003-4050-3625)
Institutions
- University of Washington (US)
Publication Details
- Journal
- Applied and Computational Engineering
- Published
- 2025-05-30
- DOI
- https://doi.org/10.54254/2755-2721/2025.gl23424
- Citations
- 1
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 2.94