JMSE, Vol. 12, Pages 2031: Vessel Trajectory Prediction Based on AIS Data: Dual-Path Spatial–Temporal Attention Network with Multi-Attribute Information

JMSE, Vol. 12, Pages 2031: Vessel Trajectory Prediction Based on AIS Data: Dual-Path Spatial–Temporal Attention Network with Multi-Attribute Information

Journal of Marine Science and Engineering doi: 10.3390/jmse12112031

Authors: Feilong Huang Zhuoran Liu Xiaohe Li Fangli Mou Pengfei Li Zide Fan

With the rapid growth of the global shipping industry, the increasing number of vessels has brought significant challenges to navigation safety and management. Vessel trajectory prediction technology plays a crucial role in route optimization and collision avoidance. However, current prediction methods face limitations when dealing with complex vessel interactions and multi-dimensional attribute information. Most models rely solely on global modeling in the temporal dimension, considering spatial interactions only later, failing to capture dynamic changes in trajectory interactions at different time points. Additionally, these methods do not fully utilize the multi-attribute information in AIS data, and the simple concatenation of attributes limits the model’s potential. To address these issues, this paper proposes a dual spacial–temporal attention network with multi-attribute information (DualSTMA). This network models vessel behavior and interactions through two distinct paths, comprehensively considering individual vessel intentions and dynamic interactions. Moreover, we divide vessel attributes into dynamic and static categories, with dynamic attributes fused during feature preprocessing, and with static attributes being controlled through a gating mechanism during spatial interactions to regulate the importance of neighboring vessel features. Benchmark tests on real AIS data show that DualSTMA significantly outperforms existing methods in prediction accuracy. Ablation studies and visual analyses further validate the model’s reliability and advantages.