SA-LSTM: A Trajectory Prediction Model for Complex Off-Road Multi-Agent Systems Considering Situation Awareness Based on Risk Field
Yuning Wang,Jiahao Wang,Junkai Jiang,Shaobing Xu,Jianqiang Wang
DOI: https://doi.org/10.1109/TVT.2023.3287227
IF: 6.8
2023-01-01
IEEE Transactions on Vehicular Technology
Abstract:Autonomous Vehicles have wide-ranging applications in off-road environments. Off-road vehicular scenes can be abstracted as multi-agent systems, and trajectory prediction is a critical process for context understanding. Agents compete and cooperate other to pursue their individual targets, which makes trajectories complicated and changeable. Hence, in order to derive precise predictions, it is necessary to reason how agents interact with each other. However, current prediction algorithms lack a unified and appropriatemethod for spatiotemporal reasoning. Previous methods mainly rely on vehicle-lane constraints to capture features, which is only applicable in structured environments. To address this issue, this article proposes a novel off-road multi-agent trajectory prediction framework called SA-LSTM. This framework comprises situation awareness extraction and Long Short Term Memory (LSTM) prediction backbones. Based on the analysis of agents' movement patterns, we decompose actions into the maintenance of previous movements and the strategy variations in response to environment situations. In situation awareness extraction, risk field and pooling layers are applied to filter interpretable awareness. As for prediction backbones, proper LSTM networks are selected to adapt the task features. In short-term predictions, a residual mechanism is used to preserve physical inertia, while for long-term predictions autoregression process is applied. We also establish a multi-agent dataset based on human-manipulated chaseand-run tasks to train and validate the performance of SA-LSTM. Experiments show that compared with the best baseline model at each prediction length, SA-LSTM reduces themean absolute error by 7.27%, 32.99%, and 28.07% at the prediction time of 0.6, 3.0, and 6.0 seconds, respectively, proving better prediction accuracy.