Recurrent Neural Networks for Multivariate Time Series with Missing Values
Zhengping CheSanjay PurushothamKyunghyun ChoDavid SontagYan Liu
Develops GRU-D, a recurrent neural network architecture that incorporates masking indicators and elapsed time intervals directly into recurrent transitions to exploit informative missing patterns in multivariate clinical time series.
Multivariate time series data in domains such as healthcare routinely contain missing values whose patterns often correlate with target outcomes, a phenomenon known as informative missingness. Conventional approaches either discard incomplete records or apply separate imputation steps before prediction; both strategies fail to exploit missingness patterns fully and can degrade performance when missing rates exceed 80 percent, as is common in intensive-care records.
The article develops and evaluates a recurrent neural-network architecture, called GRU-D, that directly incorporates two representations of missingness—binary masks and elapsed-time intervals—into a gated recurrent unit. Trainable exponential decay terms are applied to both the input features and the hidden states, allowing the model to learn how much weight to place on recent observations versus default values for each variable. The resulting end-to-end model is trained by standard back-propagation and requires no external imputation stage.
On the MIMIC-III and PhysioNet clinical datasets, GRU-D attained the highest area-under-the-ROC-curve scores for in-hospital mortality prediction (0.8527 and 0.8424, respectively) and for multi-task diagnosis classification, outperforming logistic regression, support-vector machines, random forests, and several GRU variants that rely on mean, forward, or matrix-based imputation. The advantage widened as training-set size increased and remained stable across synthetic datasets with varying degrees of informative missingness. The model also produced accurate predictions earlier in a patient’s stay, matching the 48-hour performance of the strongest non-recurrent baseline after only 36 hours of data.
These results indicate that explicitly modeling missingness inside the network improves both accuracy and timeliness of clinical forecasts without added computational cost. The learned decay parameters further reveal which variables carry the most predictive information when absent, offering clinicians interpretable insight into data-collection priorities.
The authors note that gains are modest when missingness is unrelated to outcomes and that the method has been tested only on retrospective observational data. They recommend extending the decay mechanism to explicitly missing-not-at-random regimes and conducting prospective validation before routine clinical deployment.
- Paper: Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling, Junyoung Chung et al. (2014). This paper establishes the core empirical comparison and operational mechanics of Gated Recurrent Units (GRUs), which provide the architectural foundation modified by GRU-D.
- Paper: Long Short-Term Memory, Sepp Hochreiter et al. (1997). It introduces gated recurrent architectures to resolve vanishing gradients in sequential modeling, establishing the fundamental gating concepts used in GRU-D.
- Paper: MissForest - non-parametric missing value imputation for mixed-type data, Daniel J. Stekhoven et al. (2011). It presents a benchmark external imputation technique for mixed tabular data, representing the standard multi-stage imputation baselines that the source paper seeks to replace with an end-to-end recurrent model.
- Paper: A Critical Review of Recurrent Neural Networks for Sequence Learning, Zachary C. Lipton et al. (2015). It provides an essential survey of recurrent neural network architectures, sequence learning dynamics, and gating formulations relevant to time-series modeling.
- Paper: On the difficulty of training recurrent neural networks, Razvan Pascanu et al. (2012). It analyzes the mathematical mechanisms of vanishing and exploding gradients in recurrent networks, motivating the gated recurrent structures adapted by the source.
- Paper: Scalable and accurate deep learning with electronic health records, A. Rajkomar et al. (2018). This work extends deep learning on electronic health records by scaling end-to-end neural architectures across uncurated clinical records and multiple predictive clinical tasks.
- Paper: Gate-variants of Gated Recurrent Unit (GRU) neural networks, Rahul Dey et al. (2017). It systematically simplifies the internal gating equations of GRUs to reduce computational parameter overhead while preserving sequential modeling performance.
- Paper: Deep learning for time series classification: a review, Hassan Ismail Fawaz et al. (2018). It provides an extensive benchmark review comparing diverse deep learning architectures for univariate and multivariate time-series classification.
- Paper: Modeling Long- and Short-Term Temporal Patterns with Deep Neural Networks, Guokun Lai et al. (2017). It develops an advanced multivariate time-series neural architecture combining convolutional and recurrent attention layers to jointly model short-term and long-term periodic patterns.
- Paper: Connecting the Dots: Multivariate Time Series Forecasting with Graph Neural Networks, Zonghan Wu et al. (2020). It advances multivariate time-series modeling beyond recurrent units by learning latent inter-variable dependencies directly using graph neural networks.
- Paper: An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modeling, Shaojie Bai et al. (2018). It systematically evaluates whether temporal convolutional networks can outperform recurrent networks across classical sequential modeling benchmarks.
