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

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.