STAMImputer is a Spatio-Temporal Attention Mixture of Experts network designed for traffic data imputation.
It addresses challenges in extracting features from block-wise missing data scenarios and handling distribution shifts for nonstationary traffic data.
The network incorporates a Mixture of Experts framework to capture latent spatio-temporal features and uses a Low-rank guided Sampling Graph ATtention mechanism for spatial feature propagation.
Extensive experiments on four traffic datasets show that STAMImputer outperforms existing state-of-the-art approaches in traffic data imputation.