Evaluating Deep Multivariate Imputation Models on Wearable Device Data
Authors: Skye Goodman, Roussel Desmond Nzoyem, Leandro Junges, Peter Kissack, Yasser Qureshi, Amberly Brigden, Jeff Clark, Nawid Keshtmand
Organizations: School of Engineering Mathematics and Technology, University of Bristol, UK · Centre for AI Fundamentals, University of Manchester, UK · Centre for Systems Modelling and Quantitative Biomedicine, University of Birmingham, UK
Wearable device data enables continuous health monitoring, but suffers from structured missingness: features sharing a physical sensor drop out together. Deep imputation methods such as BRITS and SAITS have seen limited evaluation on multimodal physiological data under realistic missingness, and existing benchmarks use random-point holdout protocols that incorrectly assume missingness is independent across features and time. Using data from a person with epilepsy recorded on a Garmin smartwatch, we develop an evaluation protocol that mines contiguous missing-run templates from training data, stratifies them by per-feature gap-length quantiles, and injects them as block masks with preserved co-missingness structure. A matched training protocol exposing models to the same missingness distribution reduces BRITS's severe-gap MAE by 43%, demonstrating the potential benefit of the proposed evaluation and training protocol within this single-participant dataset. We further extend BRITS with time-of-day encoding and a circadian harmonic channel. No single model dominates: linear interpolation is optimal for slow-moving features over short gaps; extended BRITS achieves lower MAE on dynamic cardiac features in moderate and severe gaps; and SAITS better preserves the ground-truth distribution by Jensen-Shannon distance despite higher MAE. Ultimately, model rankings strongly depend on evaluation designs. By exposing how traditional evaluation methods obscure true model capabilities, our transferable protocol establishes critical steps towards developing better imputation strategies for future multi-sensor wearable datasets.
Figures & tables
Feature
Description
#Runs
P50
P75
P90
Max
steps
Cumulative daily step count
37
56.0
105.0
212.4
236
steps_rate
Interval-level step rate
38
53.0
101.25
209.8
236
bodyBattery
Garmin Body Battery score (5–100)
83
35.0
98.0
207.8
1484
hr
Heart rate (bpm)
162
25.0
63.75
154.7
1492
ibi
Inter-beat interval (ms)
212
18.5
39.25
120.1
1491
sleep
Sleep stage (0–4)
107
16.0
39.0
185.0
1333
Table 1: Contiguous missing-run length statistics per feature, computed on the training set. A “missing run” is defined as one or more consecutive missing timestamps for a given feature. Run length is measured as the number of consecutive missing timestamps in the run. #Runs reports the total number of missing episodes in the training data. P50 , P75 and P90 denote the 50th, 75th and 90th percentiles of run length L , and Max denotes the longest observed run in the training set.
Figure 1: Conditional missingness matrix for the wearable features, where each cell reports P(Ymissing∣Xmissing) . The heatmap highlights sensor-coupled co-missingness, including close to global feature dropouts when the accelerometer channels are missing, and a strongly coupled PPG-derived cluster of variables.
Figure 2: Wearable-realistic training protocol used for BRITS and SAITS. Contiguous missing-run templates are extracted from training data and stored in a severity-bucketed library. During training, a template is sampled and injected into each window. Loss is computed on both naturally observed entries (reconstruction) and artificially held-out entries (masked imputation).
Feature
BRITS
BRITS-ext
SAITS
SAITS-ext
Typical ( L≤P50 )
hr
0.386 ± 0.017 / 5.93%
0.366 ± 0.016 / 5.64%
0.377 ± 0.015 / 5.84%
0.385 ± 0.016 / 5.92%
ibi
0.280 ± 0.017 / 5.12%
0.276 ± 0.014 / 5.06%
0.296 ± 0.019 / 5.39%
0.298 ± 0.016 / 5.44%
device_stress
0.268 ± 0.019 / 17.47%
0.254 ± 0.017 / 17.77%
0.309 ± 0.022 / 20.01%
0.292 ± 0.018 / 19.23%
pulseOx
0.163 ± 0.021 / 0.45%
0.173 ± 0.020 / 0.47%
0.247 ± 0.023 / 0.68%
0.231 ± 0.028 / 0.63%
steps
0.085 ± 0.008 / 69.7%
0.106 ± 0.011 / 68.6%
0.068 ± 0.006 / 68.9%
0.095 ± 0.009 / 71.3%
Table 2: Effect of the temporal context extension on BRITS and SAITS under wearable-realistic block masking for the core nine features, and extended with time-of-day and circadian auxiliary channels appended. Each cell reports test MAE (mean ± std) and sMAPE (%) per target feature and gap severity bucket over 10 seeded evaluation trials sharing identical sampled mask templates. Bold entries indicate the better variant within each model and row.
Feature
LOCF
LI
BRITS-ext
SAITS
Typical ( L≤P50 )
hr
0.527 ± 0.021 / 8.10%
0.394 ± 0.021 / 6.14%
0.366 ± 0.016 / 5.64%
0.377 ± 0.015 / 5.84%
ibi
0.415 ± 0.028 / 7.50%
0.324 ± 0.019 / 5.85%
0.276 ± 0.014 / 5.06%
0.296 ± 0.019 / 5.39%
device_stress
0.409 ± 0.028 / 24.95%
0.336 ± 0.024 / 20.89%
0.254 ± 0.017 / 17.77%
0.309 ± 0.022 / 20.01%
pulseOx
0.148 ± 0.017 / 0.41%
0.129 ± 0.015 / 0.35%
0.173 ± 0.020 / 0.47%
0.247 ± 0.023 / 0.68%
steps
0.042 ± 0.012 / 15.56%
0.021 ± 0.006 / 10.44%
0.106 ± 0.011 / 68.6%
0.068 ± 0.006 / 68.9%
Table 3: Comparison of BRITS-ext and SAITS against univariate classical baselines (LOCF - last observation carried forward, LI - linear interpolation) under wearable-realistic block masking. Each cell reports MAE (mean ± std) / sMAPE (%). Bold indicates the lowest MAE in each row.
Figure 3: Average bucket MAE for BRITS-ext and SAITS under the original training schedule (grey) versus the modified wearable-realistic block training schedule (green). Both variants are evaluated on the same masks, at the same positions, across the same 10 seeded trials. The only variable is the training objective. BRITS sees substantial gains across every bucket; SAITS gains more selectively, in the moderate and severe regimes.
Typical
Moderate
Severe
Feature
BRITS-ext
SAITS
BRITS-ext
SAITS
BRITS-ext
SAITS
hr
0.169 ± 0.023
0.152 ± 0.015
0.163 ± 0.009
0.142 ± 0.014
0.250 ± 0.014
0.231 ± 0.008
ibi
0.174 ± 0.016
0.178 ± 0.016
0.157 ± 0.013
0.143 ± 0.009
0.240 ± 0.013
0.222 ± 0.016
pulseOx
0.601 ± 0.058
0.689 ± 0.031
0.601 ± 0.058
0.689 ± 0.031
0.662 ± 0.033
0.693 ± 0.039
steps
0.244 ± 0.025
0.219 ± 0.028
0.202 ± 0.011
0.176 ± 0.010
0.192 ± 0.009
0.148 ± 0.011
steps_rate
0.212 ± 0.011
0.205 ± 0.011
0.181 ± 0.007
0.181 ± 0.006
0.169 ± 0.003
0.169 ± 0.003
Table 4: Jensen–Shannon distance (10-seed mean ± std) between imputed and ground-truth distributions, computed per feature and gap-severity bucket. Lower is better. Bold indicates the better-performing model for each feature and bucket.
Figure 4: Kernel density estimates of imputed values against ground-truth distributions at evaluation-mask positions, pooled across all three gap-severity buckets.
School of Integrated Circuits, Zhejiang University Hangzhou, China · Shanghai Innovation Institute Shanghai, China · School of Data Science, Fudan University Shanghai, China +1