中国医药科学2025,Vol.15Issue(10):4-10,7.DOI:10.20116/j.issn2095-0616.2025.10.01
基于可解释性机器学习的心力衰竭并发急性肾损伤生存预后模型研究
Research on survival prognosis model for heart failure complicated with acute kidney injury based on interpretable machine learning
摘要
Abstract
Objective This study aimed to develop and validate an interpretable machine learning-based model for predicting the risk of all-cause death during hospitalization in patients with heart failure(HF)complicated with acute kidney injury(AKI).Methods Study data were collected from MIMIC-Ⅳ database,9987 ICU patients were included,and the cases of HF combined with AKI were screened by ICD-9/10.Multiple interpolation was used to process the missing data,combined with Lasso regression and BorutaShap algorithm for feature screening,and finally identified 13 key predictors,including Charson comorbidibility index,acute physiological and chronic health scores,length of stay,etc.The study compared 10 machine learning models(e.g.XGBoost,random Forest,logistic regression,etc.).Results The AUC of random forest,gradient lift and XGBoost is the best(0.78),XGBoost had the highest accuracy rate(71.76%)and the best F1 score(75.00%).The specificity of the decision tree is the most prominent(90.80%).The gradient lift has the best sensitivity(78.03%).SHAP analysis showed that Charson comorbidity index,acute physiological and chronic health scores and respiratory rate were the core factors affecting the risk of death.The study further revealed dynamic differences in the importance of features in specific subpopulations,such as the prominent role of length of stay and bilirubin scores in local prediction.Based on the model results,an online risk assessment platform was developed to provide clinicians with individualized risk probabilities and support early intervention decisions.Conclusion Through interpretable machine learning model,this study provides accurate tools for prognostic management of patients with HF complicated with AKI,and provides auxiliary decision-making for clinical practice.关键词
心力衰竭/急性肾损伤/机器学习/可解释性分析/生存预后模型Key words
Heart failure/Acute kidney injury/Machine learning/Interpretability analysis/Survival prognosis model分类
医药卫生引用本文复制引用
王鑫宇,江洁,陈广新,胡明成..基于可解释性机器学习的心力衰竭并发急性肾损伤生存预后模型研究[J].中国医药科学,2025,15(10):4-10,7.基金项目
黑龙江省省属高校科研基本业务费科研项目(2024-KYYWF-0474). (2024-KYYWF-0474)