51. Cluster-Boosted Multi-Task Learning Framework for Survival Analysis
- Author
-
Mark Chignell, Lu Wang, Haoyan Jiang, and Nipon Charoenkitkarn
- Subjects
Structure (mathematical logic) ,Computer science ,Generalization ,Event (computing) ,business.industry ,Multi-task learning ,02 engineering and technology ,Disease cluster ,Machine learning ,computer.software_genre ,01 natural sciences ,010104 statistics & probability ,Range (mathematics) ,ComputingMethodologies_PATTERNRECOGNITION ,020204 information systems ,0202 electrical engineering, electronic engineering, information engineering ,Labeled data ,Artificial intelligence ,0101 mathematics ,business ,computer ,Survival analysis - Abstract
Accurately predicting the time to an event of interest is an important problem in a wide range of real-world applications. However, prediction is often difficult because many medical datasets have a large number of unlabeled (“censored”) instances because labeling is costly and time consuming. Survival analysis focuses on labeled data to predict the time to an event of interest, such as time of death, or conversion to a different stage in a progressive disease. Grouping structure, which naturally exists in medical datasets, can be exploited to improve generalization performance by learning multiple related survival prediction tasks for subgroups collaboratively. Thus a multi-task learning framework can connect multiple survival prediction tasks (for different subgroups) and learn them simultaneously. In order to take into account both censored information, as well as discover the grouping structure, we propose a novel cluster-boosted multitask learning framework for survival analysis that boosts survival prediction performance. We develop an efficient algorithm and demonstrate the performance of the proposed cluster-boosted multi-task survival analysis method on The Cancer Genome Atlas (TCGA) dataset. Our results show that the proposed approach can significantly improve prediction performance in survival analysis while also identifying different subgroups of cancer patients.
- Published
- 2020
- Full Text
- View/download PDF