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.

An intelligent disease prediction and monitoring system using feature selection, multi-neural network and fuzzy rules

Prediction of Heart Disease using Machine Learning Algorithms with Feature Selection Techniques

A decision support system for heart disease prediction based upon machine learning

Machine Learning-Based Model to Predict Heart Disease in Early Stage Employing Different Feature Selection Techniques

A proposed technique for predicting heart disease using machine learning algorithms and an explainable AI method

Predicting heart disease based on an intelligent healthcare monitoring system using HPM-NIA

Optimizing Disease Prediction with Artificial Intelligence Driven Feature Selection and Attention Networks

A smart healthcare monitoring system for heart disease prediction based on ensemble deep learning and feature fusion

Multi Chronic Disease Prediction System Using CNN and Random Forest

Heart Disease Prediction System Using Model Of Machine Learning and Sequential Backward Selection Algorithm for Features Selection

Efficient Heart Disease Prediction Using Hybrid Deep Learning Classification Models

An Effective Disease Prediction System using CRF based Butterfly Optimization, Fuzzy Decision Tree and DBN

A Smart Diseases Diagnosis and Classification Strategy of Electronic Healthcare Application Using Novel Hybrid Artificial Intelligence Approaches

The Efficacy of Machine-Learning-Supported Smart System for Heart Disease Prediction

Implementation of machine learning techniques with big data and IoT to create effective prediction models for health informatics

Heart Disease Prediction System using hybrid model of Multi-layer perception and XGBoost algorithms

Enhanced Cardiovascular Disease Prediction Modelling using Machine Learning Techniques: A Focus on CardioVitalnet

An Intelligent System for Heart Disease Prediction using Adaptive Neuro-Fuzzy Inference Systems and Genetic Algorithm

Heart Disease Prediction System using Ensemble of Machine Learning Algorithms

Comparative analysis of supervised learning algorithms for prediction of cardiovascular diseases

An Investigation on Disease Diagnosis and Prediction by Using Modified K-Mean clustering and Combined CNN and ELM Classification Techniques