A lightweight Spatial-Temporal Graph Neural Network for Long-term Time Series Forecasting
About
We propose Lite-STGNN, a lightweight spatial-temporal graph neural network for long-term multivariate forecasting that integrates decomposition-based temporal modeling with learnable sparse graph structure. The temporal module applies trend-seasonal decomposition, while the spatial module performs message passing with low-rank Top-$K$ adjacency learning and conservative horizon-wise gating, enabling spatial corrections that enhance a strong linear baseline. Lite-STGNN achieves state-of-the-art accuracy on four benchmark datasets for horizons up to 720 steps, while being parameter-efficient and substantially faster to train than transformer-based methods. Ablation studies show that the spatial module yields 4.6% improvement over the temporal baseline, Top-$K$ enhances locality by 3.3%, and learned adjacency matrices reveal domain-specific interaction dynamics. Lite-STGNN thus offers a compact, interpretable, and efficient framework for long-term multivariate time series forecasting.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Time Series Forecasting | Electricity (test) | MSE0.178 | 72 | |
| Multivariate Time-series Forecasting | Electricity (test) | -- | 36 | |
| Multivariate Time-series Forecasting | Weather standard (test) | MSE0.171 | 11 | |
| Time Series Forecasting | Exchange standard (test) | MSE0.088 | 11 | |
| Multivariate Time-series Forecasting | Traffic Standard (test) | MSE0.536 | 8 |