Search papers, labs, and topics across Lattice.
The paper introduces Uniform Inductive Spatio-Temporal Kriging (UniSTOK), a framework designed to improve inductive kriging models when dealing with heterogeneous missing data in spatio-temporal signal inference. UniSTOK uses a dual-branch input consisting of original observations and jigsaw-augmented data to synthesize proxy signals for missing entries, processing them in parallel with a shared backbone and missingness mask modulation. Experiments on real-world datasets show that UniSTOK consistently improves performance under various missing data patterns by adaptively fusing the outputs of the two branches via dual-channel attention.
UniSTOK tackles the messy reality of missing data in spatio-temporal kriging by cleverly synthesizing proxy signals and adaptively fusing them with original observations, leading to significant performance gains.
Spatio-temporal kriging aims to infer signals at unobserved locations from observed sensors and is critical to applications such as transportation and environmental monitoring. In practice, however, observed sensors themselves often exhibit heterogeneous missingness, forcing inductive kriging models to rely on crudely imputed inputs. This setting brings three key challenges: (1) it is unclear whether an value is a true signal or a missingness-induced artifact; (2) missingness is highly heterogeneous across sensors and time; (3) missing observations distort the local spatio-temporal structure. To address these issues, we propose Uniform Inductive Spatio-Temporal Kriging (UniSTOK), a plug-and-play framework that enhances existing inductive kriging backbones under missing observation. Our framework forms a dual-branch input consisting of the original observations and a jigsaw-augmented counterpart that synthesizes proxy signals only at missing entries. The two branches are then processed in parallel by a shared spatio-temporal backbone with explicit missingness mask modulation. Their outputs are finally adaptively fused via dual-channel attention. Experiments on multiple real-world datasets under diverse missing patterns demonstrate consistent and significant improvements.
