Machine learning prediction of tuberculosis mortality: a comparative analysis of random survival forest and cox regression models

利用机器学习预测结核病死亡率:随机生存森林模型与Cox回归模型的比较分析

阅读:1

Abstract

BACKGROUND: Survival analysis is widely used to predict time-to-event outcomes, with the Cox regression model being a standard approach. However, machine learning methods such as Random Survival Forests (RSF) can capture complex, non-linear relationships that traditional models may miss. OBJECTIVE: This study compared the predictive performance of RSF and Cox regression in modelling tuberculosis (TB) mortality. METHODS: We conducted a retrospective study of TB patients treated at the East London Central Clinic in South Africa. Patient data included demographic, clinical, and treatment-related variables. Model performance was evaluated using five metrics (C-index, Brier Score, Integrated Brier Score, Integrated Absolute Error, and Integrated Squared Error) along with time-dependent receiver operating characteristic (ROC) curves. Variable importance was assessed to identify key predictors. RESULTS: The RSF model consistently outperformed the Cox model across all evaluation metrics. RSF achieved a higher integrated AUC (0.815 vs. 0.652) and lower prediction error (IBS = 0.235 vs. 0.261). Important predictors of mortality included age, sex, weight, and disease class, with RSF capturing their time-dependent effects more accurately. The cumulative case/dynamic control ROC curve showed the strongest predictive accuracy at 120 days (AUC = 0.856). CONCLUSION: RSF demonstrated superior predictive accuracy compared with Cox regression in modelling TB mortality. Its ability to account for non-linear and time-dependent effects makes it a potentially useful tool for improving risk prediction and guiding patient management in TB care. CLINICAL TRIAL: Not applicable. SUPPLEMENTARY INFORMATION: The online version contains supplementary material available at 10.1186/s12879-026-13109-9.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。