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.

Cardiovascular disease detection using a novel stack-based ensemble classifier with aggregation layer, DOWA operator, and feature transformation

Predictive Classifier for Cardiovascular Disease Based on Stacking Model Fusion

Forecasting Heart Disease Risk with a Stacking-Based Ensemble Machine Learning Method

Machine Learning-Based Cardiovascular Disease Detection Using Optimal Feature Selection

An Improved Heart Disease Prediction Using Stacked Ensemble Method

Ensemble Framework for Cardiovascular Disease Prediction

Comparative analysis of supervised learning algorithms for prediction of cardiovascular diseases

Integrated Approach for ECG Signal Classification Using Fused Slantlet Transform and Autoregressive Features

Enhanced feature selection and ensemble learning for cardiovascular disease prediction: hybrid GOL2-2 T and adaptive boosted decision fusion with babysitting refinement

Efficient Data-Driven Machine Learning Models for Cardiovascular Diseases Risk Prediction

Detection of Diseases in Pandemic: A Predictive Approach Using Stack Ensembling on Multi-Modal Imaging Data

A Robust Multilevel DWT Densely Network for Cardiovascular Disease Classification

An efficient cardio vascular disease prediction using multi-scale weighted feature fusion-based convolutional neural network with residual gated recurrent unit

Multilayer Perceptron Neural Network with Arithmetic Optimization Algorithm-Based Feature Selection for Cardiovascular Disease Prediction

Cardiovascular disease diagnosis: a holistic approach using the integration of machine learning and deep learning models

Detection of Cardiovascular Diseases Using Data Mining Approaches: Application of an Ensemble-Based Model

Machine Learning-Based Predictive Models for Detection of Cardiovascular Diseases

Feature selection for medical diagnosis: Evaluation for using a hybrid Stacked-Genetic approach in the diagnosis of heart disease

CB-HDM: ECG signal based heart disease classification using convolutional block attention assisted hybrid deep Maxout network

Detection of cardiovascular disease cases using advanced tree-based machine learning algorithms

Cardiovascular disease detection from cardiac arrhythmia ECG signals using artificial intelligence models with hyperparameters tuning methodologies