<p>One of the major causes of mortality around the world, poor diagnosis and the early stages of the disease being asymptomatic are the primary reasons for the occurrence of coronary artery disease (CAD). Therefore, early detection of CAD is essential to decrease the risk of death and improve clinical intervention. Machine learning (ML) methods have shown good performance for CAD prediction; however, existing methods face challenges such as limited cross-dataset generalizability, suboptimal feature selection, poor performance under class imbalance, and limited interpretability. To overcome these drawbacks, this study proposes a leakage-safe ensemble learning framework, namely CardioEnsemNet, for accurate and interpretable CAD prediction from heterogeneous clinical datasets. The framework is based on a combination of dataset-aware preprocessing, advanced feature engineering techniques, including polynomial and interaction features, Boruta feature selection, and SMOTE class balancing. To enhance predictive performance, the optimised stacking ensemble architecture is used, combining classifiers such as Random Forest, Support Vector Machine, K-Nearest Neighbours, XGBoost, and a meta-learner logistic regression. Furthermore, SHAP and LIME are added to provide interpretability of model predictions at both global and local scales. Four benchmark datasets, UCI Cleveland Heart Disease, Statlog Heart, Z-Alizadeh Sani and Framingham Heart Study were used to evaluate the proposed framework. Experimental results show that predictive performance is consistently high, with accuracy values exceeding 97%, F1-scores exceeding 95%, and ROC-AUC values close to 0.99 across all datasets. The robustness and contribution of the proposed preprocessing and ensemble learning strategies are then validated and confirmed by using cross-dataset validation and ablation studies. The results indicate that the CardioEnsemNet is a reliable, interpretable model for predicting CAD across diverse clinical data conditions.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CardioEnsemNet: An Optimized Ensemble Learning Framework for Accurate Coronary Artery Disease Prediction Across Multiple Datasets

  • T. Aswani,
  • Jose Moses Gummadi,
  • G. Sharada

摘要

One of the major causes of mortality around the world, poor diagnosis and the early stages of the disease being asymptomatic are the primary reasons for the occurrence of coronary artery disease (CAD). Therefore, early detection of CAD is essential to decrease the risk of death and improve clinical intervention. Machine learning (ML) methods have shown good performance for CAD prediction; however, existing methods face challenges such as limited cross-dataset generalizability, suboptimal feature selection, poor performance under class imbalance, and limited interpretability. To overcome these drawbacks, this study proposes a leakage-safe ensemble learning framework, namely CardioEnsemNet, for accurate and interpretable CAD prediction from heterogeneous clinical datasets. The framework is based on a combination of dataset-aware preprocessing, advanced feature engineering techniques, including polynomial and interaction features, Boruta feature selection, and SMOTE class balancing. To enhance predictive performance, the optimised stacking ensemble architecture is used, combining classifiers such as Random Forest, Support Vector Machine, K-Nearest Neighbours, XGBoost, and a meta-learner logistic regression. Furthermore, SHAP and LIME are added to provide interpretability of model predictions at both global and local scales. Four benchmark datasets, UCI Cleveland Heart Disease, Statlog Heart, Z-Alizadeh Sani and Framingham Heart Study were used to evaluate the proposed framework. Experimental results show that predictive performance is consistently high, with accuracy values exceeding 97%, F1-scores exceeding 95%, and ROC-AUC values close to 0.99 across all datasets. The robustness and contribution of the proposed preprocessing and ensemble learning strategies are then validated and confirmed by using cross-dataset validation and ablation studies. The results indicate that the CardioEnsemNet is a reliable, interpretable model for predicting CAD across diverse clinical data conditions.