An Intelligent Deep Learning Framework for Traffic Flow Imputation and Short-term Prediction Based on Dynamic Features

Xianhui Zong,Yong Qi,He Yan,Qiaolin Ye
DOI: https://doi.org/10.1016/j.knosys.2024.112178
IF: 8.139
2024-01-01
Knowledge-Based Systems
Abstract:The accurate prediction of traffic flow has emerged as a focal point in the cutting -edge sphere of intelligent transportation. Extant methodologies rely on deep learning for short-term forecasts but are hampered by two critical limitations: the omission of vital traffic flow data and an overemphasis on static rather than dynamic attributes. This paper presents a solution to these challenges through the creation of a dynamic attention generative adversarial network (DATGAN) that is specifically engineered to impute missing traffic data accurately. DATGAN introduces two groundbreaking features: first, an innovative attention module that is adept at identifying local correlations, and second, novel spatial -correction graph convolutional layers that are designed to capture the ever-changing spatial dynamic dependencies. Building on this foundation, the parallel spatial - temporal Transformer (PSTTransformer) is proposed, which integrates both spatial and temporal attention blocks to manage dynamic spatial -temporal dependencies for future short-term traffic predictions. This dualblock approach not only embeds periodic information via a fusion self -attention mechanism in the temporal attention blocks, but also synergizes graph convolutional networks with self -attention mechanisms in the spatial attention blocks through a gate mechanism. A comprehensive comparative analysis conducted on traffic datasets revealed that our methodology outperforms current leading models by reducing the mean squared error by 5% in data imputation tasks and shows robust performance in traffic prediction scenarios.
What problem does this paper attempt to address?