Predicting trajectories of coastal area vessels with a lightweight Slice-Diff self attention

• 發表在: Complex & Intelligent Systems
• Main Authors: Jinxu Zhang, Jin Liu, Xiliang Zhang, Lai Wei, Zhongdai Wu, Junxiang Wang
• 格式: Article
• 語言: 英语
• 出版: Springer 2025-04-01
• 主題: Irregular trajectory feature extraction; Trajectory slice and difference; Vessel trajectory prediction; Coastal collision avoidance; Automatic identification system
• 在線閱讀: https://doi.org/10.1007/s40747-025-01877-x
• ISSN: 2199-4536; 2198-6053

Abstract Accurate prediction of vessel trajectories in coastal areas poses a significant challenge due to the large number of irregular trajectories. Existing trajectory prediction studies predominantly employ recurrent neural network (RNN) and Transformer-based methods. However, the former often encounter challenges such as gradient vanishing or exploding, and the latter tend to focus on global temporal dependencies, making it difficult to capture local irregular trajectory features in coastal maritime areas. Recently, graph-based methods have also been used to predict trajectories, however processing graph-structured data introduces significant increase in computation. In responding to these, this paper proposes a framework based on a novel lightweight Slice-Diff self attention, which consists of several key components. Firstly, the trajectory slice difference encoder (TSDE) utilizes slice embedding (SE) to enrich the cross dimensional dependencies contained in the input sequence, and then combines Slice-Diff self attention (SDSA) and fine-grained convolution (FGC) to comprehensively capture sequence-specific positional and directional information. Additionally, an auxiliary model, stepping bidirectional long short-term memory (S-BiLSTM) is developed to capture global temporal dependencies within the whole sequence. Finally, the fine-grained trajectory features obtained from TSDE and the global temporal dependencies compensated by S-BiLSTM are combined through the fully connected layer to predict coastal vessel trajectories. Extensive experimental results on three real-world automatic identification system (AIS) datasets demonstrate the effectiveness of proposed framework against other baselines.