Prediction of cardiovascular and cerebrovascular diseases based on machine learning models

Yadong Tian,Donghong Yu,Hongji Liu
DOI: https://doi.org/10.54254/2755-2721/46/20241068
2024-03-15
Abstract:Recently we had the fact that cardiovascular disease has become one of the major threats to human life, which leads to the significance of the research around the prevention and cure of such disease. Recently, machine learning algorithms are utilized for the prediction of a certain person who has an illness or not. To verify the effectiveness of predicting cardiovascular disease using machine learning methods, we predict cardiovascular disease given features of a persons life habits and illness history from the Behavioral Risk Factor Surveillance System. Therefore, 5 models are selected, including SVM, logistic regression, decision tree, fully connected network, and XGBoost to evaluate the performance via confusion matrix and ROG curves. Plus, the dataset is highly unbalanced, so we also implemented SMOTETomek re-sampling algorithms to evaluate the models performance on such kinds of datasets. Results exhibited that XGBoost performs the best on the given dataset, hence deep research on improving the performance using XGBoost is highly recommended.
Medicine,Computer Science
What problem does this paper attempt to address?