Development of AI-Based Prediction of Heart Attack Risk as an Element of Preventive Medicine

Izabela Rojek,Piotr Kotlarz,Mirosław Kozielski,Mieczysław Jagodziński,Zbyszko Królikowski
DOI: https://doi.org/10.3390/electronics13020272
IF: 2.9
2024-01-07
Electronics
Abstract:The future paradigm of early cardiac diagnostics is shifting the focus towards heart attack preventive medicine based on non-invasive medical imaging with the support of artificial intelligence. It is necessary to preventively detect its increased risk early and respond with preventive drugs before moving on to more effective, but also more invasive, forms of therapy. The main motivation of our study was to improve existing and develop new AI-based solutions for cardiac preventive medicine, with particular emphasis on the prevention of heart attacks. This is due to the fact that the epidemic of lifestyle diseases (including cardiologic ones) has been stopped but not reversed; hence, automatically supervised prevention using AI seems to be a key opportunity to introduce progress in the above-mentioned areas. This can have major effects not only scientific and clinical in nature, but also economic and social. The aim of this article is to develop and test an AI-based tool designed to predict the occurrence of a heart attack for the purposes of preventive medicine. It used the combination and comparison of multiple AI methods and techniques to determine a personalized heart attack probability based on a wide range of patient characteristics and, from a computational point of view, determine the minimum set of characteristics necessary to do so. When applied to a specific patient, this represents progress in this field of research, resulting in improvements in preclinical care and diagnostics, as well as predictive accuracy in preventive medicine. After an initial selection based on the authors' knowledge and experience, four solutions turned out to be the best: linear support vector machine (Linear SVC), logistic regression, k-nearest neighbors algorithm (KNN, k-NN), and random forest. A comparison of the models developed in the study shows that models based on logistic regression proved to be the most accurate, although their predictive value is moderate, but sufficient for the initial screening diagnosis—selecting patients who require further, more accurate testing. In addition, this can be performed based on a reduced set of parameters, particularly heart rate, age, BMI, and cholesterol. This allows the development of a prevention strategy based on modifiable factors (e.g., in the form of diet, activity modification, or a hybrid combining different factors) combined with the monitoring of heart attack risk by the proposed system. The novelty and contribution of the described system lies in the use of AI for a widely available, cheap, and quick predictive analysis of cardiovascular functions in a group of patients classified as at risk, and over time in all patients as a standard periodic examination qualifying them for further, more advanced diagnosis of heart diseases.
engineering, electrical & electronic,computer science, information systems,physics, applied
What problem does this paper attempt to address?
### Problems the Paper Attempts to Solve The main purpose of this paper is to develop and test an artificial intelligence (AI)-based tool for predicting the risk of heart attacks to support the development of preventive medicine. Specifically, the focus of the research is on the early detection of heart disease risk and taking preventive measures before the condition worsens. By combining and comparing various AI methods and techniques, the study aims to determine the personalized probability of heart attacks based on a wide range of patient characteristics and, from a computational perspective, identify the minimal feature set required to achieve this goal. ### Specific Objectives 1. **Improve Existing AI Solutions**: Enhance existing AI solutions and develop new AI technologies to support preventive cardiology, particularly the prevention of heart attacks. 2. **Personalized Risk Assessment**: Provide personalized heart attack risk assessments based on a wide range of patient characteristics. 3. **Reduce Feature Set**: Identify the minimal feature set required to achieve accurate predictions, thereby simplifying the data collection process. 4. **Improve Diagnostic Accuracy**: Enhance diagnostic accuracy and predictive precision in preventive medicine, thereby improving the level of clinical care. ### Methods and Results - **Dataset**: A dataset from Kaggle was used, containing 8763 records, each with 26 features. - **Model Selection**: Four optimal models were selected for comparison: Linear Support Vector Machine (Linear SVC), Logistic Regression, k-Nearest Neighbors (KNN), and Random Forest. - **Model Tuning**: Hyperparameter tuning was performed on different models to optimize their performance. - **Results**: The results showed that the model based on Logistic Regression performed the best. Although its predictive values were moderate, they were sufficient for preliminary screening diagnoses. Additionally, the study indicated that age, gender, and smoking status were the most significant risk factors, with smoking being a modifiable factor that can help in formulating preventive strategies.