MTGCN: A Multitask Deep Learning Model for Traffic Flow Prediction

摘要

The prediction of traffic flow is of great importance to urban planning and intelligent transportation systems. Recently, deep learning models have been applied to study this problem. However, there still exist two main limitations: (1) They do not effectively model dynamic traffic patterns in irregular regions; (2) The traffic flow of a region is strongly correlated to the transition-flow between different regions, while this issue is largely ignored by existing approaches. To address these issues, we propose a multitask deep learning model called MTGCN for a more accurate traffic flow prediction. First, to process the input traffic network data, we propose using graph convolution in place of traditional grid-based convolution to model spatial dependencies between irregular regions. Second, as original graph convolution can not well respond to traffic dynamics, we design a novel attention mechanism to capture dynamic traffic patterns. At last, to obtain a more accurate prediction result, we integrate two correlated tasks which respectively predict two types of traffic flows (region-flow and transition-flow) as a whole, by combining the representations learned from each task in a rational way. We conduct extensive experiments on two real-world datasets and the results show that our proposed method achieves better performance compared with other baseline models.