A novel reinforced dynamic graph convolutional network model with data imputation for network-wide traffic flow prediction

Yong Chen,Xiqun (Michael) Chen
DOI: https://doi.org/10.1016/j.trc.2022.103820
2022-10-01
Abstract:Traffic data missing issues due to unpredictable equipment failure, extreme weather, and other reasons have brought great challenges to traffic flow prediction modeling. In this paper, a novel reinforced dynamic graph convolutional network model is proposed to simultaneously conduct data imputation and network-wide traffic flow prediction. First, a multi-graph convolutional fusion network is proposed for data imputation by using the graph convolutional network to analyze the propagation law of traffic states between traffic flow detection stations in both time and space dimensions. Second, to enhance the robustness of network-wide traffic flow prediction, a dynamic graph learning method based on deep reinforcement learning is proposed to adaptively generate the graph adjacency matrix to represent the dynamic spatiotemporal dependencies between the stations. Finally, experimental results on two real-world traffic datasets show that the proposed method outperforms other baseline methods and can effectively extract the data missing features and spatiotemporal dependence features between the stations. The visualization results of the graph adjacency matrix indicate that the proposed method can effectively identify the influential traffic stations in the process of traffic flow prediction, and the extracted dependencies between the stations are interpretable. The proposed model has strong generalization in tackling network-wide traffic flow prediction tasks with different data missing rates and missing patterns, and can be extended to assist decision-makers in enhancing traffic management and mitigating traffic congestion.
What problem does this paper attempt to address?