An explainable AI based new deep learning solution for efficient heart disease prediction at early stages
摘要
Accurate predictive models are crucial for early detection and intervention of Heart Disease (HD), which continues to be a major cause of death worldwide. However, the challenges of high dimensionality and data imbalance affect the precision and generalizability of predictions. We propose a novel deep model, the Fully Connected Wave Network (FCW-Net), for early HD prediction. To mitigate class imbalance, Proximity Weighted Synthetic (ProWSyn) oversampling technique is employed, while Principal Component Analysis (PCA) is used to reduce dimensionality, enhancing model efficiency and prediction accuracy. With an AUC-ROC of 0.9622, an accuracy of 0.9237, a precision of 0.8856, an F1-score of 0.9268, and a recall of 0.9721, our results demonstrate that the FCW-Net deep model achieves superior performance than baseline models when using PCA and ProWSyn. Model transparency is further improved by eXplainable Artificial Intelligence technique, SHapley Additive exPlanations analysis, which provides insights into feature contributions to HD risk.