Ping Wang, Tian Shi, Chandan Reddy


Survival analysis aims at predicting the time to event of interest along with its probability on longitudinal data. It is commonly used to make predictions for a single specific event of interest at a given time point. However, predicting the occurrence of multiple events of interest simultaneously and dynamically is needed in many real-world applications. An intuitive way to solve this problem is to simply apply the standard survival analysis method independently to each prediction task at each time point. However, it often leads to a sub-optimal solution since the underlying dependencies between these tasks are ignored. This motivates us to analyze these prediction tasks jointly in order to select the common features shared across all the tasks. In this paper, we formulate a temporal ( M ultiple T ime points) M ulti- T ask learning framework (MTMT) for survival analysis problems using tensor representation. More specifically, given a survival dataset and a sequence of time points, which are considered as the monitored time points for the events of interest, we reformulate the survival analysis problem to jointly handle each task at each time point and optimize them simultaneously. We demonstrate the performance of the proposed MTMT model on important real-world datasets, including employee attrition and medical records. We show the superior performance of the MTMT model compared to several state-of-the-art models using standard metrics. We also provide the list of important features selected by our MTMT model thus demonstrating the interpretability of the proposed model.


Ping Wang

Tian Shi

Chandan Reddy

Publication Details

Date of publication:
January 17, 2020
IEEE Transactions on Knowledge and Data Engineering
Page number(s):
Issue Number:
Publication note:

Ping Wang, Tian Shi, Chandan K. Reddy: A Novel Tensor-Based Temporal Multi-Task Survival Analysis Model. IEEE Trans. Knowl. Data Eng. 33(9): 3311-3322 (2021)