arXiv:2507.23491cs.LG2025-07被引 4

用机器学习预测糖尿病患者死亡风险,结果可解释且准确。

Explainable artificial intelligence model predicting the risk of all-cause mortality in patients with type 2 diabetes mellitus

  • 基于10个关键特征构建生存树模型,提升预测精度。
  • 5年死亡预测AUC达0.86,16.8年预测仍超0.82。
  • 通过SHAP分析实现结果可解释,适合临床使用。

2型糖尿病(T2DM)是常见慢性病,显著缩短寿命。本研究分析了554名年龄40-87岁的T2DM患者,随访最长16.8年,其中202人(36%)死亡。通过识别关键生存相关特征,训练并验证多种机器学习模型以预测全因死亡风险。采用分层加性解释(SHAP)增强最佳模型的可解释性。结果显示,融合10个关键特征的额外生存树(EST)模型表现最优,其C-statistic为0.776,5、10、15及16.8年全因死亡预测的AUC分别为0.86、0.80、0.841和0.826。通过SHAP方法揭示模型个体决策逻辑。该模型具备强预测性能与临床可解释性,有助于识别高危患者,支持治疗优化。

原文摘要 · Abstract (English)

Objective. Type 2 diabetes mellitus (T2DM) is a highly prevalent non-communicable chronic disease that substantially reduces life expectancy. Accurate estimation of all-cause mortality risk in T2DM patients is crucial for personalizing and optimizing treatment strategies. Research Design and Methods. This study analyzed a cohort of 554 patients (aged 40-87 years) with diagnosed T2DM over a maximum follow-up period of 16.8 years, during which 202 patients (36%) died. Key survival-associated features were identified, and multiple machine learning (ML) models were trained and validated to predict all-cause mortality risk. To improve model interpretability, Shapley additive explanations (SHAP) was applied to the best-performing model. Results. The extra survival trees (EST) model, incorporating ten key features, demonstrated the best predictive performance. The model achieved a C-statistic of 0.776, with the area under the receiver operating characteristic curve (AUC) values of 0.86, 0.80, 0.841, and 0.826 for 5-, 10-, 15-, and 16.8-year all-cause mortality predictions, respectively. The SHAP approach was employed to interpret the model's individual decision-making processes. Conclusions. The developed model exhibited strong predictive performance for mortality risk assessment. Its clinically interpretable outputs enable potential bedside application, improving the identification of high-risk patients and supporting timely treatment optimization.

糖尿病死亡风险可解释AI机器学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。