Use of machine learning models to predict mortality in dialysis patients

作者：Junmin Huang, Zhiyi Chen, Hongying Luo, Junhao Song, Zhuming Bi, Keyu Chen, Xiu Xian Chia, Ming Liu, Tianyang Wang, Benji Peng, Zihan Wei, Zhiqing Huang, Zhihang Li, Xincheng Liu, Hongjiu Zhou, Weihuang Zhang, Wen Wen, Mian-Na Luo, Shujun Wang, Huafeng Liu, Chunjie Tian, Jibin Guan, Joe Yeong, Yongzhi Xu, Peng Wang, Junfeng Hao · 发表于：Frontiers in Public Health · 年份：2025 · DOI：10.3389/fpubh.2025.1683285 · 被引用次数：2 · 研究领域：Machine Learning in Healthcare、Artificial Intelligence in Healthcare、Dialysis and Renal Disease Management

Background: Mortality among maintenance hemodialysis patients remains high, and traditional statistical models often fail to capture complex clinical relationships. This study aimed to systematically develop, compare, and validate 19 machine learning algorithms for predicting all-cause mortality in maintenance hemodialysis patients. Methods: This retrospective study included data from 538 maintenance hemodialysis patients (2018.1-2023.12), with 70% used for training and 30% for testing. Each model underwent hyperparameter optimization based on three performance metrics (accuracy, F1-score, and ROC Area Under the Curve [AUC]) to evaluate the impact of different clinical priorities. Results: Gradient boosting models demonstrated consistent superiority, with performance outcomes highly sensitive to the selected optimization target. XGBoost optimized for accuracy achieved an F1 score of 0.683 and a ROC AUC of 0.899. AdaBoost optimized for F1 score attained the highest ROC AUC of 0.903 and an F1 score of 0.682. AdaBoost also demonstrated robust performance across optimization strategies, suggesting its suitability for clinical implementation where balanced risk prediction is essential. Conclusion: A systematic ML framework can yield tailored, high-performing models for mortality risk stratification in maintenance hemodialysis patients, with significant potential to enhance identification and management of high-risk individuals in clinical practice. Clinical trial number Registry: ...