ST-GMLP: A concise spatial-temporal framework based on gated multi-layer perceptron for traffic flow forecasting

• Published in: Neural networks Vol. 184; p. 107074
• Main Authors: Luo, Yong, Zheng, Jianying, Wang, Xiang, E, Wenjuan, Jiang, Xingxing, Zhu, Zhongkui
• Format: Journal Article
• Language: English
• Published: United States Elsevier Ltd 01.04.2025
• Subjects: Deep learning; Forecasting - methods; Gated mechanisms; Humans; Multi-layer perceptron; Multilayer Perceptrons; Neural Networks, Computer; Spatial-temporal data; Spatio-Temporal Analysis; Traffic prediction; Spatial-temporal data; Deep learning; Traffic prediction; Gated mechanisms; Multi-layer perceptron
• ISSN: 0893-6080; 1879-2782; 1879-2782
• DOI: 10.1016/j.neunet.2024.107074

The field of traffic forecasting has been the subject of considerable attention as a critical component in alleviating traffic congestion and improving urban services. Given the regular patterns of human activities, it is evident that traffic flow is inherently periodic. However, most of existing studies restrict themselves to recent historical observations and typically yield structurally and computationally complex models, which greatly limits the forecasting accuracy and hinders the application of models in realistic situations. To this end, this paper proposes a concise framework named Spatial-Temporal Gated Multi-Layer Perceptron (ST-GMLP), aiming to enhance the forecasting performance by leveraging the temporal patterns of different scales with a simple and effective structure. Nevertheless, due to the incorporation of more historical features, the presence of distribution shifts between periods further restricts the forecasting accuracy. To address the above issue, ST-GMLP employs a parallel structure of learning the interdependencies of traffic flow in both spatial node and temporal directions, and then establishes the interactions between time and space to effectively mitigate the adverse effects due to temporal distribution shifts. Owing to the utilization of MLP with gated mechanisms (GMLP) for modeling the spatial-temporal interdependencies, ST-GMLP has significant advantages in terms of training efficiency and resources occupation. Extensive experimental findings indicate that ST-GMLP exhibits superior performance in comparison to state-of-the-art methods.