همايش ملی پيشرفت های معماری سازمانی

صفحه اصلی / هشتمین همايش ملی پيشرفت های معماری سازمانی

A combination of EDA, Machine Learning, and Artificial Neural Networks for Accurate Prediction of Heart Disease

نویسندگان :

Ali Rashedi Gazari¹ Shayan Rokhva² Toktam Khatibi³ Elham Akhondzadeh Noughabi⁴ Babak Teymourpour⁵

1- دانشگاه تربیت مدرس 2- دانشگاه تربیت مدرس 3- دانشگاه تربیت مدرس 4- دانشگاه تربیت مدرس 5- دانشگاه تربیت مدرس

کلمات کلیدی :

Heart Disease،Exploratory Data Analysis،Machine Learning،Artificial Intelligence،Healthcare

چکیده :

Cardiovascular diseases cause millions of deaths each year, with cases continuing to rise, making early prediction increasingly important. Although data science and artificial intelligence have been utilized to address this issue, further studies that enhance predictability and generalization are crucial as they significantly reduce mortality rates and healthcare costs. This study employs exploratory data analysis, a variety of conventional machine learning algorithms, and an artificial neural network to predict heart disease accurately and fill research gaps. A dataset from Kaggle, containing 1025 training samples and 303 test samples, with 14 attributes, including 13 predictive variables and a binary target indicating heart disease presence, was used. Normalization, feature importance analysis, K-fold cross-validation, and grid search were meticulously applied to improve model performance, generalization, and robustness. These methodologies led to impressive results, with most models achieving 100% accuracy, precision, recall, and F1-score on the test data, without signs of overfitting, data leakage, or bias. Principal component analysis was also conducted to evaluate the richness of the features and their potential for dimension reduction. Lastly, in-depth discussions were made to clarify the study’s outcomes, compare results with the most related studies, and comprehensively examine real-world applicability.