Abstract:BACKGROUND: With the advent of artificial intelligence technology, machine learning algorithms have been widely used in the area of disease prediction. OBJECTIVE: Cardiovascular disease (CVD) seriously jeopardizes human health worldwide, thereby needing the establishment of an effective CVD prediction model that can be of great significance for controlling the risk of the disease and safeguarding the physical and mental health of the population. METHODS: Considering the UCI heart disease dataset as an example, initially, a single machine learning prediction model was constructed. Subsequently, six methods such as Pearson, chi-squared, RFE and LightGBM were comprehensively used for the feature screening. On the basis of the base classifiers, Soft Voting fusion and Stacking fusion was carried out to build a prediction model for cardiovascular diseases, in order to realize an early warning and disease intervention for high-risk populations. To address the data imbalance problem, the SMOTE method was adopted to process the data set, and the prediction effect of the model was analyzed using multi-dimensional and multi-indicators. RESULTS: In the single classifier model, the MLP algorithm performed optimally on the preprocessed heart disease dataset. After feature selection, five features eliminated. The ENSEM_SV algorithm that combines the base classifiers to determine the prediction results by soft voting on the results of the classifiers achieved the optimal value on five metrics such as Accuracy, Jaccard_Score, Hamm_Loss, AUC, etc., and the AUC value reached 0.951. The RF, ET, GBDT, and LGB algorithms were employed in the first stage sub-model composed of base classifiers. The AB algorithm was selected as the second stage model, and the ensemble algorithm ENSEM_ST, obtained by Stacking fusion of the two stages exhibited the best performance on 7 indicators such as Accuracy, Sensitivity, F1_Score, Mathew_Corrcoef, etc., and the AUC reached 0.952. Furthermore, a comparison of the algorithms' classification effects based on different training set occupancy was carried out. The results indicated that the prediction performance of both the fusion models was better than the single models, and the overall effect of ENSEM_ST fusion was stronger than the ENSEM_SV fusion. CONCLUSIONS: The fusion model established in this study improved the overall classification accuracy and stability of the model to a significant extent. It has a good application value in the predictive analysis of CVD diagnosis, and can provide a valuable reference in the disease diagnosis and intervention strategies.

High-risk Prediction of Cardiovascular Diseases via Attention-based Deep Neural Networks.

Heart Disease Prediction Based on the Embedded Feature Selection Method and Deep Neural Network

Highrisk Prediction from Electronic Medical Records via Deep Attention Networks

Artificial Neural Network Based Model for Cardiovascular Risk Stratification in Hypertension.

Improving Cardiovascular Risk Prediction Through Machine Learning Modelling of Irregularly Repeated Electronic Health Records

Deep Featured Adaptive Dense Net Convolutional Neural Network Based Cardiac Risk Prediction in Big Data Healthcare Environment

Application of an end-to-end model with self-attention mechanism in cardiac disease prediction

Prediction of cardiovascular and cerebrovascular diseases based on machine learning models

Deep Learning Based Healthcare Method for Effective Heart Disease Prediction

Enhanced Character-Level Deep Convolutional Neural Networks for Cardiovascular Disease Prediction.

Deep Cardiovascular Disease Prediction with Risk Factors Powered Bi-attention.

Disease Prediction by Machine Learning Over Big Data From Healthcare Communities

Prediction of cardiovascular diseases by integrating multi-modal features with machine learning methods

Deep Cardiovascular Disease Prediction with Risk Factors Powered Bi-Attention

Efficient Prediction Model for Cardiovascular Disease Using Deep Learning Techniques

Research on Early Warning Model of Cardiovascular Disease Based on Computer Deep Learning

Prediction of Cardiovascular Disease using DeepLearning Algorithm

Predicting blood pressure from face videos using face diagnosis theory and deep neural networks technique

An Efficient Prediction Method for Coronary Heart Disease Risk Based on Two Deep Neural Networks Trained on Well-Ordered Training Datasets

Risk prediction of heart diseases in patients with breast cancer: A deep learning approach with longitudinal electronic health records data

Comparative analysis of supervised learning algorithms for prediction of cardiovascular diseases