用动态因果分析提升中风风险预测准确率
Machine learning algorithms to predict stroke in China based on causal inference of time series analysis
- 结合向量自回归与图神经网络构建动态因果特征
- 模型AUC达0.78至0.83,梯度提升效果最佳
- 揭示随时间变化的健康指标对中风的关键影响
本研究纳入11,789名参与者(女性6,334人,占比53.73%;男性5,455人,占比46.27%),平均年龄65岁。采用向量自回归(VAR)模型与图神经网络(GNN)构建动态因果推断特征,对比随机森林、逻辑回归、XGBoost、支持向量机(SVM)、K近邻(KNN)、梯度提升和多层感知机(MLP)等经典分类算法。通过SMOTE处理样本不平衡,并采用分层K折交叉验证。引入动态因果特征后,各模型性能显著提升,所有模型的受试者工作特征曲线下面积(AUC)在0.78至0.83之间,差异具有统计学意义(P < 0.01)。其中梯度提升模型表现最优且最稳定。模型解释与特征重要性分析揭示了与中风风险密切相关的关键健康因素。研究提出一种融合动态因果推断与机器学习的中风风险预测方法,显著提升预测精度,强调动态健康变化对中风风险的预测价值,为未来预防策略提供理论支持与实践指导。
原文摘要 · Abstract (English)
Participants: This study employed a combination of Vector Autoregression (VAR) model and Graph Neural Networks (GNN) to systematically construct dynamic causal inference. Multiple classic classification algorithms were compared, including Random Forest, Logistic Regression, XGBoost, Support Vector Machine (SVM), K-Nearest Neighbor (KNN), Gradient Boosting, and Multi Layer Perceptron (MLP). The SMOTE algorithm was used to undersample a small number of samples and employed Stratified K-fold Cross Validation. Results: This study included a total of 11,789 participants, including 6,334 females (53.73%) and 5,455 males (46.27%), with an average age of 65 years. Introduction of dynamic causal inference features has significantly improved the performance of almost all models. The area under the ROC curve of each model ranged from 0.78 to 0.83, indicating significant difference (P < 0.01). Among all the models, the Gradient Boosting model demonstrated the highest performance and stability. Model explanation and feature importance analysis generated model interpretation that illustrated significant contributors associated with risks of stroke. Conclusions and Relevance: This study proposes a stroke risk prediction method that combines dynamic causal inference with machine learning models, significantly improving prediction accuracy and revealing key health factors that affect stroke. The research results indicate that dynamic causal inference features have important value in predicting stroke risk, especially in capturing the impact of changes in health status over time on stroke risk. By further optimizing the model and introducing more variables, this study provides theoretical basis and practical guidance for future stroke prevention and intervention strategies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。