Survival Analysis Algorithms based on Decision Trees with Weighted Log-rank Criteria

Vasilev, I.; Petrovskiy, M.; Mashechkin, I.

Авторы: Vasilev Iulii, Petrovskiy Mikhail, Mashechkin Igor
Сборник: In Proceedings of the 11th International Conference on Pattern Recognition Applications and Methods
Год издания: 2022
Место издания: SciTePress
Первая страница: 132
Последняя страница: 140
DOI: 10.5220/0010987100003122
Аннотация: Survival Analysis is an important tool to predict time-to-event in many applications, including but not limited to medicine, insurance, manufacturing, and others. The state-of-the-art statistical approach is based on Cox proportional hazards. Though, from a practical point of view, it has several important disadvantages, such as strong assumptions on proportional over time hazard functions and linear relationship between time-independent covariates and the log hazard. Another technical issue is the inability to deal with missing data directly. To overcome these disadvantages machine learning survival models based on recursive partitioning approach have been developed recently. In this paper, we propose a new survival decision tree model that uses weighted log-rank split criteria. Unlike traditional log-rank criteria, the weighted ones allow giving different priority to events with different time stamps. It works with missing data directly while searching for the best splitting point, its size is controlled by p-value threshold with Bonferroni adjustment and quantile-based discretization is used to decrease the number of potential candidates for splitting points. Also, we investigate how to improve the accuracy of the model with a bagging ensemble of the proposed decision tree models.We introduce an experimental comparison of the proposed methods against Cox proportional risk regression and existing tree-based survival models and their ensembles.According to the obtained experimental results, the proposed methods show better performance on several benchmark public medical datasets in terms of Concordance index and Integrated Brier Score metrics.
Добавил в систему: Васильев Юлий Алексеевич

	ИСТИНА	Войти в систему Регистрация
	ИСТИНА ЦЭМИ РАН
	Главная Поиск Статистика О проекте Помощь

ИСТИНА

ИСТИНА ЦЭМИ РАН

Survival Analysis Algorithms based on Decision Trees with Weighted Log-rank Criteriaстатья

Прикрепленные файлы