基于多任务Informer模型的船舶轨迹预测及行为识别研究

    Research on ship trajectory prediction and behavior recognition based on multi-task Informer model

    • 摘要: 为有效预判航行风险,并为船舶避碰、交通管理等决策提供重要依据,研究了一种基于多任务Informer模型的船舶轨迹预测及行为识别模型。该模型以Informer框架为基础,并引入多任务学习模式,通过设计多任务损失函数将船舶行为识别与轨迹预测并联训练,解决了AIS数据中船舶行为不准确无法作为模型输入的问题;在模型训练时,并设计基于同方差不确定性的损失函数自适应更新策略,自适应分配两个任务的损失权重。利用太仓航段水域中的真实AIS数据进行试验中多任务的Informer船舶轨迹预测模型在轨迹预测中的损失比LSTM和Informer模型分别降低了40.2%和14.7%;在行为识别任务中多任务模型的识别准确率比LSTM和Informer模型分别提升了11.7%和5.95%。表明了多任务模型能在有效提升船舶轨迹预测的性能的同时实现船舶对行为的准确识别。

       

      Abstract: Ship trajectory prediction and behavior recognition can help effectively assess navigational risks and provide an important basis for decision-making in collision avoidance and traffic management. To improve the accuracy of ship trajectory prediction and behavior recognition, this paper studies a multi-task Informer model for simultaneous trajectory prediction and behavior recognition. Based on the Informer framework, the model incorporates a multi-task learning approach. It addresses the issue that inaccurate ship behavior records in AIS data cannot be directly used as model inputs by designing a multi-task loss function that jointly trains behavior recognition and trajectory prediction in parallel. During training, an adaptive updating strategy for the loss function-based on homoscedastic uncertainty-is designed to automatically allocate weights to the losses of the two tasks. Evaluated using real AIS data from the Taicang sector waters, the multi-task Informer model reduces trajectory prediction loss by 40. 2% and 14. 7% compared to LSTM and Informer models, respectively. In behavior recognition, the multi-task model improves accuracy by 11. 7% and 5. 95% compared to LSTM and Informer models, respectively. The results demonstrate that the multi-task model effectively enhances the performance of ship trajectory prediction while achieving accurate recognition of ship behavior.

       

    /

    返回文章
    返回