Prediction for cardiovascular diseases based on laboratory data: An analysis of random forest model

Abstract

Background To establish a prediction model for cardiovascular diseases (CVD) in the general population based on random forests. Methods A retrospective study involving 498 subjects was conducted in Xi'an Medical University between 2011 and 2018. The random forest algorithm was used to screen out the variables that greatly affected the CVD prediction and to establish a prediction model. The important variables were included in the multifactorial logistic regression analysis. The area under the curve (AUC) was compared between logistic regression model and random forest model. Results The random forest model revealed the variables, including the age, body mass index (BMI), fasting blood glucose (FBG), diastolic blood pressure (DBP), triglyceride (TG), systolic blood pressure (SBP), total cholesterol (TC), waist circumference, and high‐density lipoprotein‐cholesterol (HDL‐C), were more significant for CVD prediction; the AUC was 0.802 in CVD prediction. Multifactorial logistic regression analysis indicated that the risk factors for CVD included the age [odds ratio (OR): 1.14, 95% confidence intervals (CI): 1.10‐1.17, P < .001], BMI (OR: 1.13, 95% CI: 1.06‐1.20, P < .001), TG (OR: 1.11, 95% CI: 1.02‐1.22, P = .023), and DBP (OR: 1.04, 95% CI: 1.02‐1.06, P = .001); the AUC was 0.843 in CVD prediction. The established logistic regression prediction model was Logit P = Log[P/(1 − P)] = −11.47 + 0.13 × age + 0.12 × BMI + 0.11 × TG + 0.04 × DBP; P = 1/[1 + exp(−Logit P)]. People were prone to develop CVD at the time of P > .51. Conclusions A prediction model for CVD is developed in the general population based on random forests, which provides a simple tool for the early prediction of CVD.

Risk Assessment of Pulmonary Metastasis for Cervical Cancer Patients by Ensemble Learning Models: A Large Population Based Real-World Study

“…Unlike other established risk assessment methods, the random forest algorithm can achieve higher accuracy in the disease prediction by using bootstrap aggregation and randomization of predictors. 26 , 27 Interestingly, using machine learning algorithms, we found that the age, tumor size, histological grade, organ-site metastasis, pathologic types, and TNM stage were indicated to be the most contributive risk factors of PM, consistent with the results of linear regression analysis. However, in this study, the aforementioned factors were included, and correlations within those clinicopathological characteristics were considered, which accounted for a better result than multivariable logistic regression.…”

Development of a random forest model for hypotension prediction after anesthesia induction for cardiac surgery

“…As shown in Figure 2 , the variables that achieved peak values were placed at the top node location, closest to the root of the tree[ 8 ]. The value of a variable was indicated by the mean decrease in the Gini (MDG) index[ 14 ]. It has been established that the higher the index value, the more important is the value given by the variable.…”

Identifying Coronary Artery Lesions by Feature Analysis of Radial Pulse Wave: A Case-Control Study

“…In the DT, the tree models are composed of nodes and directed edges, and optimal classifications are achieved through learning processes that involve recursive feature selection, DT generation, and pruning [28]. RF uses a large series of decision trees with low reciprocal correlation and randomly selected features using the method of bagging to obtain more precise and stable classifications and predictions [29].…”

“…Existing research [36] has demonstrated that machine learning algorithms can produce accurate results when sorting epidemiological data. The random forest, a key data mining method in machine learning field that depends on a computer to learn all the complicated and nonlinear interactions among variables through minimization of errors between observed and predicted outcomes, can achieve a higher accuracy in the disease prediction by using bootstrap aggregation and randomization of predictors [29]. Besides, RF models are less prone to overfitting.…”