Curriculum-Aware Interpolate-then-Refine: Learned Physiological Time-Series Imputation under Realistic Missingness
Organizations: UNITES Lab, University of North Carolina at Chapel Hill
Abstract
Imputing physiological time series (arterial blood pressure, blood glucose, etc.) is essential for addressing the missingness that pervades clinical data. Yet modern imputation methods perform poorly in this domain: a recent benchmark found that simple linear interpolation outperformed every learned imputer on real-world clinical signals with realistic gaps. We show that this reflects two properties of physiological missingness that generic imputers ignore: gaps may occur when the signal is clinically extreme rather than typical, and gap lengths can easily span orders of magnitude. To this end, we introduce Curriculum-Aware Interpolate-then-Refine (CAIR), a two-stage framework for physiological time-series imputation. Our key motivation is to learn a coarse base curve and then repeatedly correct it toward physiological realism, rather than predict a gap in a single pass. Consequently, CAIR couples a bidirectional-GRU interpolator with a Transformer refiner that corrects its own estimate over three successive passes, trained jointly under a broad, signal-agnostic random-gap curriculum. We evaluate imputers stratified by gap length and missingness mechanism (MCAR, MAR, NMAR) rather than by a single average, and CAIR is the most accurate under every mechanism on continuous glucose monitoring (AI-READI) and arterial pressure in intensive care (MIMIC-III). Its margin over the strongest baseline grows with difficulty, from 9% under MCAR to 19% under value-dependent dropout, where generic learned imputers are weakest. We further show low reconstruction error alone does not recover the burden metrics clinicians act on: interpolants matching CAIR's error fail to preserve those metrics, imputers that recover them are far less accurate, and CAIR alone ranks among the best on both axes.
Figures & tables
| Method | MCAR | MAR | NMAR |
| CAIR (Ours) | 2.66 | 10.33 | 23.50 |
| linear interp | 2.91 | 12.34 | 28.94 |
| MICE | 3.26 | 19.83 | 49.35 |
| missForest | 3.33 | 16.54 | 40.90 |
| hot-deck | 5.04 | 16.06 | 43.44 |
| NN | 5.19 | 15.45 | 43.54 |
| Method | 15 min | 30 min | 45 min | 60 min | mean |
| CAIR (Ours) | 2.84 | 5.03 | 6.66 | 8.23 | 5.69 |
| akima | 2.76 | 5.01 | 6.87 | 8.61 | 5.81 |
| PCHIP | 2.84 | 5.14 | 6.92 | 8.58 | 5.87 |
| linear | 3.20 | 5.64 | 7.45 | 9.09 | 6.34 |
| SAITS | 4.60 | 7.54 | 9.76 | 11.79 | 8.42 |
| BRITS | 6.25 | 9.42 | 11.67 | 13.33 | 10.17 |
| Method | meal | sleep | asc | dip | comb | AVG |
| CAIR (Ours) | 15.25 | 24.94 | 9.01 | 7.49 | 16.37 | 14.61 |
| PCHIP | 16.56 | 29.26 | 7.86 | 7.36 | 18.65 | 15.94 |
| linear | 16.97 | 29.27 | 8.42 | 7.31 | 18.30 | 16.05 |
| akima | 17.82 | 36.73 | 8.02 | 7.24 | 19.64 | 17.89 |
| Kalman | 17.97 | 34.70 | 8.88 | 8.09 | 20.29 | 17.99 |
| Savitzky–Golay | 17.27 | 29.45 | 13.41 | 11.22 | 19.43 | 18.15 |
| (a) Physiological vs. random-gap training masks | |||
| Training masks | ABP RMSE | ||
| MCAR | MAR | NMAR | |
| linear | 4.40 | 9.23 | 11.85 |
| CAIR, CGM masks | 5.47 | 8.74 | 12.10 |
| CAIR, random-gap | 4.30 | 8.60 | 11.42 |
| (b) Clinical-burden recovery | |||
| Heart rate (bpm) | Resp. (br/min) | |||||||
| Method | 15 | 30 | 45 | 60 | 15 | 30 | 45 | 60 |
| Akima | 4.54 | 5.93 | 6.86 | 7.87 | 2.73 | 3.23 | 3.72 | 4.30 |
| AR (bidir.) | 4.71 | 5.62 | 6.09 | 6.72 | 2.77 | 3.09 | 3.33 | 3.57 |
| PCHIP | 4.45 | 5.62 | 6.27 | 7.05 | 2.63 | 3.07 | 3.43 | 3.76 |
| Linear | 4.40 | 5.52 | 6.12 | 6.86 | 2.61 | 3.00 | 3.34 | 3.65 |
| CAIR (ours) | ||||||||
| ABP-mean (mmHg) | HR (bpm) | |||||
| Method | MCAR | MAR | NMAR | MCAR | MAR | NMAR |
| CAIR (Ours) | 4.30 | 8.60 | 11.42 | 2.26 | 5.28 | 6.48 |
| linear interp | 4.40 | 9.23 | 11.85 | 2.22 | 5.50 | 6.42 |
| GP-VAE | 11.03 | 12.82 | 17.94 | 9.17 | 14.01 | 18.00 |
| M-RNN | 13.45 | 13.19 | 18.35 | 15.28 | 14.53 | 18.60 |
| Variant | MCAR | MAR | NMAR | ALL | |
| CAIR (full) | 2.66 | 10.33 | 23.50 | 12.16 | n/a |
| w/o aux. loss | 6.36 | 10.75 | 27.59 | 14.90 | |
| w/o residual | 4.45 | 10.34 | 27.38 | 14.06 | |
| GRU self-attn. | 4.15 | 10.22 | 27.14 | 13.84 | |
| linear interp | 2.91 | 12.34 | 28.94 | 14.73 |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Pretrain, then fine-tune | meal | sleep | asc | dip | comb | AVG |
| control (AI-READI only) | 15.60 | 27.52 | 5.96 | 5.14 | 14.74 | 13.79 |
| HUPA-UCM (T1D) | 15.99 | 26.32 | 6.21 | 5.36 | 15.66 | 13.91 |
| OhioT1DM (T1D) | 15.77 | 27.14 | 6.08 | 5.21 | 15.25 | 13.89 |
| Shanghai (T2D) | 15.71 | 27.58 | 6.13 | 5.20 | 15.18 | 13.96 |
| Method | meal | sleep | asc | dip | comb | AVG |
| CAIR (Ours) | 15.25 | 24.94 | 9.01 | 7.49 | 16.37 | 14.61 |
| Savitzky–Golay | 17.27 | 29.45 | 13.41 | 11.22 | 19.43 | 18.15 |
| EWMA | 20.59 | 31.69 | 21.45 | 18.32 | 23.11 | 23.03 |
| local mean | 22.66 | 32.23 | 24.00 | 20.65 | 24.40 | 24.79 |
| cubic spline | 22.64 | 45.22 | 21.90 | 18.62 | 23.85 | 26.44 |
| LOCF | 25.91 | 37.33 | 34.19 | 29.36 | 31.81 | 31.72 |
| Conditioning set | meal | sleep | asc | dip | AVG |
| target only (control) | 14.94 | 25.17 | 6.07 | 4.93 | 13.11 |
| heart rate | 14.93 | 25.16 | 6.03 | 4.92 | 13.10 |
| steps, calories | 14.89 | 25.25 | 6.10 | 4.97 | 13.14 |
| sleep state | 14.89 | 25.27 | 5.98 | 4.92 | 13.09 |
| respiration, stress | 14.79 | 24.96 | 6.00 | 4.89 | 12.98 |
| environment | 14.82 | 24.83 | 6.24 | 5.06 | 13.10 |
| Method | MCAR | MAR | NMAR |
| CAIR (univariate) | 4.26 | 8.58 | 11.44 |
| CAIR (multivariate) | 4.30 | 8.60 | 11.42 |
| linear interp | 4.40 | 9.23 | 11.85 |
| missForest | 4.62 | 9.60 | 12.91 |
| MICE | 4.88 | 9.02 | 12.65 |
| NN | 5.27 | 8.91 | 12.69 |
| Method | MCAR | MAR | NMAR |
| CAIR (univariate) | 2.26 | 5.36 | 6.72 |
| CAIR (multivariate) | 2.26 | 5.28 | 6.48 |
| linear interp | 2.22 | 5.50 | 6.42 |
| missForest | 2.37 | 7.40 | 9.25 |
| MICE | 2.49 | 7.55 | 9.63 |
| NN | 2.69 | 7.09 | 9.21 |
| Signal | Mechanism | ||
| ABP-mean (mmHg) | MCAR | ||
| MAR | |||
| NMAR | |||
| Heart rate (bpm) | MCAR | ||
| MAR | |||
| NMAR |
| Method | RMSE (mmHg) | Burden MRR |
| Low RMSE, burden not preserved | ||
| linear interp | 11.85 | |
| LOCF | 12.79 | |
| Fourier | 13.56 | |
| Recovers burden, high RMSE | ||
| MICE | 12.65 | |
| Fill | MCAR AUROC | NMAR AUROC |
| oracle (recorded trace) | 0.679 | 0.678 |
| mean-fill | 0.677 | 0.683 |
| NN | 0.680 | 0.685 |
| CAIR (multivar.) | 0.688 | 0.688 |
| linear interp | 0.692 | 0.688 |
| PCHIP | 0.692 | 0.689 |
| RMSE (mg/dL) | TIR recovery | |||
| Study group | CAIR | linear | CAIR | linear |
| healthy | 18.29 | 21.49 | 0.165 | 0.097 |
| pre-diabetes | 15.90 | 21.15 | 0.467 | 0.382 |
| oral medication | 33.96 | 40.19 | 0.601 | 0.556 |
| insulin-dependent | 27.23 | 35.17 | 0.608 | 0.489 |
Time Series Imputation
Time series imputation reconstructs missing observations in sequential data using temporal, cross-variable, and domain-specific patterns.
See all 58 papers