English

Temporal Graph MLP Mixer for Spatio-Temporal Forecasting

Machine Learning 2025-01-20 v1

Abstract

Spatiotemporal forecasting is critical in applications such as traffic prediction, climate modeling, and environmental monitoring. However, the prevalence of missing data in real-world sensor networks significantly complicates this task. In this paper, we introduce the Temporal Graph MLP-Mixer (T-GMM), a novel architecture designed to address these challenges. The model combines node-level processing with patch-level subgraph encoding to capture localized spatial dependencies while leveraging a three-dimensional MLP-Mixer to handle temporal, spatial, and feature-based dependencies. Experiments on the AQI, ENGRAD, PV-US and METR-LA datasets demonstrate the model's ability to effectively forecast even in the presence of significant missing data. While not surpassing state-of-the-art models in all scenarios, the T-GMM exhibits strong learning capabilities, particularly in capturing long-range dependencies. These results highlight its potential for robust, scalable spatiotemporal forecasting.

Keywords

Cite

@article{arxiv.2501.10214,
  title  = {Temporal Graph MLP Mixer for Spatio-Temporal Forecasting},
  author = {Muhammad Bilal and Luis Carretero Lopez},
  journal= {arXiv preprint arXiv:2501.10214},
  year   = {2025}
}
R2 v1 2026-06-28T21:09:22.529Z