Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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.

Henok Tenaw Moges, Deshendran Moodley• 2025

Related benchmarks

TaskDatasetResultRank
Time Series ForecastingElectricity (test)
MSE0.178
72
Multivariate Time-series ForecastingElectricity (test)--
36
Multivariate Time-series ForecastingWeather standard (test)
MSE0.171
11
Time Series ForecastingExchange standard (test)
MSE0.088
11
Multivariate Time-series ForecastingTraffic Standard (test)
MSE0.536
8
Showing 5 of 5 rows

Other info

Follow for update