CatBoost algorithm and Bayesian network model analysis based on risk prediction of cardiovascular and cerebro vascular diseases
10.13481/j.1671-587X.202404019
- VernacularTitle:基于心脑血管疾病发病风险预测的CatBoost算法和贝叶斯网络模型分析
- Author:
Aimin WANG
1
;
Fenglin WANG
;
Yiming HUANG
;
Yaqi XU
;
Wenjing ZHANG
;
Xianzhu CONG
;
Weiqiang SU
;
Suzhen WANG
;
Mengyao GAO
;
Shuang LI
;
Yujia KONG
;
Fuyan SHI
;
Enxue TAO
Author Information
1. 山东第二医科大学公共卫生学院卫生统计学教研室,山东潍坊 261053
- Keywords:
Cardiovascular and cerebrovascular disease;
CatBoost algorithm;
Bayesian network;
Risk inference
- From:
Journal of Jilin University(Medicine Edition)
2024;50(4):1044-1054
- CountryChina
- Language:Chinese
-
Abstract:
Objective:To screen the main characteristic variables affecting the incidence of cardiovascular and cerebrovascular diseases,and to construct the Bayesian network model of cardiovascular and cerebrovascular disease incidence risk based on the top 10 characteristic variables,and to provide the reference for predicting the risk of cardiovascular and cerebrovascular disease incidence.Methods:From the UK Biobank Database,315 896 participants and related variables were included.The feature selection was performed by categorical boosting(CatBoost)algorithm,and the participants were randomly divided into training set and test set in the ratio of 7∶3.A Bayesian network model was constructed based on the max-min hill-climbing(MMHC)algorithm.Results:The prevalence of cardiovascular and cerebrovascular diseases in this study was 28.8％.The top 10 variables selected by the CatBoost algorithm were age,body mass index(BMI),low-density lipoprotein cholesterol(LDL-C),total cholesterol(TC),the triglyceride-glucose(TyG)index,family history,apolipoprotein A/B ratio,high-density lipoprotein cholesterol(HDL-C),smoking status,and gender.The area under the receiver operating characteristic(ROC)curve(AUC)for the CatBoost training set model was 0.770,and the model accuracy was 0.764;the AUC of validation set model was 0.759 and the model accuracy was 0.763.The clinical efficacy analysis results showed that the threshold range for the training set was 0.06-0.85 and the threshold range for the validation set was 0.09-0.81.The Bayesian network model analysis results indicated that age,gender,smoking status,family history,BMI,and apolipoprotein A/B ratio were directly related to the incidence of cardiovascular and cerebrovascular diseases and they were the significant risk factors.TyG index,HDL-C,LDL-C,and TC indirectly affect the risk of cardiovascular and cerebrovascular diseases through their impact on BMI and apolipoprotein A/B ratio.Conclusion:Controlling BMI,apolipoprotein A/B ratio,and smoking behavior can reduce the incidence risk of cardiovascular and cerebrovascular diseases.The Bayesian network model can be used to predict the risk of cardiovascular and cerebrovascular disease incidence.
