Spatial Temporal Attention based Target Vehicle Trajectory Prediction for Internet of Vehicles

Ouhan Huang,Huanle Rao,Xiaowen Cai,Tianyun Wang,Aolong Sun,Sizhe Xing,Yifan Sun,Gangyong Jia
2025-01-02
Abstract:Forecasting vehicle behavior within complex traffic environments is pivotal within Intelligent Transportation Systems (ITS). Though this technology plays a significant role in alleviating the prevalent operational difficulties in logistics and transportation systems, the precise prediction of vehicle trajectories still poses a substantial challenge. To address this, our study introduces the Spatio Temporal Attention-based methodology for Target Vehicle Trajectory Prediction (STATVTPred). This approach integrates Global Positioning System(GPS) localization technology to track target movement and dynamically predict the vehicle's future path using comprehensive spatio-temporal trajectory data. We map the vehicle trajectory onto a directed graph, after which spatial attributes are extracted via a Graph Attention Networks(GATs). The Transformer technology is employed to yield temporal features from the sequence. These elements are then amalgamated with local road network structure maps to filter and deliver a smooth trajectory sequence, resulting in precise vehicle trajectory <a class="link-external link-http" href="http://prediction.This" rel="external noopener nofollow">this http URL</a> study validates our proposed STATVTPred method on T-Drive and Chengdu taxi-trajectory datasets. The experimental results demonstrate that STATVTPred achieves 6.38% and 10.55% higher Average Match Rate (AMR) than the Transformer model on the Beijing and Chengdu datasets, respectively. Compared to the LSTM Encoder-Decoder model, STATVTPred boosts AMR by 37.45% and 36.06% on the same datasets. This is expected to establish STATVTPred as a new approach for handling trajectory prediction of targets in logistics and transportation scenarios, thereby enhancing prediction accuracy.
Robotics,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to accurately predict the trajectory of the target vehicle in a complex traffic environment. Specifically, although the Intelligent Transportation System (ITS) has played an important role in alleviating the operational problems of logistics and transportation systems, **accurately predicting vehicle trajectories remains a major challenge**. ### Problem Background With the rapid development of the global economy and population growth, the demand for transportation, public transportation, and other urban services is constantly increasing, leading to the intensification of urban problems such as safety, environmental pollution, and congestion. To solve these problems, the Intelligent Transportation System (ITS) emerged, using advanced automation technologies (such as sensors, communication networks, and AI - based decision - making modules) to improve urban traffic efficiency, safety, and sustainability. ### Research Motivation In the Intelligent Transportation System, **autonomous driving technology and vehicle trajectory prediction are key components**. High - precision vehicle trajectory prediction can support key functions in self - driving cars and traffic management systems, such as real - time decision - making, safe and coordinated interaction, etc. However, the performance of existing methods in complex scenarios is limited, especially when dealing with long - term sequence prediction and the connectivity between road segments, and the accuracy decreases. ### Solution To solve the above problems, this research introduced a target - vehicle - trajectory - prediction method based on the spatio - temporal attention mechanism (STATVTPred). The main innovations of this method include: 1. **Spatio - temporal Feature Extraction**: Represent the road network as a directed graph and map vehicle trajectories onto these graphs. Extract spatial features through Graph Attention Networks (GATs) and extract temporal features using the Transformer model. 2. **Combining Local Road Network Structure**: Combine the extracted spatial and temporal features with the local road network structure diagram, filter and generate a smooth trajectory sequence, thereby achieving accurate vehicle trajectory prediction. 3. **Experimental Verification**: Verification was carried out on the T - Drive and Chengdu taxi trajectory data sets, and the results show that STATVTPred is significantly superior to the existing Transformer and LSTM encoder - decoder models in terms of the Average Matching Rate (AMR). ### Main Contributions - **Map Matching**: Realize the map - matching process, align the moving - target trajectory with the road network, and convert the resulting road - segment sequence into a feature vector through an embedding layer. - **Spatio - temporal Feature Capture**: Use the encoding module of the Transformer to capture temporal features, and at the same time use the graph - attention module to extract spatial features. Both use the self - attention mechanism to effectively allocate dependency weights. - **Continuous Trajectory Generation**: Generate a continuous predicted road - segment sequence by filtering the output of the Transformer decoding module to achieve accurate target - trajectory prediction. Through these improvements, STATVTPred can provide more accurate vehicle - trajectory prediction in complex traffic scenarios, thereby enhancing the overall performance and reliability of the Intelligent Transportation System.