LAPSE:2023.2626v1
Published Article

LAPSE:2023.2626v1
Predictive Classifier for Cardiovascular Disease Based on Stacking Model Fusion
February 21, 2023
Abstract
The etiology of cardiovascular disease is still an unsolved world problem, and high morbidity, disability, and mortality are the main characteristics of cardiovascular diseases. There is, therefore, a need for effective and rapid early prediction of likely outcomes in patients with cardiovascular disease using artificial intelligence (AI) techniques. The Internet of Things (IoT) is becoming a catalyst for enhancing the capabilities of AI applications. Data are collected through IoT sensors and analyzed and predicted using machine learning (ML). Existing traditional ML models do not handle data inequities well and have relatively low model prediction accuracy. To address this problem, considering the data observation mechanism and training methods of different algorithms, this paper proposes an ensemble framework based on stacking model fusion, from Support Vector Machines (SVM), K-Nearest Neighbor (KNN), Logistic Regression (LR), Random Forest (RF), Extra Tree (ET), Gradient Boosting Decision Tree (GBDT), XGBoost, LightGBM, CatBoost, and Multilayer Perceptron (MLP) (10 classifiers to select the optimal base learners). In order to avoid the overfitting phenomenon generated by the base learners, we use the Logistic Regression (LR) simple linear classifier as the meta learner. We validated the proposed algorithm using a fused Heart Dataset from several UCI machine learning repositories and another publicly available Heart Attack Dataset, and compared it with 10 single classifier models. The experimental results show that the proposed stacking classifier outperforms other classifiers in terms of accuracy and applicability.
The etiology of cardiovascular disease is still an unsolved world problem, and high morbidity, disability, and mortality are the main characteristics of cardiovascular diseases. There is, therefore, a need for effective and rapid early prediction of likely outcomes in patients with cardiovascular disease using artificial intelligence (AI) techniques. The Internet of Things (IoT) is becoming a catalyst for enhancing the capabilities of AI applications. Data are collected through IoT sensors and analyzed and predicted using machine learning (ML). Existing traditional ML models do not handle data inequities well and have relatively low model prediction accuracy. To address this problem, considering the data observation mechanism and training methods of different algorithms, this paper proposes an ensemble framework based on stacking model fusion, from Support Vector Machines (SVM), K-Nearest Neighbor (KNN), Logistic Regression (LR), Random Forest (RF), Extra Tree (ET), Gradient Boosting Decision Tree (GBDT), XGBoost, LightGBM, CatBoost, and Multilayer Perceptron (MLP) (10 classifiers to select the optimal base learners). In order to avoid the overfitting phenomenon generated by the base learners, we use the Logistic Regression (LR) simple linear classifier as the meta learner. We validated the proposed algorithm using a fused Heart Dataset from several UCI machine learning repositories and another publicly available Heart Attack Dataset, and compared it with 10 single classifier models. The experimental results show that the proposed stacking classifier outperforms other classifiers in terms of accuracy and applicability.
Record ID
Keywords
cardiovascular disease, SHAP values, stacking model fusion
Subject
Suggested Citation
Liu J, Dong X, Zhao H, Tian Y. Predictive Classifier for Cardiovascular Disease Based on Stacking Model Fusion. (2023). LAPSE:2023.2626v1
Author Affiliations
Liu J: College of Intelligence Equipment, Shandong University of Science and Technology, Tai’an 271000, China
Dong X: College of Intelligence Equipment, Shandong University of Science and Technology, Tai’an 271000, China [ORCID]
Zhao H: College of Intelligence Equipment, Shandong University of Science and Technology, Tai’an 271000, China
Tian Y: College of Intelligence Equipment, Shandong University of Science and Technology, Tai’an 271000, China [ORCID]
Dong X: College of Intelligence Equipment, Shandong University of Science and Technology, Tai’an 271000, China [ORCID]
Zhao H: College of Intelligence Equipment, Shandong University of Science and Technology, Tai’an 271000, China
Tian Y: College of Intelligence Equipment, Shandong University of Science and Technology, Tai’an 271000, China [ORCID]
Journal Name
Processes
Volume
10
Issue
4
First Page
749
Year
2022
Publication Date
2022-04-13
ISSN
2227-9717
Version Comments
Original Submission
Other Meta
PII: pr10040749, Publication Type: Journal Article
Record Map
Published Article

LAPSE:2023.2626v1
This Record
External Link

https://doi.org/10.3390/pr10040749
Publisher Version
Download
Meta
Record Statistics
Record Views
416
Version History
[v1] (Original Submission)
Feb 21, 2023
Verified by curator on
Feb 21, 2023
This Version Number
v1
Citations
Most Recent
This Version
URL Here
http://psecommunity.org/LAPSE:2023.2626v1
Record Owner
Auto Uploader for LAPSE
Links to Related Works
