CVTA: Cross-Variable Temporal Attention for Multivariate Irregular Time Series Prediction
Organizations: Department of Computer Science, University of Pittsburgh, Pittsburgh, PA 15260
Abstract
Irregularly sampled multivariate time series arise in many real-world domains such as healthcare, climate science, and large-scale monitoring systems. Characterized by asynchronous observations and heterogeneity across variables, such time series pose significant challenges for forecasting. In this work, we introduce Cross-Variable Temporal Attention (CVTA), a short-term, continuous-time forecasting framework that learns prediction functions from its compact, interpretable state summarizing recent temporal dynamics across variables. CVTA uses Variable-Conditioned Embeddings to project heterogeneous variables into variable-specific latent spaces, and applies temporal attention to capture cross-variable dependencies from this state. Furthermore, we demonstrate that CVTA can be integrated as an auxiliary module into existing state-of-the-art prediction models through a dynamic meta-decision model. Extensive experiments on multiple real-world benchmarks show that CVTA achieves strong short-term predictive accuracy, supports efficient online inference, improves existing prediction models when used as an auxiliary module, and learns attention patterns that align well with domain knowledge.
Figures & tables
| Regular Time Series | Irregular Time Series | Event Time Series | |
| Marking | Continuous | Continuous | Discrete |
| Sampling | Uniform | Non-Uniform | Non-Uniform |
| USHCN | Physionet | MIMIC-III | MIMIC-IV | MIMIC-HiRes | |
| Missingness type | MCAR | NMAR | NMAR | NMAR | NMAR |
| # Vars. | 5 | 37 | 96 | 100 | 393 |
| Mean len | 730.0 | 72.16 | 30.23 | 154.34 | 244.30 |
| Max len | 730 | 185 | 96 | 920 | 4299 |
| Interarrival Variability | 0.894 | 0.609 | 1.385 | 1.609 | 1.015 |
| Missingness (%) | 22.36 | 83.85 | 95.59 | 97.83 | 98.74 |
| Model | USHCN | Physionet | MIMIC-III | MIMIC-IV | MIMIC-HiRes | |||||
| MSE | MAE | MSE | MAE | MSE | MAE | MSE | MAE | MSE | MAE | |
| CRU | 0.030\scalebox{0.75}{\pm 0.019} | 0.519\scalebox{0.75}{\pm 0.166} | 0.807\scalebox{0.75}{\pm 0.035} | 5.233\scalebox{0.75}{\pm 0.242} | 6.913\scalebox{0.75}{\pm \text{0.040}} | 5.797\scalebox{0.75}{\pm \text{0.103}} | 4.013\scalebox{0.75}{\pm \text{0.022}} | 4.128\scalebox{0.75}{\pm \text{0.141}} | 1.236\scalebox{0.75}{\pm 0.035} | 6.735\scalebox{0.75}{\pm 0.139} |
| f-CRU | 0.020\scalebox{0.75}{\pm 0.007} | 0.455\scalebox{0.75}{\pm 0.077} | 1.095\scalebox{0.75}{\pm 0.069} | 5.295\scalebox{0.75}{\pm 0.132} | 7.021\scalebox{0.75}{\pm \text{0.030}} | 6.134\scalebox{0.75}{\pm \text{0.052}} | 4.142\scalebox{0.75}{\pm \text{0.021}} | 4.290\scalebox{0.75}{\pm \text{0.076}} | 1.191\scalebox{0.75}{\pm 0.043} | 6.535\scalebox{0.75}{\pm 0.214} |
| RKN- | 0.015\scalebox{0.75}{\pm 0.016} | 0.367\scalebox{0.75}{\pm 0.178} | 0.680\scalebox{0.75}{\pm 0.042} | 4.854\scalebox{0.75}{\pm 0.197} | 7.784\scalebox{0.75}{\pm 0.086} | 6.102\scalebox{0.75}{\pm 0.038} | 5.707\scalebox{0.75}{\pm \text{0.031}} | 5.961\scalebox{0.75}{\pm \text{0.029}} | 1.292\scalebox{0.75}{\pm 0.042} | 6.820\scalebox{0.75}{\pm 0.081} |
| GRU- | 0.035\scalebox{0.75}{\pm 0.001} | 0.717\scalebox{0.75}{\pm 0.018} | 0.449\scalebox{0.75}{\pm 0.018} | 4.160\scalebox{0.75}{\pm 0.154} | 5.731\scalebox{0.75}{\pm 0.009} | 4.946\scalebox{0.75}{\pm 0.021} | 4.359\scalebox{0.75}{\pm \text{0.005}} | 4.729\scalebox{0.75}{\pm \text{0.008}} | 1.414\scalebox{0.75}{\pm 0.051} | 7.226\scalebox{0.75}{\pm 0.086} |
| GRU-D | 0.015\scalebox{0.75}{\pm 0.009} | 0.386\scalebox{0.75}{\pm 0.110} | 0.672\scalebox{0.75}{\pm 0.026} | 5.459\scalebox{0.75}{\pm 0.094} | 5.906\scalebox{0.75}{\pm \text{0.028}} | 5.362\scalebox{0.75}{\pm \text{0.046}} | 4.741\scalebox{0.75}{\pm \text{0.011}} | 4.867\scalebox{0.75}{\pm \text{0.038}} | 0.984\scalebox{0.75}{\pm 0.028} | 5.853\scalebox{0.75}{\pm 0.102} |
| Model | USHCN | % | Physionet | % | MIMIC-III | % | MIMIC-IV | % | MIMIC-HiRes | % | Avg. % |
| CVTA | 0.003 0.002 | – | 0.226 0.009 | – | 3.957 0.009 | – | 2.026 0.006 | – | 0.513 0.018 | – | – |
| w/o VCE | 0.013\scalebox{0.75}{\pm 0.003} | 333.3 | 0.494\scalebox{0.75}{\pm 0.008} | 118.6 | 5.236\scalebox{0.75}{\pm 0.008} | 32.3 | 2.932\scalebox{0.75}{\pm 0.006} | 44.8 | 1.016\scalebox{0.75}{\pm 0.036} | 98.1 | 125.4 |
| w/o local slope | 0.006\scalebox{0.75}{\pm 0.004} | 100.0 | 0.241\scalebox{0.75}{\pm 0.009} | 6.6 | 4.005\scalebox{0.75}{\pm 0.009} | 1.2 | 2.129\scalebox{0.75}{\pm 0.009} | 5.1 | 0.903\scalebox{0.75}{\pm 0.018} | 76.0 | 37.8 |
| w/o key mask | 0.004\scalebox{0.75}{\pm 0.003} | 33.3 | 0.227\scalebox{0.75}{\pm 0.010} | 0.4 | 3.960\scalebox{0.75}{\pm 0.009} | 0.1 | 2.080\scalebox{0.75}{\pm 0.006} | 2.7 | 0.809\scalebox{0.75}{\pm 0.016} | 57.7 | 18.8 |
| w/o value mask | 0.010\scalebox{0.75}{\pm 0.001} | 233.3 | 0.237\scalebox{0.75}{\pm 0.007} | 4.9 | 4.110\scalebox{0.75}{\pm 0.007} | 3.9 | 2.117\scalebox{0.75}{\pm 0.006} | 4.5 | 0.832\scalebox{0.75}{\pm 0.018} | 62.2 | 61.8 |
| w/o elapsed time | 0.004\scalebox{0.75}{\pm 0.003} | 33.3 | 0.241\scalebox{0.75}{\pm 0.004} | 6.6 | 4.112\scalebox{0.75}{\pm 0.008} | 3.9 | 2.168\scalebox{0.75}{\pm 0.005} | 7.0 | 0.869\scalebox{0.75}{\pm 0.019} | 69.4 | 24.0 |
| Method | USHCN | Physionet | MIMIC-III | MIMIC-IV | MIMIC-HiRes | |||||||||||||||
| H=1 | H=3 | H=5 | H=7 | H=1 | H=3 | H=5 | H=7 | H=1 | H=3 | H=5 | H=7 | H=1 | H=3 | H=5 | H=7 | H=1 | H=3 | H=5 | H=7 | |
| CVTA | 0.003 | 0.010 | 0.013 | 0.015 | 0.226 | 0.245 | 0.254 | 0.274 | 3.957 | 4.155 | 4.282 | 4.435 | 2.026 | 2.259 | 2.382 | 2.452 | 0.513 | 0.832 | 0.913 | 1.041 |
| GRUwE (PM) | 0.011 | 0.033 | 0.118 | 0.184 | 0.234 | 0.250 | 0.262 | 0.279 | 4.010 | 4.113 | 4.216 | 4.322 | 2.110 | 2.287 | 2.376 | 2.461 | 0.809 | 1.224 | 1.326 | 1.329 |
| GRUwE+CVTA | 0.004 | 0.017 | 0.024 | 0.028 | 0.211 | 0.230 | 0.239 | 0.258 | 3.783 | 3.925 | 4.039 | 4.151 | 2.015 | 2.191 | 2.285 | 2.334 | 0.604 | 0.931 | 0.959 | 0.989 |
| Improv. ( %) | +63.6 | +48.5 | +79.7 | +84.8 | +9.8 | +8.0 | +8.8 | +7.5 | +5.7 | +4.6 | +4.2 | +4.0 | +4.5 | +4.2 | +3.8 | +5.2 | +25.3 | +23.9 | +27.7 | +25.6 |
| GraFITi (PM) | 0.072 | 0.081 | 0.099 | 0.113 | 0.234 | 0.243 | 0.247 | 0.256 | 4.059 | 4.083 | 4.216 | 4.264 | 2.146 | 2.277 | 2.370 | 2.371 | 1.409 | 1.445 | 1.462 | 1.703 |
Appendix figures & tables11 assets
Supplementary material from the paper’s appendix.
Appendix
| Variable | ||||
| (hours) | (hours) | (hours) | (hours) | |
| BUN | ||||
| Creatinine | ||||
| GCS | ||||
| HCT | ||||
| Na |
| Model | Latent Dim. | # Layers | # Heads | Embed. Dim. | Refs / Patches |
| mTAND | 16, 32, 64, 128 | 1 | 1, 2, 4 | Time: 16, 32, 64 | Ref pts: 32, 64 |
| GRU-D | 16, 32, 64, 128 | 1 | – | – | – |
| CRU | 16, 32, 64, 128 | 1 | – | – | Basis: 10, 20 |
| f-CRU | 16, 32, 64, 128 | 1 | – | – | Basis: 10, 20 |
| Latent ODE | 16, 32, 64, 128 | Gen/Rec: 3 | – | Rec dim: 20, 32, 64 | GRU units: 20 |
| ODE-RNN | 16, 32, 64, 128 | 1 | – | – | – |
| Model | USHCN | Physionet | MIMIC-III | |||
| MSE | MAE | MSE | MAE | MSE | MAE | |
| FFN( D =2, lsd =256, residual =F) | 0.032 | 0.398 | 0.337 | 3.654 | 4.995 | 4.389 |
| FFN( D =2, lsd =64, residual =T) | 0.033 | 0.359 | 0.295 | 2.952 | 4.752 | 4.169 |
| FFN( D =2, lsd =128, residual =T) | 0.024 | 0.223 | 0.271 | 2.911 | 4.631 | 4.050 |
| FFN( D =2, lsd =256, residual =T) | 0.023 | 0.183 | 0.269 | 2.924 | 4.626 | 4.032 |
| CVTA | 0.003 | 0.112 | 0.226 | 2.543 | 3.957 | 3.778 |
| Model | USHCN | Physionet | MIMIC-III | MIMIC-IV | MIMIC-HiRes | |||||
| MSE | MAE | MSE | MAE | MSE | MAE | MSE | MAE | MSE | MAE | |
| CVTA | 0.003 0.002 | 0.112 0.033 | 0.226 0.009 | 2.543 0.024 | 3.957 0.009 | 3.778 0.016 | 2.026 0.006 | 2.811 0.016 | 0.513 0.018 | 3.941 0.034 |
| w/o VCE | 0.013\scalebox{0.75}{\pm 0.003} | 0.191\scalebox{0.75}{\pm 0.022} | 0.494\scalebox{0.75}{\pm 0.008} | 3.259\scalebox{0.75}{\pm 0.036} | 5.236\scalebox{0.75}{\pm 0.008} | 4.224\scalebox{0.75}{\pm 0.023} | 2.932\scalebox{0.75}{\pm 0.006} | 3.284\scalebox{0.75}{\pm 0.019} | 1.016\scalebox{0.75}{\pm 0.036} | 6.132\scalebox{0.75}{\pm 0.061} |
| w/o local slope | 0.006\scalebox{0.75}{\pm 0.004} | 0.124\scalebox{0.75}{\pm 0.031} | 0.241\scalebox{0.75}{\pm 0.009} | 2.626\scalebox{0.75}{\pm 0.032} | 4.005\scalebox{0.75}{\pm 0.009} | 3.829\scalebox{0.75}{\pm 0.021} | 2.129\scalebox{0.75}{\pm 0.009} | 2.886\scalebox{0.75}{\pm 0.021} | 0.903\scalebox{0.75}{\pm 0.018} | 4.606\scalebox{0.75}{\pm 0.031} |
| w/o key mask | 0.004\scalebox{0.75}{\pm 0.003} | 0.114\scalebox{0.75}{\pm 0.032} | 0.227\scalebox{0.75}{\pm 0.010} | 2.563\scalebox{0.75}{\pm 0.034} | 3.960\scalebox{0.75}{\pm 0.009} | 3.791\scalebox{0.75}{\pm 0.018} | 2.080\scalebox{0.75}{\pm 0.006} | 2.838\scalebox{0.75}{\pm 0.015} | 0.809\scalebox{0.75}{\pm 0.016} | 4.541\scalebox{0.75}{\pm 0.032} |
| w/o value mask | 0.010\scalebox{0.75}{\pm 0.001} | 0.213\scalebox{0.75}{\pm 0.031} | 0.237\scalebox{0.75}{\pm 0.007} | 2.595\scalebox{0.75}{\pm 0.021} | 4.110\scalebox{0.75}{\pm 0.007} | 3.875\scalebox{0.75}{\pm 0.023} | 2.117\scalebox{0.75}{\pm 0.006} | 2.888\scalebox{0.75}{\pm 0.014} | 0.832\scalebox{0.75}{\pm 0.018} | 4.688\scalebox{0.75}{\pm 0.030} |
| Model | Metric | USHCN | Physionet | MIMIC-III | MIMIC-IV | MIMIC-HiRes | |||||||||||||||
| H=1 | H=3 | H=5 | H=7 | H=1 | H=3 | H=5 | H=7 | H=1 | H=3 | H=5 | H=7 | H=1 | H=3 | H=5 | H=7 | H=1 | H=3 | H=5 | H=7 | ||
| CVTA | Mean MSE | 0.003 | 0.010 | 0.013 | 0.015 | 0.226 | 0.245 | 0.254 | 0.274 | 3.957 | 4.155 | 4.282 | 4.435 | 2.026 | 2.259 | 2.382 | 2.452 | 0.513 | 0.832 | 0.913 | 1.041 |
| Std MSE | 0.002 | 0.002 | 0.004 | 0.007 | 0.009 | 0.014 | 0.012 | 0.021 | 0.009 | 0.008 | 0.014 | 0.013 | 0.006 | 0.007 | 0.011 | 0.023 | 0.018 | 0.020 | 0.021 | 0.034 | |
| GRUwE (PM) | Mean MSE | 0.011 | 0.033 | 0.118 | 0.184 | 0.234 | 0.250 | 0.262 | 0.279 | 4.010 | 4.113 | 4.216 | 4.322 | 2.110 | 2.287 | 2.376 | 2.461 | 0.809 | 1.224 | 1.326 | 1.329 |
| Std MSE | 0.014 | 0.015 | 0.016 | 0.021 | 0.007 | 0.004 | 0.008 | 0.015 | 0.008 | 0.008 | 0.013 | 0.012 | 0.008 | 0.013 | 0.011 | 0.016 | 0.022 | 0.021 | 0.032 | 0.030 | |
| GRUwE+CVTA | Mean MSE | 0.004 | 0.017 | 0.024 | 0.028 | 0.211 | 0.230 | 0.239 | 0.258 | 3.783 | 3.925 | 4.039 | 4.151 | 2.015 | 2.191 | 2.285 | 2.334 | 0.604 | 0.931 | 0.959 | 0.989 |
| Target | Top-8 attended variables with attention scores |
| Lactate | HCO 3 (0.2208), Lactate (0.1651), RespRate (0.1095), TroponinT (0.0771), Platelets (0.0636), ALT (0.0589), PaCO 2 (0.0376), FiO 2 (0.0357) |
| Creatinine | Creatinine (0.8387), TroponinT (0.0392), Urine (0.0317), BUN (0.0146), Cholesterol (0.0133), Bilirubin (0.0112), HCO 3 (0.0110), ALP (0.0076) |
| Variable | Baseline MSE | Masked Top-3 MSE | Masked Bottom-3 MSE |
| Albumin | |||
| ALT | |||
| Bilirubin | |||
| BUN | |||
| Creatinine | |||
Multivariate Time Series Forecasting
Multivariate time series forecasting predicts future values across related variables by learning their temporal and cross-variable dependencies.
See all 289 papers