Multi-task Graph Neural Network for Truck Speed Prediction under Extreme Weather Conditions

Reza Safarzadeh Ramhormozi,Arash Mozhdehi,Saeid Kalantari,Yunli Wang,Sun,Xin Wang
DOI: https://doi.org/10.1145/3557915.3561029
2022-01-01
Abstract:Truck speed prediction plays a key role in truck transportation management. However, it is a very challenging task since the truck traffic usually shows complex patterns. Most of the existing traffic prediction methods lack the ability to model the dynamic spatial-temporal correlations of truck traffic or ignore contributing contextual factors that impact traffic. Also, truck traffic data is typically sparse and noisy, which makes the truck speed prediction an even more challenging task. How to improve the truck speed prediction by taking advantage of other relevant truck traffic information (such as the truck flow) has not been investigated in depth. Additionally, traffic congestions and poor driving conditions caused by extreme weather conditions can make sudden changes in the general pattern of the truck speed. In this paper, we propose a novel Multi-Task Context Based Gated Recurrent Unit Graph Convolutional Network (MT-C2G) to predict the truck speed under extreme weather conditions. MT-C2G includes four major components: The spatial dependence learning component captures the spatial dependencies shaped by the topological structure of the road network. Truck traffic feature temporal dependence modeling component is built to acquire the temporal dependencies involved in the truck traffic features, and contextual feature temporal dependence modeling component employs a layer of GRU units to capture the temporal dependencies of contextual factors. The multi-task learning component then leverages the information between the truck speed and flow prediction tasks through attention mechanism for improving the performance. Moreover, a data augmentation method SMOTE is utilized to balance the data with the extreme weather conditions. Experiments on two real datasets demonstrate that the proposed MT-C2G fairly outperforms six state-of-the-art traffic prediction methods.
What problem does this paper attempt to address?