The problem of missing values in multivariable time series is a key challengein many applications such as clinical data mining. Although many imputationmethods show their effectiveness in many applications, few of them are designedto accommodate clinical multivariable time series. In this work, we proposemultiple imputation models that capture both cross-sectional information andtemporal correlations. We integrate Gaussian processes with mixture models andintroduce individualized mixing weights to handle the variance of predictiveconfidence of Gaussian process models. The proposed models are compared withseveral state-of-the-art imputation algorithms on both real-world and syntheticdatasets. Experiments show that our best model can provide more accurateimputation than the benchmarks on all of our datasets.
Quick Read (beta)
Mixture-based Multiple Imputation Models for Clinical Data with a Temporal Dimension
††thanks: This work is supported in part by NIH grant R21LM012618.
The problem of missing values in multivariable time series is a key challenge in many applications such as clinical data mining. Although many imputation methods show their effectiveness in many applications, few of them are designed to accommodate clinical multivariable time series. In this work, we propose multiple imputation models that capture both cross-sectional information and temporal correlations. We integrate Gaussian processes with mixture models and introduce individualized mixing weights to handle the variance of predictive confidence of Gaussian process models. The proposed models are compared with several state-of-the-art imputation algorithms on both real-world and synthetic datasets. Experiments show that our best model can provide more accurate imputation than the benchmarks on all of our datasets.
The computational modeling in clinical applications attracts growing interest with the realization that the quantitative understanding of patient pathophysiological progression is crucial to clinical studies . With a comprehensive and precise modeling, we can have a better understanding of a patient’s state, offer more precise diagnosis and provide better individualized therapies . Researchers are increasingly motivated to build more accurate computational models from multiple types of clinical data. However, missing values in clinical data challenge researchers using analytic techniques for modeling, as many of the techniques are designed for complete data.
Traditional strategies used in clinical studies to handle missing values include deleting records with missing values and imputing missing entries by mean values. However, deleting records with missing values and some other filtering strategies can introduce biases  that can impact modeling in many ways, thus limiting its generalizability. Mean imputation is widely used by researchers to handle missing values. However, it is shown to yield less effective estimates than many other modern imputation techniques [5, 58, 1, 55], such as maximum likelihood approaches and multiple imputation methods (e.g. multiple imputation by chained equations (MICE) ). The maximum likelihood and multiple imputation methods are based on solid statistical foundations and become standard in the last few decades [17, 41].
Multiple imputation is originally proposed by Rubin  and with the idea of replacing each missing value with multiple estimates that reflect variation within each imputation model. The classic multiple imputation method treats the complete variables as predictors and incomplete variables as outcomes, and uses multivariate regression models to estimate missing values [39, 43]. Another imputation approach relies on chained equations where a sequence of regression models is fitted by treating one incomplete variable as the outcome and other variables with possible best estimates for missing values as predictors [37, 54, 53]. Multiple imputation has its success in healthcare applications, where many imputation methods designed for various tasks are based on multiple imputation [18, 36, 47, 22, 31, 52, 10].
In recent years, additional imputation methods are proposed. Although many imputation methods [55, 6, 46, 37, 27, 32, 64, 51] show their effectiveness in many applications, few of them are designed for time series-based clinical data. These clinical data are usually multivariable time series, where patients have measurements of multiple laboratory tests at different times. Many methods are designed for cross-sectional imputation (measurements taken at the same time point) and do not consider temporal information that is useful in making predictions or imputing missing values. Ignoring informative temporal correlations and only capturing cross-sectional information may yield less effective imputation.
In order to address the limitations mentioned above, we present mixture-based multiple imputation models for clinical time series. Our models capture both cross-sectional information and temporal correlations to estimate missing values using mixture models. We model the distribution of measurements using a mixture model. The mixture is composed of linear regression to model cross-sectional correlations and Gaussian processes (GPs) to capture temporal correlations. The problem of integrating GP within a standard mixture model is that GP models in all patient cases get the same mixing weights, while the confidence of predictions by GP models can vary largely across different patient cases. We overcome this problem by introducing individualized mixing weights for each patient cases, instead of assigning a fixed weight. We train our models using the Expectation-Maximization (EM) algorithm. We demonstrate the effectiveness of our models by comparing them with several state-of-the-art imputation algorithms on multiple clinical datasets.
Our main contributions are summarized as follows.
1. To the best of our knowledge, we are the first to build imputation models for time series by integrating GP within mixture models. We address the problem that all GP models in all patient cases get a fixed mixing weight by introducing individualized mixing weights.
2. We test the performance of our models on two real-world clinical datasets and several synthetic datasets. Our best model outperforms all comparison models including several state-of-the-art imputation models. Using synthetic datasets, we also explore and discover the properties of the data that benefit our models and/or comparison models. Experiments show that our best model is robust to the variation of these properties and outperforms comparison models on all synthetic datasets.
The remainder of this paper is structured as follows. Section II discusses related work while in Section III, the proposed method is described. The experimental setup, including dataset collection and evaluation procedure, is described in Section IV. Section V discusses the computational results and underlying analyses. The conclusions are drawn in Section VI.
II Related work
Research in designing imputation methods for multivariable time series attracts growing interest in recent decades. Previous studies generally fall into two categories. One comes from methods using linear or other simple parametric functions to estimate missing values. The other is the methods treating time series as smooth curves and estimating missing values using GP or other nonparametric methods.
In the first category, multivariable time series are modeled based on either linear models , linear mixed models [29, 42] or autoregressive models [20, 3]. However, in these methods, the potential trajectories of variables are only limited to linear or other simple parametric functions. Alternatively, many authors choose GPs or other nonparametric functions to model time series. Compared to linear models, GPs only have locality constraints in which close time points in a time series usually have close values. Therefore, GPs bring in more flexibility in capturing temporal trajectories of variables.
The straightforward way of applying GPs to the imputation on multivariable time series is to fit a single GP model on each time series and then make predictions for missing entries separately. However, without taking into account similarities and correlations across multiple time series, only fitting a single GP model on each time series may yield less effective imputation. Many researchers attempt to extend GP-based methods to multivariable settings. Futoma et al.  apply the multi-task Gaussian Processes (MTGP) prediction , a multiple-outcome modeling approach in the context of GPs, to impute missing values in longitudinal data. However, the quality of estimating missing values relies on the estimation of covariance structure among variables when using MTGP or other multi-task functional approaches [19, 8, 25]. To make a confident estimation of the covariance, a large amount of time points with shared observations of these variables are often required by these multi-task approaches [4, 62]. Due to the fact that many patients only have records with a limited number of time points, time series of inpatient clinical laboratory tests fall short of such a requirement. Hori et al.  extend MTGP to imputation on multi-environmental trial data in a third-order tensor. However, this approach is not applicable to clinical data with a large number of patients, since the covariance matrix of observed values needs to be computed together with its inverse, which is intractable for our datasets.
Recently, Luo et al.  explore the application of GPs in clinical data and propose an algorithm, 3-dimensional multiple imputation with chained equations (3D-MICE), that combines the imputation from GP and traditional MICE based on weighting equations. However, the weighting equations are calculated only based on the standard deviations of GP and MICE predictions for missing values. The weighting strategy is static and not optimized. We postulate that calculating weights through an optimization problem can help to improve the imputation quality. In our work, instead of the predictive mean matching used in , we choose linear regression as one component of our model. Our model is also based on a statistical model and thus statistically justified which is not the case for .
The mixture model proposed in this work differs from the mixtures of Gaussian processes (MGP)  in several ways. The mixing components in our model are not limited to Gaussian process, thus bringing in possibilities of capturing correlations with various choices of parametric functions. More importantly, our model is designed for multivariable time series and captures both temporal and cross-sectional correlations. The latter is ignored in  when MGP is applied to multivariable time series, which yields less effective predictions since some clinical tests have strong cross-sectional relations. Without a Gaussian process component, the standard Gaussian mixture model (GMM) is well studied and has been extended to imputation tasks [11, 9, 60, 45]. These approaches still suffer from losing informative temporal correlations when imputing missing values in longitudinal data.
In order to effectively model the interaction between different aspects, we represent the data as a tensor with each aspect being one mode. For that reason, our method is also considered as a tensor completion approach. Tensor completion problems are extensively studied. However, the classic tensor completion methods [12, 30, 49] focus on general tensors and usually do not consider temporal aspects. In recent years, many studies explore the application of temporal augmented tensor completion on imputing missing values in time series [2, 63, 15, 48, 7]. These methods discretize time into evenly sampled intervals. However, due to the fact that inpatient clinical laboratory tests are usually measured at varying intervals, assembling clinical data over regularly sampled time periods might have several drawbacks, such as leading to sparse tensors if discretizing time at fine granularity (e.g. every minute) while some laboratory tests are measured less frequently (e.g. daily). Furthermore, extending these methods to the case, where time is not regularly sampled, is not easy and straightforward, requiring changing design details and the objective functions to be optimized. Recently, Yang et al.  propose a tensor completion method that can deal with irregularly sampled time. They extend the PACIFIER imputation framework  and propose a time-aware matrix decomposition method to estimate missing values in predicting septic shock. However, most components of this approach are tailored to the characteristics of septic patients. In this work, we implement the imputation approach proposed in  with only the time-aware mechanism, which is general and applicable to our experimental settings. However, this approach is not so effective in our experiments and thus it is not included as a benchmark in this paper. Lately, Zhe et al.  propose a Bayesian nonparametric tensor decomposition model that captures temporal correlations between interacting events. However, this approach is not directly applicable to continuous multivariable time series because it focuses on discrete events and captures temporal correlations between the occurrences of events.
III-A Imputation Framework
In many predictive tasks on temporal clinical data, time series are often aligned into the same-length sequences to derive more robust patient phenotypes through matrix decomposition or discover feature groups by applying sequence/graph mining techniques [61, 59]. We model under this assumption. In this work, we use tensor representation, in which patients have the same number of time points. We represent the data collected from patients with laboratory tests and time points as two 3D tensors and , shown in Fig. 1. Each laboratory test measurement is stored in the measurement tensor . Each time , when is measured, is stored in the time tensor .
Table I lists the main symbols we use throughout the paper. Missing values in the measurement tensor are denoted as , showing that the value of test at time index for patient is missing. Correspondingly, denotes an observed value. The time tensor is complete, since we only collect patient records at the time when at least one laboratory test measurement is available. We assume we know the “prescribed” time a missing measurement should have been taken. In the matrix at time index , the measurement time and can be different when , whereas for a given patient , we have for . That is, all tests for a particular patient are taken at the same time.
Disregarding the temporal dimension, the imputation problem is well studied. If one dimension is time, in order to apply imputation methods that are not designed for time series, we need to disregard the temporal aspect or ignore temporal correlations of the data. However, temporal trajectories can reveal patient’s underlying pathophysiological evolution, the modeling of which can help to better estimate missing values. For the reason that both cross-sectional information and temporal aspects can impact the estimation of missing measurements, we explore mixture models, which are composed of several base models through either a cross-sectional or temporal view. We introduce these base models in Section III-B.
|,||Measurement and time tensor|
|Measurements in fiber excluding|
|Times in fiber excluding|
|Mixture model for and .|
|Concatenation of and|
The th prior multivariate normal distribution in
|,||Predictive mean and variance of a GP model|
The set of all trainable parameters of
|The identity matrix|
In our imputation framework, a mixture model is trained for each variable and time index. We use to denote the mixture model to impute missing values of variable at time index . The missing values in the fiber are imputed by the optimized . In each iteration of the algorithm, it is assumed that all other values are known and only is imputed. This is wrapped in an outer loop. We call this procedure simple-pass tensor imputation, i.e., one pass through all . Since several simple-pass tensor imputations are conducted, our approach is also considered as an iterative imputation , which can also be regarded as a sampling-based approach where a Gibbs sampler is used to approximate convergence. The convergence of iterative imputation methods can be quite fast with a few iterations .
In detail, the iterative imputation approaches start by replacing all missing data with values from simple guesses; we fill in all missing values with initial estimates by taking random draws from observed values. This procedure is called an initial imputation. Then we perform iterative tensor imputation on each copy separately. The training procedure and imputation for fibers are introduced in Section III-C and III-D. We also rely on the concept of multiple imputations, where several iterative imputations are performed and the imputed values are averaged at the end. Each iterative imputation starts with a different iterative imputation tensor and/or uses a different order of ,.
In summary, the algorithm creates different copies of , each one filled with different random . For each , we then perform simple-pass tensor imputations. Each simple-pass has a loop over all ,, which uses to adjust . At the end of the imputations, are averaged across all to yield the final imputed tensor.
The whole imputation process involves imputation models. We next first focus on the base models behind and then on the actual mixture model.
III-B Base Models
Our mixture models are composed of three components that are derived from two base models, linear regression and Gaussian processes. One component consists of GP models and the other two components are linear models through two different views of the measurement tensor. Through a cross-sectional view, the tensor can be considered as a vector of patient-by-variable matrices at different time indices. Through a temporal view, we can view the tensor as a vector of patient-by-time matrices for different variables.
III-B1 Linear model through cross-sectional view
We can view the measurement tensor as a vector of patient-by-variable matrices. On the slice , we use a linear regression model to fit the target variable as a function of the other variables except . The target values are modeled as
where is the column vector of coefficients and is the standard deviation of the error , regarding the regression model through cross-sectional view for variable and time index .
The likelihood distribution of is then given by
The training data consists of observed target values and the input data , where is the training patient set and includes only if is observed.
III-B2 Linear model through temporal view
In addition to the cross-sectional view, we can also view the measurement tensor as a vector of patient-by-time matrices. On matrix , we use linear regression to model the measurements at time index against those at other indices.
The target values are modeled as
where is the column vector of coefficients and is the standard deviation of the error , regarding the linear regression model though temporal view for variable and time index .
The likelihood distribution of is given by
The training data consists of observed target values and input data .
III-B3 Gaussian processes through temporal view
Gaussian processes are commonly used to capture trajectories of variables, thus used in our mixture model to capture temporal correlations. Through the same temporal view as introduced above, on matrix , we fit GPs on time series for each patient.
The target value is modeled as
where is the overall mean of the model, is a Gaussian process with mean of and a covariance matrix of time pairs (,). Then the likelihood distribution of is written as
where are the kernel parameters of the GP models, and the predictive mean and variance are given by and ; see more details in Appendix B. For a certain and , all GP models share the same kernel parameters .
III-C The Mixture Model
Given the likelihood distribution of all three components, we model the joint mixture distribution, regarding the variable and time index , in the following way
where we define , , and is the mixing weight for the th component. This model can be interpreted as the joint distribution between observed data and missing values , which consists of a mixture of three distributions. The first one is modeled as
the remaining two follow the same logic.
By marginalizing over , the prior probability distribution is written as
which is a mixture of Gaussians. It also follows that
where we define .
We train our mixture model on observed target values by maximizing the log likelihood of the joint mixture distribution
where is the training input data and defined as the concatenation of and , and is the set of all trainable parameters
After training, the missing values are imputed using individualized (per-patient) mixing weights that are derived from the conditional distribution. The missing value is imputed by
where the individualized mixing weight of the th component for patient , variable and time index is defined as
where is the observed data and defined as the concatenation of and .
III-D Mixture Parameter Estimation
Let be the log likelihood . Explicitly maximizing is hard. Instead, we use the EM algorithm to repeatedly construct a lower-bound on and then optimize that lower-bound . We first define a latent indicator variable that specifies which mixing component that data points come from. Then we use Jensen’s inequality to get the lower-bound , which is given by
In (12) and (13), the marginal distribution over is specified by the mixing coefficients . We can view as the responsibility that component of the mixture model takes to “explain” . We use the standard EM algorithm to maximize the lower-bound ; see more details about the estimation of the parameters in Appendix A.
|Real-world MIMIC (=0)||0.13072||0.12599||0.11741||0.11763||0.11186||0.09304||0.09285|
|Synthetic MIMIC (=0.5)||0.10466||-||0.11455||0.11573||0.09087||0.08612||0.08448|
|Synthetic MIMIC (=1)||0.09220||-||0.11436||0.11561||0.07715||0.08427||0.07538|
|MTGP cannot have multiple inputs, thus not applicable to the synthetic datasets.|
|Blood urea nitrogen||0.12358||0.12720||0.15160||0.15189||0.11814||0.09479||0.09410|
III-E Special Cases and an Ensemble Model
Our imputation model provides flexibility in changing base models. The mixture model mentioned above consists of three base models. Two of them are applied through the same temporal view of the measurement tensor. We can drop one of the models through temporal view to yield a new mixture model. For example, if the GP model through temporal view is removed from the mixture model for variable and time point , the new mixture model is now a mixture of two linear models.
Each mixture model can be a mixture of all three base models (the Linear-Linear-GP [LLG] model), a mixture of two linear models (the Linear-Linear [LL] model) or a mixture of linear and GP models. All these mixtures may have an overfitting problem due to the large amount of trainable parameters. To further improve the imputation quality, we build an ensemble model (En-LLG) to reduce variance by allowing each mixture model to be a mixture of either two linear models or all three base models.
In En-LLG, for each variable and time index, we train the mixture model with two linear models and the mixture model with all three base models, and then select the one with less training error as the final mixture model, which is used to do imputation. Although we train our mixture models by maximizing the likelihood, we use the absolute training error as the selection criteria because the likelihood of the mixture model with two linear models is not in the same scale as the mixture model with three base models.
IV Experimental Setup
IV-A Real-world Datasets
We collect two real-world datasets from the Medical Information Mart for Intensive Care (MIMIC-III) database  and the Northwestern Medicine Enterprise Data Warehouse (NMEDW). Each dataset contains inpatient test results from 13 laboratory tests. These tests are quantitative and frequently measured on hospital inpatients. They are the same as those used in  in their imputation study. We organize the data by unique admissions. We distinguish multiple admissions of the same patient. Each admission consists of time series of the 13 laboratory tests. Although our models are evaluated only on continuous variables, they are also applicable on categorical variables with a simple extension (e.g. through predictive mean matching) .
In both MIMIC-III and NMEDW datasets, the length of time series varies across admissions. To apply our imputation models on these datasets, we truncate time series so that they have the same length. The length is the average number of time points across all admissions. Before truncating, the average number of time points in MIMIC-III dataset is 11. We first exclude admissions that have less than 11 time points, and then we truncate time series by removing measurements taken after the 11-th time point. We also exclude admissions that contain time series that have no observed values. Our MIMIC-III dataset includes 26,154 unique admissions and the missing rate is about 28.71%. The same data collection procedure is applied on the NMEDW dataset (approved by Northwestern IRB) where we end up with 13,892 unique admissions that have 7 time points. The missing rate of the NMEDW dataset is 24.22%.
IV-B Synthetic Datasets
We create synthetic datasets to explore and discover the properties of the data that might benefit our models and/or comparison models. In synthetic datasets, we augment the correlation between measurements and times. We do not augment correlations by imposing strong constraints on time series where closer measurements have closer values. Instead, we generate synthetic times by altering real times so that the constraints in synthetic data are “slightly” stronger than real-world data. We also introduce a scaling factor to control the strength of the constraints in synthetic data.
The synthetic datasets are generated based on the real-world MIMIC-III dataset. We move two consecutive times of a time series closer, if the relative difference in two consecutive measurements is smaller than the relative difference in two consecutive times. The relative differences and of a time series are given by
The scaling factor controls how much we move times. If , we do not move times. In other words, the synthetic dataset at is the same as the real-world MIMIC-III dataset. As increases, stronger constraints are introduced to synthetic data. The synthetic time for a time series is generated as follows:
If a time series has missing values, we first calculate the synthetic times for the observed measurements. Then we perform a linear interpolation between real times and synthetic times for observed measurements to generate synthetic times for missing measurements.
IV-C Evaluation of Imputation Quality
We randomly mask 20% observed measurements in a data set as missing and treat the masked values as the test set. The remaining observed values are used in training. We impute originally missing and masked values together, and compare the imputed values with the ground truth for masked data to evaluate imputation performance.
We use Mean Absolute Scaled Error (MASE)  to measure the quality of imputation on the test set. MASE is a scale-free measure of the accuracy of predictions and recommended by Hyndman et al [23, 13] to measure the accuracy of predictions for series. In this work, we calculate MASE for all tests (variables) and take a weighted average, according to the number of masked values of a variable, to get an overall MASE per dataset.
Let be the set of cardinality of all time indices that have been masked for patient and variable . Also let be the sequence of length of all observed values for patient and variable , and let represent the imputed value. The MASE for variable is defined as
To show the effectiveness of our imputation models, we compare the MASE scores of our models (LL and En-LLG) with other five imputation methods: (a) MICE  with 100 imputations, where the average of all imputations are used for evaluation; (b) the pure Gaussian processes, where a GP model is fitted to the observed data of each time series using GPfit  in R and missing values are replaced with the predictions from the fitted GP models; (c) 3D-MICE, a state-of-the-art imputation model  for which we obtain their code and adapt it to account for our use of the tensor representation; (d) multi-task Gaussian process prediction (MTGP)  and (e) a Gaussian mixture model for imputation (GMM) [11, 35]. To tune hyperparameters if any in these models, we mask out 20% observed measurements in the training set as a validation set and tune hyperparameters on the validation set.
We introduce and as hyperparameters of our models. In our experiments, all mixture models start with the same mixing weights, i.e., for . Other trainable parameters are initialized directly from the data after initial imputation and do not require a manual specification.
We run our En-LLG model for multiple imputations with iterations and run the LL model with more multiple imputations (M=5) and more iterations (K=5). We run more iterations of the LL model, because the LL model takes less time than the En-LLG model. For 3D-MICE, we set the number of multiple imputation to 40, instead of 100 that is suggested in , to balance the performance and running time.
V-A Performance Comparison
Table II and Fig. 2 compare all the imputation models on 4 datasets using MASE. Table II shows the overall MASE score of each imputation model on all datasets. Fig. 2 provides a comparison for all imputation models in the MASE score over 3D-MICE by showing the percentage deviation against 3D-MICE. We select 3D-MICE since it is a state-of-the-art benchmark model. We observe that our En-LLG model outperforms all comparison models on all 4 datasets. The LL model is outperformed only by 3D-MICE on the synthetic MIMIC (=1) dataset. The En-LLG model is significantly better than the second best model (=.001, permutation test with 1000 replicates) on all 4 datasets.
Table III shows a variable-wise comparison of the imputation models on the real-world MIMIC (=0) dataset. Our two imputation models outperform all comparison models on all variables. The En-LLG model is better than the LL model on most variables. All models except GP and MTGP achieve a much lower error on Hematocrit and Hemoglobin than on other variables. The reason is that these two variables are highly correlated. Those methods that capture the correlation between variables can reasonably infer missing values for Hematocrit from observed measurements of Hemoglobin, and vice versa. Compared to MICE, our models achieve even lower errors on these two variables, which indicates that temporal correlations captured by our models help to make better estimation of missing values, even when there is a more dominant cross-sectional correlation.
As shown in Table II, all models except MTGP benefit from the increment of , the scaling factor when generating synthetic data. The reason is that all models take into account temporal aspects and the measurements in the synthetic time series have stronger temporal correlations as increases. The reason that MICE and GMM also benefit from the temporal correlations is that we include time as a feature in addition to variable features. We also try to exclude times, however, experiments show that these two models perform better when times are included. As shown in Table II, GP and 3D-MICE benefit the most as increases from to , MICE and GMM benefit the least and our models (LL and En-LLG) are in the middle. The LL model is outperformed by 3D-MICE when increases to . However, En-LLG shows its robustness to the variation of in our current experimental settings.
V-B Individualized Weights
By introducing individualized (per patient) mixing weights defined in (11), we improve the performance of our En-LLG model in the MASE score from 0.08351 to 0.07538, an improvement of 9.73% compared against the model where each mixture component has a fixed weight for all patient cases. The reason that individualized weights are better than fixed weights in our model might be that they better approximate the responsibilities.
In training, we can optimize the responsibility a component should take to “explain” an observed target value for . These correspond to in (13). However, when making inference, we can not calculate the responsibility each component should take to “explain” missing values, because responsibilities depend on observed target values, according to (13). We have to use , the individualized mixing weights. In a standard mixture model, we could use , which is the average of responsibilities of the th component across all training patients, as a fixed weight that the th component should contribute to impute missing values for all test patients. However, patient time series can be very different and the confidence of predictions by the GP component can vary largely across different patient cases. In our mixture model, therefore, a fixed weight can not reflect such variation in prediction confidence.
We shall view an individualized mixing weight as an approximation of how much responsibility a component should take to impute the missing value for a particular patient case. It is tailored for each patient. As defined in (11), the individualized mixing weights only depend on the inputs, therefore, we can calculate them when making inferences on the test set.
In Fig. 3, we plot the distribution of the individualized weights of the GP component in the training set and compare it with the distribution of the optimized responsibility values . The responsibilities that the GP component should take can vary a lot in different patient cases, especially on the synthetic dataset, which implies that it is more reasonable for patients to get individualized mixing weights than a fixed weight. We also observe that the individualized mixing weights reasonably mimic the distribution of the optimized responsibilities on the training set. The improvement of our model on the test set attests that the individualized weights approximate the responsibilities better than fixed weights.
In addition, 3D-MICE also assigns individualized weights at the same level of granularity as our models, however, weights in 3D-MICE are calculated only based on the deviations of the cross-sectional imputation and the temporal imputation, and are not optimized. As shown in Table III, the improvement of our models over 3D-MICE implies that our models provide a more accurate weighting solution when combining the cross-sectional and temporal imputation.
V-C Time Complexity
We compare the running times of our proposed models and all comparison models on the real-world MIMIC dataset. All models ran on the same Linux server. The MTGP and GMM models are executed sequentially and other models run in parallel with 20 cores. The LL model (taking 4.2 hours), GP (1.1 hours), MTGP (1.2 hours) and GMM (2.2 hours) are the four fastest models; 3D-MICE (156.1 hours) is the slowest; MICE (77.5 hours) and the En-LLG model (109.5 hours) are in the middle.
The LL model is much faster than the En-LLG model because we can explicitly calculate the estimates of parameters for the mixture model with only linear components. However, it is hard to directly calculate the parameter estimates of the GP component. In En-LLG, we use the Adam optimizer to update the parameter estimates of the GP component in each EM iteration. In our experiments, the EM algorithm and the Adam optimization procedure converge in a few iterations. As shown in Fig. 3(a), the increment speed of optimizing the log likelihood of the mixture model decreases dramatically after a few EM iterations. Fig. 3(b) shows that the log likelihood with respect to the GP component converges within 10 Adam iterations. We observe a similar convergence property in other mixture models.
We notice that our models can have an overfitting problem due to the large amount of trainable parameters. We observe that, in many mixture models, the log likelihood keeps increasing, however, the imputation error stops decreasing after a few EM iterations. We alleviate the overfitting problem by terminating the EM algorithm earlier. We terminate it when the mean absolute error on the training set stops decreasing. In our experiment, this strategy is better than a vanilla ridge regularization for handling the overfitting problem. In addition to the improvement of imputation quality, this strategy also helps to reduce the running time of our model, as the EM algorithm stops earlier with this strategy. Although we alleviate the overfitting problem, we expect further improvements in the performance of our imputation model with better strategies for handling overfitting.
We present and demonstrate mixture-based imputation models for multivariable clinical time series. Our models can capture both cross-sectional and temporal correlations in time series. We integrate Gaussian processes with mixture models and introduce individualized mixing weights to further improve imputation accuracy. We show that our best model can provide more accurate imputation than MICE, GMM, GP, MTGP and 3D-MICE, a state-of-the-art imputation model that integrates cross-sectional and longitudinal imputation. Although in this work our models are tested on inpatient clinical data, they can also be applied to other multivariable time series data in healthcare and other domains with necessary adaptation.
-  (1996) Full information estimation in the presence of incomplete data. In Advanced Structural Equation Modeling: Issues and Techniques, G. A. Marcoulides and R. E. Schumacker (Eds.), Vol. 243, pp. 277. Cited by: §I.
-  (2014) Fast multivariate spatio-temporal analysis via low rank tensor learning. In Advances in Neural Information Processing Systems, pp. 3491–3499. Cited by: §II.
-  (2017) Handling missing data in multivariate time series using a vector autoregressive model-imputation (var-im) algorithm. Neurocomputing. Cited by: §II.
-  (2008) Multi-task gaussian process prediction. In Advances in Neural Information Processing Systems, pp. 153–160. Cited by: §II, §IV-C.
-  (1994) Efficacy of the indirect approach for estimating structural equation models with missing data: a comparison of five methods. Structural Equation Modeling: A Multidisciplinary Journal 1 (4), pp. 287–316. Cited by: §I.
-  (2011) Mice: multivariate imputation by chained equations in r. Journal of Statistical Software 45 (3). Cited by: §I, §I, §III-A, §IV-C.
-  (2015) Facets: fast comprehensive mining of coevolving high-order time series. In Proceedings of the 21th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 79–88. Cited by: §II.
-  (2014) A functional data approach to missing value imputation and outlier detection for traffic flow data. Transportmetrica B: Transport Dynamics 2 (2), pp. 106–129. Cited by: §II.
-  (2012) Efficient em training of gaussian mixtures with missing data. arXiv preprint arXiv:1209.0521. Cited by: §II.
-  (2016) Multiple imputation for general missing data patterns in the presence of high-dimensional data. Scientific Reports 6, pp. 21689. Cited by: §I.
-  (2007) Imputation through finite gaussian mixture models. Computational Statistics & Data Analysis 51 (11), pp. 5305–5316. Cited by: §II, §IV-C.
-  (2015) Tucker factorization with missing data with application to low--rank tensor completion. Multidimensional Systems and Signal Processing 26 (3), pp. 677–692. Cited by: §II.
-  (2016) A note on the mean absolute scaled error. International Journal of Forecasting 32 (1), pp. 20–22. Cited by: §IV-C.
-  (2017) Learning to detect sepsis with a multitask gaussian process rnn classifier. In Proceedings of the 34th International Conference on Machine Learning-Volume 70, pp. 1174–1182. Cited by: §II.
-  (2016) Uncovering the spatio-temporal dynamics of memes in the presence of incomplete information. In Proceedings of the 25th ACM International on Conference on Information and Knowledge Management, pp. 1493–1502. Cited by: §II.
-  (2004) Parameterization and bayesian modeling. Journal of the American Statistical Association 99 (466), pp. 537–545. Cited by: §III-A.
-  (2009) Missing data analysis: making it work in the real world. Annual Review of Psychology 60, pp. 549–576. Cited by: §I.
-  (2007) Multiple imputation for the comparison of two screening tests in two-phase alzheimer studies. Statistics in Medicine 26 (11), pp. 2370–2388. Cited by: §I.
-  (2011) A functional multiple imputation approach to incomplete longitudinal data. Statistics in Medicine 30 (10), pp. 1137–1156. Cited by: §II.
-  (2012) MARSS: multivariate autoregressive state-space models for analyzing time-series data. R Journal 4 (1). Cited by: §II.
-  (2016) Multi-task gaussian process for imputing missing data in multi-trait and multi-environment trials. Theoretical and Applied Genetics 129 (11), pp. 2101–2115. Cited by: §II.
-  (2006) Survival analysis using auxiliary variables via non-parametric multiple imputation. Statistics in Medicine 25 (20), pp. 3503–3517. Cited by: §I.
-  (2006) Another look at measures of forecast accuracy. International Journal of Forecasting 22 (4), pp. 679–688. Cited by: §IV-C.
-  (2016) MIMIC-iii, a freely accessible critical care database. Scientific Data 3, pp. 160035. Cited by: §IV-A.
-  (2014) A bayesian approach to functional mixed-effects modeling for longitudinal data with binomial outcomes. Statistics in Medicine 33 (18), pp. 3130–3146. Cited by: §II.
-  (2015) Ten things we have to do to achieve precision medicine. Science 349 (6243), pp. 37–38. Cited by: §I.
-  (2004) Robust likelihood-based analysis of multivariate data with missing values. Statistica Sinica, pp. 949–968. Cited by: §I.
-  (1988) Missing-data adjustments in large surveys. Journal of Business & Economic Statistics 6 (3), pp. 287–296. Cited by: §IV-A.
-  (2000) Multiple imputation and posterior simulation for multivariate missing data in longitudinal studies. Biometrics 56 (4), pp. 1157–1163. Cited by: §II.
-  (2015) Trace norm regularized candecomp/parafac decomposition with missing data. IEEE Transactions on Cybernetics 45 (11), pp. 2437–2448. Cited by: §II.
-  (2012) Doubly robust nonparametric multiple imputation for ignorable missing data. Statistica Sinica 22, pp. 149. Cited by: §I.
-  (2016) Using machine learning to predict laboratory test results. American Journal of Clinical Pathology 145 (6), pp. 778–788. Cited by: §I.
-  (2017) 3D-mice: integration of cross-sectional and longitudinal imputation for multi-analyte longitudinal clinical data. Journal of the American Medical Informatics Association 25 (6), pp. 645–653. Cited by: §II, §IV-A, §IV-C, §IV-C.
-  (2015) GPfit: an r package for fitting a gaussian process model to deterministic simulator outputs. Journal of Statistical Software 64 (1), pp. 1–23. Cited by: §IV-C.
-  (2012) Machine learning: a probabilistic perspective. MIT press. Cited by: §IV-C.
-  (2010) A comparison of multiple imputation and fully augmented weighted estimators for cox regression with missing covariates. Statistics in Medicine 29 (25), pp. 2592–2604. Cited by: §I.
-  (2001) A multivariate technique for multiply imputing missing values using a sequence of regression models. Survey Methodology 27 (1), pp. 85–96. Cited by: §I, §I.
-  (1978) Multiple imputations in sample surveys-a phenomenological bayesian approach to nonresponse. In Proceedings of the survey research methods section of the American Statistical Association, Vol. 1, pp. 20–34. Cited by: §I.
-  (1987) Multiple imputation for nonresponse in surveys. John Wiley & Sons. Cited by: §I.
-  (1989) Design and analysis of computer experiments. Statistical Science, pp. 409–423. Cited by: Appendix B.
-  (2002) Missing data: our view of the state of the art.. Psychological Methods 7 (2), pp. 147. Cited by: §I.
-  (2002) Computational strategies for multivariate linear mixed-effects models with missing values. Journal of Computational and Graphical Statistics 11 (2), pp. 437–457. Cited by: §II.
-  (1997) Analysis of incomplete multivariate data. Chapman and Hall/CRC. Cited by: §I.
-  (1997) A random-effects model for multiple characteristics with possibly missing data. Journal of the American Statistical Association 92 (438), pp. 775–779. Cited by: §II.
-  (2018) Multivariate data imputation using gaussian mixture models. Spatial statistics 27, pp. 74–90. Cited by: §II.
-  (2011) MissForest—non-parametric missing value imputation for mixed-type data. Bioinformatics 28 (1), pp. 112–118. Cited by: §I.
-  (2011) Multiple imputation with diagnostics (mi) in r: opening windows into the black box. Journal of Statistical Software 45 (2), pp. 1–31. Cited by: §I.
-  (2017) Autoregressive tensor factorization for spatio-temporal predictions. IEEE International Conference on Data Mining. Cited by: §II.
-  (2010)(Website) External Links: Cited by: §II.
-  (2001) Mixtures of gaussian processes. In Advances in neural information processing systems, pp. 654–660. Cited by: §II.
-  (2001) Missing value estimation methods for dna microarrays. Bioinformatics 17 (6), pp. 520–525. Cited by: §I.
-  (1999) Multiple imputation of missing blood pressure covariates in survival analysis. Statistics in Medicine 18 (6), pp. 681–694. Cited by: §I.
-  (2006) Fully conditional specification in multivariate imputation. Journal of statistical computation and simulation 76 (12), pp. 1049–1064. Cited by: §I.
-  (2018) Flexible imputation of missing data. Chapman and Hall/CRC. Cited by: §I.
-  (2013) Comparison of imputation methods for missing laboratory data in medicine. BMJ Open 3 (8), pp. e002847. Cited by: §I, §I.
-  (2017) Biases introduced by filtering electronic health records for patients with “complete data”. Journal of the American Medical Informatics Association 24 (6), pp. 1134–1141. Cited by: §I.
-  (2012) Computational medicine: translating models to clinical care. Science Translational Medicine 4 (158), pp. 158rv11–158rv11. Cited by: §I.
-  (2000) Longitudinal and multi-group modeling with missing data. In Modeling longitudinal and multiplegroup data: Practical issues, applied approaches, and specific examples, T. D. Little, K. U. Schnabel, and J. Baumert (Eds.), pp. 219–240. Cited by: §I.
-  (2018) Predicting icu readmission using grouped physiological and medication trends. Artificial Intelligence in Medicine. Cited by: §III-A.
-  (2015) Missing value imputation based on gaussian mixture model for the internet of things. Mathematical Problems in Engineering 2015. Cited by: §II.
-  (2018) Time-aware subgroup matrix decomposition: imputing missing data using forecasting events. In 2018 IEEE International Conference on Big Data, pp. 1524–1533. Cited by: §II, §III-A.
-  (2005) Learning gaussian processes from multiple tasks. In Proceedings of the 22nd International Conference on Machine Learning, pp. 1012–1019. Cited by: §II.
-  (2015) Accelerated online low-rank tensor learning for multivariate spatio-temporal streams. In Proceedings of the 32nd International Conference on Machine Learning (ICML-15), pp. 238–247. Cited by: §II.
-  (2009) Extensions of the penalized spline of propensity prediction method of imputation. Biometrics 65 (3), pp. 911–918. Cited by: §I.
-  (2018) Stochastic nonparametric event-tensor decomposition. In Advances in Neural Information Processing Systems, pp. 6855–6865. Cited by: §II.
-  (2014) From micro to macro: data driven phenotyping by densification of longitudinal electronic medical records. In Proceedings of the 20th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 135–144. Cited by: §II.
Appendix A Parameter Estimation in EM
In the E (Expectation) step, we calculate the responsibilities for using the current values of the parameters in iteration :
Let and . In the M (Maximization) step, we re-estimate the parameters in iteration using the th responsibilities:
where is the vector of for in iteration . The kernel parameters of GP models are evaluated by function , a gradient descent method that calculates the estimates of to maximize , using as the starting point. The first order derivatives of with respect to that are used in are given in Appendix C.
Appendix B GP Model
We assume the GP model discussed here in a mixture model for a certain variable and time, and thus we exclude the subscripts and . We use to denote a measurement of the time series at time for patient of a certain variable. We use to denote a time series without the measurement at time . The GP model is given by
where is the overall mean of the model and is a Gaussian process with mean of and covariance of . Following the maximum likelihood approach, the best linear unbiased predictor (BLUP)  at and the mean squared error are
where , is the vector of for all possible , is a vector of time except for time , is the correlation matrix and the correlation function is given by
The estimator is given by
where is a vector with length of all ones.
Appendix C Partial Derivatives in GP
To simplify the notations, we assume that the likelihood function under consideration is for a mixture model for a certain variable and time. The partial derivative with respect to Gaussian process parameters is
Letting and , we have
Then and are given by
where , , and are given as follows: