Heart Disease Diagnosis Using Machine Learning

The growing impact of heart disease on global health requires to improve diagnostic techniques for more timely and accurate diagnosis. Machine Learning (ML) has demonstrated significant potential in supporting the identification and classification of heart diseases, thanks to its ability to analyze large volumes of data and learn complex patterns. The aim of this work is to explore the application of ML algorithms for heart disease diagnosis, using two datasets ‘Heart Disease Cleveland’ and ‘Heart Failure Prediction Dataset’ available on the web. Each dataset is enriched with 1000 synthetic instances, generated by a designed Generative Adversarial Network model. Different ML-based classification models including Random Forest, Logistic Regression, Stochastic Gradient Descent and XGBoost are compared based on standard performance metrics. In addition, a stacking model as an ensemble method based on the combination of the above four models has been developed and tested. The obtained results show the effectiveness of ML models in the diagnosis of cardiac diseases, with the stacking model standing out for its superior performance according to the majority of metrics.