Cluster-Boosted Multi-Task Learning Framework for Survival Analysis
Conference proceedings article
Authors/Editors
Strategic Research Themes
Publication Details
Author list: Wang, Lu; Chignell, Mark; Jiang, Haoyan; Charoenkitkarn, Nipon;
Publisher: Hindawi
Publication year: 2020
Start page: 255
End page: 262
Number of pages: 8
ISBN: 9781728195742
ISSN: 0146-9428
eISSN: 1745-4557
Languages: English-Great Britain (EN-GB)
View in Web of Science | View on publisher site | View citing articles in Web of Science
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. © 2020 IEEE.
Keywords
Cluster-boosted framework, Multi-task learning, Prediction of death time
Contact Information