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.

Risk Factors Selection for SGA Prediction

Feature Selection and Prediction of Small-for-gestational-age Infants

Detecting Biomarkers for Major Adverse Cardiac Events Using SVM with PLS Feature Selection and Extraction.

A data-driven approach to predict Small-for-Gestational-Age infants

Lasso Algorithm and Support Vector Machine Strategy to Screen Pulmonary Arterial Hypertension Gene Diagnostic Markers

SVM Based Predictive Model for SGA Detection.

Multiclass Cancer Classification by Using Fuzzy Support Vector Machine and Binary Decision Tree with Gene Selection

Study on the Application of Classification Tree Model in Screening the Risk Factors of Malignant Tumor

Machine Learning Driven Biomarker Selection for Medical Diagnosis

Machine learning discrimination of Gleason scores below GG3 and above GG4 for HSPC patients diagnosis

Gene Features Selection for Three-Class Disease Classification via Multiple Orthogonal Partial Least Square Discriminant Analysis and S-Plot Using Microarray Data

Feature Selection and Prediction of Sub-health State Using SVM-RFE

The Unsupervised Feature Selection Algorithms Based on Standard Deviation and Cosine Similarity for Genomic Data Analysis

Analyzing Longitudinal Health Screening Data with Feature Ensemble and Machine Learning Techniques: Investigating Diagnostic Risk Factors of Metabolic Syndrome for Chronic Kidney Disease Stages 3a to 3b

Optimized Clinical Feature Analysis for Improved Cardiovascular Disease Risk Screening

FS–GBDT: identification multicancer-risk module via a feature selection algorithm by integrating Fisher score and GBDT

Machine Learning for Selecting Important Clinical Markers of Imaging Subgroups of Cerebral Small Vessel Disease Based on a Common Data Model

Comparative analysis of supervised learning algorithms for prediction of cardiovascular diseases

Fasting Blood Glucose Change Prediction Model Based on Medical Examination Data and Data Mining Techniques

Robust biomarker screening from gene expression data by stable machine learning-recursive feature elimination methods

An ensemble learning-based feature selection algorithm for identification of biomarkers of renal cell carcinoma