TVGCN: Time-Variant Graph Convolutional Network for Traffic Forecasting

Abstract Traffic forecasting is a very challenging task due to the complicated and dynamic spatial-temporal correlations between traffic nodes. Most existing methods measure the spatial correlations by defining physical or virtual graphs with distance or similarity measurement, which is constructed with stable edge connections by some prior knowledge. However, the use of such graphs with stable edge connections limits the variations of spatial correlations between traffic nodes at different times, which can not capture the hidden dynamic patterns of traffic graphs. This paper proposes a Time-Variant Graph Convolutional Network (TVGCN) to overcome this limitation. Architecturally, a time-variant spatial convolutional module (TV-SCM) is developed on two graphs without any prior knowledge. One graph is learned to capture the stable spatial correlations of the traffic graph, while the other graph is evolved to model dynamic spatial correlations at different times. Such two graphs are combined hierarchically together under the framework of graph convolutional network (GCN). Moreover, a gated multi-scale temporal convolutional module (GMS-TCM) is designed to extract long-range temporal dependencies within traffic nodes, which are further supplied to the TV-SCM to mutually explore the spatial correlations between traffic nodes. Extensive experiments conducted on three real-world traffic datasets indicate the effectiveness and superiority of our proposed approach.