RNN-SURV: A Deep Recurrent Model for Survival Analysis
| Autor: | Mihaela van der Schaar, Anton Nemchenko, Eleonora Giunchiglia |
|---|---|
| Rok vydání: | 2018 |
| Předmět: |
Structure (mathematical logic)
Computer science business.industry Deep learning 010501 environmental sciences Machine learning computer.software_genre 01 natural sciences Censoring (statistics) 010104 statistics & probability Survival function Embedding Artificial intelligence State (computer science) 0101 mathematics business computer Survival analysis 0105 earth and related environmental sciences |
| Zdroj: | Artificial Neural Networks and Machine Learning – ICANN 2018 ISBN: 9783030014230 ICANN (3) |
| DOI: | 10.1007/978-3-030-01424-7_3 |
| Popis: | Current medical practice is driven by clinical guidelines which are designed for the “average” patient. Deep learning is enabling medicine to become personalized to the patient at hand. In this paper we present a new recurrent neural network model for personalized survival analysis called rnn-surv. Our model is able to exploit censored data to compute both the risk score and the survival function of each patient. At each time step, the network takes as input the features characterizing the patient and the identifier of the time step, creates an embedding, and outputs the value of the survival function in that time step. Finally, the values of the survival function are linearly combined to compute the unique risk score. Thanks to the model structure and the training designed to exploit two loss functions, our model gets better concordance index (C-index) than the state of the art approaches. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|