An Optimal Ensemble Learning Framework With PCA‐Based Dimensionality Reduction for Heart Disease Prediction

Read full text
Authors: Narayan Jee, Gesu Thakur, Sumit Kumar
Cited by: 2
References: 15

Abstract

ABSTRACT Cardiovascular disease remains a leading cause of global mortality, necessitating the development of reliable, interpretable, and computationally efficient diagnostic support systems. This study proposes a novel ensemble learning framework for heart disease prediction that integrates Principal Component Analysis (PCA) for dimensionality reduction with a weighted soft voting classifier combining Random Forest, XGBoost, and Logistic Regression. The proposed pipeline incorporates robust preprocessing, including imputation, categorical encoding, standardisation, and class balancing via Synthetic Minority Over‐sampling Technique (SMOTE). Performance was evaluated on the Cleveland Heart Disease dataset using 10‐fold stratified cross‐validation, with comprehensive tuning of hyperparameters. The ensemble achieved an F1‐score of 93.3%, accuracy of 93.3%, and an area under the receiver operating characteristic curve of 94.5%, outperforming several recent state‐of‐the‐art models. Detailed ablation studies and interpretability analyses using SHapley Additive exPlanations (SHAP) and feature importance ranking confirmed the critical role of both PCA and ensemble integration. The methodology demonstrates strong generalisation, robustness to noise and missing data, and alignment with clinical interpretability standards. This framework offers a reproducible and transparent approach for deploying machine learning models in diagnostic cardiology.

Fields of Study & Subjects

Artificial Intelligence in Healthcare Machine Learning in Healthcare Explainable Artificial Intelligence (XAI)

Publication details

Journal / source
Digital twins and applications.
Publisher
Institution of Engineering and Technology (IET)
Volume
3
Issue
1
Publication date
1 January 2026
Language
EN