CardiaTics: An explainable AI integrated heart disease diagnosis model with feature engineering and stacked ensemble approach
摘要
Heart disease is a leading global cause of morbidity and mortality. Accurate and prompt diagnoses are crucial for its effective prevention and management. Integrating multiple machine learning algorithms, this research introduces a stacked ensemble machine learning model, called CardiaTics (stands for Cardiac DiagnosTics), toward improving heart disease detection. We detect outliers and remove them as a first-step to ensure data quality and maintain integrity. Ten distinct machine learning algorithms are then individually applied, culminating in the creation of a stacked ensemble model. We use feature engineering to refine the model further applying three well-known techniques –Pearson correlation, Chi-Square Test (Chi-2), and Recursive Feature Elimination. The implementation of these techniques on the benchmark dataset results in an optimized feature set. Experimental results show that CardiaTics delivers 89.3% accuracy on raw data, and significantly improves its accuracy after feature selection to 93.3%, outperforming the individual classifiers. However, can human professionals rely on algorithms for prediction when the underlying process is not fully understood? To address concerns regarding interpretability, trust, and transparency in black-box predictions, we propose utilizing SHapley Additive exPlanations (SHAP) and Explain Like I’m 5 (ELI5) in the second phase to elucidate feature importance in our model. The SHAP summary plots of CardiaTics reveal that the positive and negative contributors to heart disease are comparable, thereby enhancing the model’s interpretability and reliability and helping refine the decision-making process.