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.

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

Heart Disease Diagnosis Using Machine Learning

  • Michele Roccotelli,
  • Wasim A. Ali,
  • Maria Pia Fanti

摘要

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.