An explainable AI-driven hybrid feature selection approach for coronary artery disease diagnosis
摘要
Coronary artery disease (CAD), where the heart does not get enough oxygen-rich blood due to a buildup of fatty matter, is a leading cause of death worldwide. Since its symptoms may not be recognized until a cardiac attack occurs, its early diagnosis is crucial. In this paper, we introduce the SHAP Optimized Wrapper (SHOW) feature selection algorithm, which works in two steps. First, a SHapley Additive exPlanations (SHAP) method is developed using XGBoost, Random Forest (RF), and Support Vector Machine (SVM) classifiers, to rank the features based on their diagnostic significance. Second, an optimized sequential forward selection wrapper technique is employed, whereby the ranked features are evaluated to select the optimal subset. To validate the algorithm, it is used in seven classifiers to classify three public domain CAD data sets. The classifiers are XGBoost, RF, SVM, Decision Tree (DT), Logistic Regression (LR), K-Nearest Neighbors (KNN), and Multilayer Perceptron (MLP). The data sets are the Z-Alizadeh Sani, Cleveland, and Statlog. Leveraging stratified 10-fold cross-validation and delicate hyperparameter tuning, the results reveal that the SHOW algorithm significantly outperforms 14 state-of-the-art competitive algorithms in terms of accuracy and the number of selected features, while also demonstrating favorable performance in clinically relevant metrics such as sensitivity, specificity, AUC, and F1-score. For example, using the XGBoost classifier, the algorithm selects 14 features (out of 55) from the Z-Alizadeh Sani data set, achieving 93.79% accuracy, 93.98% sensitivity, 89.81% specificity, 0.97 AUC, and 93.98% F1-score; 5 features (out of 13) from the Cleveland data set, achieving 86.52% accuracy, 88.55% sensitivity, 85% specificity, 0.89 AUC, and 84.84% F1-score; and 5 features (out of 13) from the Statlog data set, achieving 87.78% accuracy, 80% sensitivity, 92.67% specificity, 0.90 AUC, and 85.18% F1-score. These figures are not matched by any of the 14 competitive algorithms.