A hybrid prediction model of vessel trajectory based on attention mechanism and CNN-GRU

Jian Cen,JiaXi Li,Xi Liu,JiaHao Chen,HaiSheng Li,WeiSheng Huang,LinZhe Zeng,JunXi Kang,Silin Ke
DOI: https://doi.org/10.1177/14750902231226162
2024-02-07
Proceedings of the Institution of Mechanical Engineers Part M Journal of Engineering for the Maritime Environment
Abstract:Proceedings of the Institution of Mechanical Engineers, Part M: Journal of Engineering for the Maritime Environment, Ahead of Print. With the increase in global shipping volumes and the complexity of maritime transport systems, vessel trajectory prediction serves an important tool in improving maritime safety. However, most existing vessel trajectory prediction methods focus on a single feature and unable fuse high-dimensional features. To solve these problems, CNN-GRU model with a hybrid attention mechanism (AM) is proposed based on Automatic Identification System (AIS) data. First convolutional neural network (CNN) is proposed to extract the spatio-temporal information of the trajectory data. Then a gated recurrent unit (GRU) is designed to extract the temporal relationship of the trajectories. Finally, AM is introduced to learn the deep-level features and predict the vessel trajectories. To validate the effectiveness of the model, experiments are conducted on three real AIS datasets. In comparison with other models, the method has a high trajectory prediction accuracy.
engineering, marine
What problem does this paper attempt to address?