Uniform Inductive Spatio-Temporal Kriging
About
Inductive spatio-temporal kriging infers signals at unobserved locations from observed sensors, but real-world observations are often incomplete and exhibit block-wise missingness caused by failures, interruptions, or maintenance. A common impute-then-krige pipeline suffers from objective mismatch: better reconstruction on observed sensors does not necessarily improve downstream kriging, and value-dependent imputation bias can be propagated to unobserved nodes. We propose UniSTOK, a plug-and-play framework for inductive spatio-temporal kriging under incomplete observations. We first introduce Reliability-guided Signal Regulation (RSR), which estimates entry-wise reliability from temporal continuity and spatial support, and uses it to regulate the input signals so that reliable observations are emphasized while long-gap or weakly supported entries are suppressed before spatial propagation. We further introduce Residual Bias Calibration (RBC), which estimates value-conditioned residual prototypes after the main predictor converges and learns context-correction amplitudes to adaptively calibrate systematic over- or under-estimation in final kriging predictions. Extensive experiments on real-world datasets show that UniSTOK consistently improves multiple kriging backbones.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Spatio-Temporal Kriging | METR-LA (Random) | MAE5.79 | 10 | |
| Spatio-Temporal Kriging | METR-LA (Block) | MAE5.92 | 10 | |
| Spatio-Temporal Kriging | METR-LA (Mixed) | MAE5.84 | 10 | |
| Spatio-Temporal Kriging | PEMS-BAY (random) | MAE3.39 | 10 | |
| Spatio-Temporal Kriging | PEMS-BAY (Block) | MAE3.5 | 10 | |
| Spatio-Temporal Kriging | PEMS-BAY (Mixed) | MAE3.37 | 10 | |
| Spatio-Temporal Kriging | NREL (Random) | MAE1.16 | 10 | |
| Spatio-Temporal Kriging | NREL (Block) | MAE1.49 | 10 | |
| Spatio-Temporal Kriging | NREL (Mixed) | MAE1.16 | 10 |