A multi-view bidirectional spatiotemporal graph network for urban traffic flow imputation

Accurate estimation of missing traffic data is one of the essential components in intelligent transportation systems (ITS). The non-Euclidean data structure and complex missing traffic flow patterns make it challenging to capture nonlinear spatiotemporal correlations of missing traffic flow, which are critical for the imputation of missing traffic data. In this study, we propose a novel multi-view bidirectional spatiotemporal graph network called Multi-BiSTGN to impute urban traffic data with complex missing patterns. First, three spatiotemporal graph sequences are constructed to comprehensively describe traffic conditions from different temporal correlation views, i.e. temporal closeness view, daily periodicity view, and weekly periodicity view. Then, three bidirectional spatiotemporal graph networks are fused by a parametric-matrix-based method to obtain the final imputation results. To train the Multi-BiSTGN model, a novel loss function that considers the interactions between three temporal correlation views is designed to optimize the parameters of the Multi-BiSTGN model. The proposed model was validated on real-world traffic datasets collected in Wuhan, China. Experimental results showed that Multi-BiSTGN outperformed ten existing baselines under different missing types (random missing, block missing, and mixed missing) and missing rates.