Enhancing Survival Analysis Model Selection Through XAI(t) in Healthcare
APPLIED SCIENCES-BASEL(2024)
摘要
Artificial intelligence algorithms have become extensively utilized in survival analysis for high-dimensional, multi-source data. However, due to their complexity, these methods often yield poorly interpretable outcomes, posing challenges in the analysis of several conditions. One of these conditions is obstructive sleep apnea, a sleep disorder characterized by the simultaneous occurrence of comorbidities. Survival analysis provides a potential solution for assessing and categorizing the severity of obstructive sleep apnea, aiding personalized treatment strategies. Given the critical role of time in such scenarios and considering limitations in model interpretability, time-dependent explainable artificial intelligence algorithms have been developed in recent years for direct application to basic Machine Learning models, such as Cox regression and survival random forest. Our work aims to enhance model selection in OSA survival analysis using time-dependent XAI for Machine Learning and Deep Learning models. We developed an end-to-end pipeline, training several survival models and selecting the best performers. Our top models—Cox regression, Cox time, and logistic hazard—achieved good performance, with C-index scores of 0.81, 0.78, and 0.77, and Brier scores of 0.10, 0.12, and 0.11 on the test set. We applied SurvSHAP methods to Cox regression and logistic hazard to investigate their behavior. Although the models showed similar performance, our analysis established that the results of the log hazard model were more reliable and useful in clinical practice compared to those of Cox regression in OSA scenarios.
更多查看译文
关键词
survival analysis,Machine Learning,Deep Learning,explainable artificial intelligence,rehabilitation medicine,obstructive sleep apnea
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
数据免责声明
页面数据均来自互联网公开来源、合作出版商和通过AI技术自动分析结果,我们不对页面数据的有效性、准确性、正确性、可靠性、完整性和及时性做出任何承诺和保证。若有疑问,可以通过电子邮件方式联系我们:report@aminer.cn