TPST: A Traffic Flow Prediction Model Based on Spatial–Temporal Identity

Abstract

With the constant dynamics of temporal dependence and spatial correlation, the interaction between them has become intricate. Existing work attempts to model precise temporal dependency and spatial correlation to make their interactions more accurate but ignores the importance of understanding how the two interact with each other. Thus, this article mines deeper into their interaction mechanism and proposes a new traffic prediction model called traffic flow prediction model based on spatial–temporal identity (TPST). It provides a new way named the spatial–temporal identity mechanism to model spatial–temporal interactions, which convert complex temporal dependence and spatial correlation into their identity information. Meanwhile, in order to improve spatial–temporal interaction resolution of the model, the method utilizes the down-sampling cross-convolution technique to contain more spatial–temporal history information and parses spatial–temporal interactions at different granularity. Experiments conducted with four real traffic flow datasets show that TPST consistently outperforms the other seven benchmark models, providing higher prediction accuracy with lower computational cost.