MT-CNN: A Lightweight Spatial-Temporal Convolutional Neural Network for Deep Learning of Complex Trajectory Distributions based on Area Partitioning

Rongkun Ye,Zhihao Xu,Zhiqiang Lv,Jianbo Li
DOI: https://doi.org/10.1109/IJCNN60899.2024.10651261
2024-06-30
Abstract:With the rapid development of 5G technology and deep learning, intelligent transportation systems (ITS) have been constantly improved and perfected. In the ITS, trajectory prediction, as a key part, requires the processing of a large amount of urban data, which places a high demand on computing resources. To reduce computational complexity and demand, this work proposes a lightweight spatial-temporal model based on convolutional neural networks which has fewer model parameters, faster prediction speed and higher prediction accuracy. This work transforms the complex trajectory into regular areas, reducing model computation while also more intuitively representing the change in trajectory. This work extracts multi-scale spatial features through spatial convolutions of different sizes to comprehensively understand the spatial variation of trajectory. Then, the temporal correlation of the trajectory is extracted through multi-layer time convolution. In addition, this paper introduces the trajectory relationship matrix(TRM) to enhance the global spatial relationship of the pedestrian trajectory and capture different spatial correlations for different areas. It analyzes the relationship between the areas where the trajectory is located by learning the historical trajectory. This work introduces extra factors such as pedestrian motion direction and speed to assist prediction, improving the model’s prediction ability and robustness. Experimental results show that the proposed MT-CNN model improves the accuracy by 22.7% compared to other models, with better performance and generalization ability, while also having excellent computational time and energy consumption.
Engineering,Computer Science
What problem does this paper attempt to address?