Open AccessA survival model generalized to regression learning algorithms
Author(s) -
Yuanfang Guan,
Hongyang Li,
Daiyao Yi,
Dongdong Zhang,
Changchang Yin,
Keyu Li,
Ping Zhang
Publication year - 2021
Publication title -
nature computational science
Language(s) - English
Resource type - Journals
ISSN - 2662-8457
DOI - 10.1038/s43588-021-00083-2
Subject(s) - computer science , regression analysis , regression , algorithm , artificial intelligence , machine learning , mathematics , statistics
Survival prediction is an important problem that is encountered widely in industry and medicine. Despite the explosion of artificial intelligence technologies, no uniformed method allows the application of any type of regression learning algorithm to a survival prediction problem. Here, we present a statistical modeling method that is generalized to all types of regression learning algorithm, including deep learning. We present its empirical advantage when it is applied to traditional survival problems. We demonstrate its expanded applications in different types of regression learning algorithm, such as gradient boosted trees, convolutional neural networks and recurrent neural networks. Additionally, we demonstrate its application in clinical informatic data, pathological images and the hardware industry. We expect that this algorithm will be widely applicable for diverse types of survival data, including discrete data types and those suitable for deep learning such as those with time or spatial continuity.