Deep-learning in survival analysis

In this paper, hospitalisation duration is modelled using traditional survival model, machine-learning models and deep-learning models. Machine-learning and deep-learning algorithms typically assume that all event of interest are known at the time of modelling. However, hospitalisation duration i...

Full description

Saved in:
Bibliographic Details
Main Author: Ho, Jeff
Other Authors: Xiang Liming
Format: Final Year Project
Language:English
Published: Nanyang Technological University 2021
Subjects:
Online Access:https://hdl.handle.net/10356/148490
Tags: Add Tag
No Tags, Be the first to tag this record!
Institution: Nanyang Technological University
Language: English
Description
Summary:In this paper, hospitalisation duration is modelled using traditional survival model, machine-learning models and deep-learning models. Machine-learning and deep-learning algorithms typically assume that all event of interest are known at the time of modelling. However, hospitalisation duration is a time-to-event data with right-censoring (not all events are known at the time of modelling). Hence, specific techniques were employed to deal with this inconsistency. Subsequently, the various models are evaluated using the Concordance-index (C-index). It is a ranking evaluation metrics that can account for censored observation. The empirical results showed that the deep-learning model is best in predicting hospitalisation despite the small dataset. This paper can be further improved by incorporating geo-spatial data in the analysis.