Variability Aware Recursive Neural Network (VARNN): A Residual-Memory Model for Capturing Temporal Deviation in Sequence Regression Modeling
Organizations: Department of Computer Science, Illinois Institute of Technology, Chicago, IL, USA · Department of Computer Science, Emory University, Atlanta, GA, USA
Abstract
Real-world time-series regression often involves non-stationarity, heteroscedasticity, and regime changes, under which recent prediction errors may contain structured information about local temporal mismatch between model predictions and observations. Learning how to represent and reuse these errors can therefore provide useful information for subsequent prediction. We introduce the Variability-Aware Recursive Neural Network (VARNN), a residual-aware architecture for supervised time-series regression that learns an explicit residual-memory state from recent prediction errors and uses it to condition subsequent predictions. Specifically, VARNN maps scalar prediction innovations into a learned nonlinear, vector-valued residual representation over a short context. Across nine datasets spanning energy, healthcare, and environmental domains, VARNN achieves lower test MSE than the compared static, lag-based, and sequence-model baselines. Targeted ablations further show that learned projected residual memory improves predictive accuracy over direct scalar residual feedback, supporting the benefit of a learned nonlinear representation of prediction deviations.
Figures & tables
| Domain | Dataset | Size | Missing | #Feat. | Target |
|---|---|---|---|---|---|
| Energy | Appliances | 19,735 | No | 27 | Appliance energy use (Wh) |
| ETTh1 | 17,420 | No | 6 | Oil temperature | |
| ETTh2 | 17,420 | No | 6 | Oil temperature | |
| Healthcare | BIDMC–HR | 25,436 | No | 375 | Heart rate (bpm) |
| BIDMC–RR | 25,436 | No | 375 | Respiratory rate (breaths/min) | |
| BIDMC–SPO2 | 25,436 | No | 375 | Oxygen saturation (SpO 2 ) |
| Model | ENERGY | Healthcare | Environmental | |||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Appliances | ETTh1 | ETTh2 | BIDMC HR | BIDMC RR | BIDMC SPO2 | Beijing PM2.5 | Beijing PM10 | WEATHER | ||||||||||
| Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | |
| Static (no lags) | ||||||||||||||||||
| LR | 0.00799 | 0.00657 | 0.02792 | 0.03335 | 0.02243 | 0.04112 | 0.01830 | 0.02256 | 0.00984 | 0.00822 | 0.03659 | 0.03103 | 0.00212 | 0.00171 | 0.00375 | 0.00242 | 0.01516 | 0.01483 |
| RF | 0.00057 | 0.04815 | 0.00234 | 0.02888 | 0.00174 | 0.03274 | 0.00031 | 0.01842 | 0.00032 | 0.00705 | 0.00057 | 0.03925 | 0.00012 | 0.00162 | 0.00025 | 0.00227 | 0.00229 | 0.01235 |
| MLP | 0.00833 | 0.00778 | 0.00833 | 0.00778 | 0.02622 | 0.02049 | 0.00216 | 0.00168 | 0.01276 | 0.00933 | 0.04683 | 0.03452 | 0.00216 | 0.00168 | 0.00392 | 0.00224 | 0.01413 | 1.21301 |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Appliances | BIDMC-HR | Beijing PM2.5 | |
|---|---|---|---|
| 2 | 0.00338 | 0.00016 | 0.00028 |
| 3 | 0.00358 | 0.00018 | 0.00029 |
| 5 | 0.00328 | 0.00015 | 0.00026 |
| 7 | 0.00340 | 0.00016 | 0.00026 |
| 9 | 0.00340 | 0.00020 | 0.00027 |
| Power | HR | PM2.5 | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Model | ||||||||||||
| RNN | 0.007057 | 0.007459 | 0.007310 | 0.007337 | 0.010834 | 0.019918 | 0.017077 | 0.016677 | 0.001407 | 0.001773 | 0.001995 | 0.002268 |
| GRU | 0.006663 | 0.006711 | 0.006870 | 0.006772 | 0.016132 | 0.017971 | 0.017825 | 0.015146 | 0.001412 | 0.001765 | 0.002015 | 0.002303 |
| LSTM-H | 0.006522 | 0.006510 | 0.006793 | 0.006731 | 0.017026 | 0.018671 | 0.016860 | 0.017115 | 0.001365 | 0.001747 | 0.001984 | 0.002257 |
| LSTM-H+C | 0.006484 | 0.007736 | 0.007053 | 0.006923 | 0.017054 | 0.017482 | 0.018643 | 0.016395 | 0.001408 | 0.001757 | 0.002021 | 0.002260 |
| VARNN-RM | 0.003430 | 0.005029 | 0.005817 | 0.006147 | 0.000153 | 0.000335 | 0.000397 | 0.000483 | 0.000230 | 0.000656 | 0.001082 | 0.001493 |
| Model | ENERGY | Healthcare | Environmental | |||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Appliances | ETTh1 | ETTh2 | BIDMC HR | BIDMC RR | BIDMC SPO2 | Beijing PM2.5 | Beijing PM10 | WEATHER | ||||||||||
| Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | Train | Test | |
| VARNN–RM | 0.00412 | 0.00328 | 0.00037 | 0.00020 | 0.00041 | 0.00048 | 0.00016 | 0.00015 | 0.00017 | 0.00031 | 0.00015 | 0.00024 | 0.00024 | 0.00026 | 0.00085 | 0.00058 | 0.00028 | 0.00045 |
| VARNN–RM+AM | 0.00378 | 0.00329 | 0.00035 | 0.00018 | 0.00044 | 0.00049 | 0.00016 | 0.00015 | 0.00017 | 0.00029 | 0.00015 | 0.00022 | 0.00022 | 0.00025 | 0.00077 | 0.00057 | 0.00037 | 0.00042 |
| VARNN–ARM | 0.00382 | 0.00330 | 0.00035 | 0.00017 | 0.00040 | 0.00046 | 0.00015 | 0.00015 | 0.00017 | 0.00029 | 0.00016 | 0.00024 | 0.00022 | 0.00025 | 0.00077 | 0.00058 | 0.00026 | 0.00031 |
| VARNN–RM+ARM | 0.00374 | 0.00329 | 0.00037 | 0.00017 | 0.00041 | 0.00051 | 0.00016 | 0.00015 | 0.00017 | 0.00029 | 0.00015 | 0.00024 | 0.00021 | 0.00024 | 0.00078 | 0.00055 | 0.00024 | 0.00032 |