Deep State-Space Generative Model For Correlated Time-to-Event Predictions

Capturing the inter-dependencies among multiple types of clinically-critical events is critical not only to accurate future event prediction, but also to better treatment planning. In this work, we propose a deep latent state-space generative model to capture the interactions among different types of correlated clinical events (e.g., kidney failure, mortality) by explicitly modeling the temporal dynamics of patients' latent states. Based on these learned patient states, we further develop a new general discrete-time formulation of the hazard rate function to estimate the survival distribution of patients with significantly improved accuracy. Extensive evaluations over real EMR data show that our proposed model compares favorably to various state-of-the-art baselines. Furthermore, our method also uncovers meaningful insights about the latent correlations among mortality and different types of organ failures.

[1]  Charles Elkan,et al.  Learning to Diagnose with LSTM Recurrent Neural Networks , 2015, ICLR.

[2]  Yan Liu,et al.  Recurrent Neural Networks for Multivariate Time Series with Missing Values , 2016, Scientific Reports.

[3]  Jiayu Zhou,et al.  Multi-task Survival Analysis , 2017, 2017 IEEE International Conference on Data Mining (ICDM).

[4]  Mihaela van der Schaar,et al.  RNN-SURV: A Deep Recurrent Model for Survival Analysis , 2018, ICANN.

[5]  Ole Winther,et al.  Sequential Neural Models with Stochastic Layers , 2016, NIPS.

[6]  Uri Shalit,et al.  Deep Kalman Filters , 2015, ArXiv.

[7]  Uri Shaham,et al.  DeepSurv: personalized treatment recommender system using a Cox proportional hazards deep neural network , 2016, BMC Medical Research Methodology.

[8]  Ole Winther,et al.  A Disentangled Recognition and Nonlinear Dynamics Model for Unsupervised Learning , 2017, NIPS.

[9]  Lei Zheng,et al.  Deep Recurrent Survival Analysis , 2018, AAAI.

[10]  Uri Shalit,et al.  Structured Inference Networks for Nonlinear State Space Models , 2016, AAAI.

[11]  Milos Hauskrecht,et al.  Learning Adaptive Forecasting Models from Irregularly Sampled Multivariate Clinical Data , 2016, AAAI.

[12]  Quan Zhang,et al.  Nonparametric Bayesian Lomax delegate racing for survival analysis with competing risks , 2018, NeurIPS.

[13]  Milos Hauskrecht,et al.  Clinical Time Series Prediction with a Hierarchical Dynamical System , 2013, AIME.

[14]  Jieping Ye,et al.  A Multi-Task Learning Formulation for Survival Analysis , 2016, KDD.

[15]  May D. Wang,et al.  Interpretable Predictions of Clinical Outcomes with An Attention-based Recurrent Neural Network , 2017, BCB.

[16]  Peter Szolovits,et al.  Understanding vasopressor intervention and weaning: risk prediction in a public heterogeneous clinical time series database , 2017, J. Am. Medical Informatics Assoc..

[17]  Yuan Yu,et al.  TensorFlow: A system for large-scale machine learning , 2016, OSDI.

[18]  Andreas Spanias,et al.  Attend and Diagnose: Clinical Time Series Analysis using Attention Models , 2017, AAAI.

[19]  Anthony C. C. Coolen,et al.  Gaussian process regression for survival data with competing risks , 2013, 1312.1591.

[20]  Adler J. Perotte,et al.  Deep Survival Analysis , 2016, MLHC.

[21]  Peter Szolovits,et al.  MIMIC-III, a freely accessible critical care database , 2016, Scientific Data.

[22]  Maximilian Karl,et al.  Deep Variational Bayes Filters: Unsupervised Learning of State Space Models from Raw Data , 2016, ICLR.

[23]  Robert Gray,et al.  A Proportional Hazards Model for the Subdistribution of a Competing Risk , 1999 .

[24]  Alan E Jones,et al.  The Sequential Organ Failure Assessment score for predicting outcome in patients with severe sepsis and evidence of hypoperfusion at the time of emergency department presentation* , 2009, Critical care medicine.

[25]  D.,et al.  Regression Models and Life-Tables , 2022 .

[26]  Lawrence Carin,et al.  Adversarial Time-to-Event Modeling , 2018, ICML.

[27]  Changhee Lee,et al.  DeepHit: A Deep Learning Approach to Survival Analysis With Competing Risks , 2018, AAAI.

[28]  Yee Whye Teh,et al.  Gaussian Processes for Survival Analysis , 2016, NIPS.

[29]  Ahmed M. Alaa,et al.  Deep Multi-task Gaussian Processes for Survival Analysis with Competing Risks , 2017, NIPS.

[30]  Yuan Xue,et al.  Deep Physiological State Space Model for Clinical Forecasting , 2019, ArXiv.

[31]  Jeffrey Dean,et al.  Scalable and accurate deep learning with electronic health records , 2018, npj Digital Medicine.

[32]  Walter F. Stewart,et al.  Doctor AI: Predicting Clinical Events via Recurrent Neural Networks , 2015, MLHC.

[33]  Mihaela van der Schaar,et al.  Attentive State-Space Modeling of Disease Progression , 2019, NeurIPS.

[34]  Mihaela van der Schaar,et al.  Multitask Boosting for Survival Analysis with Competing Risks , 2018, NeurIPS.

[35]  Peter Szolovits,et al.  Predicting intervention onset in the ICU with switching state space models , 2017, CRI.

[36]  Peter Szolovits,et al.  Continuous State-Space Models for Optimal Sepsis Treatment: a Deep Reinforcement Learning Approach , 2017, MLHC.