Deep learning framework for vessel trajectory prediction using auxiliary tasks and convolutional networks
Yuyol Shin,Namwoo Kim,Hyeyeong Lee,Soh Young In,Mark Hansen,Yoonjin Yoon
DOI: https://doi.org/10.1016/j.engappai.2024.107936
IF: 8
2024-02-01
Engineering Applications of Artificial Intelligence
Abstract:With the exponential growth in vessel traffic and the increasing complexity of maritime operations, there is a pressing need for reliable and efficient methods to forecast vessel movements. The accurate prediction of vessel trajectories plays a pivotal role in various maritime applications, including route planning, collision avoidance, and maritime traffic management. Traditional statistical and machine learning approaches have shown limitations in capturing the complex spatial–temporal patterns of vessel movements. Deep learning techniques have emerged as a promising solution due to their ability to handle large-scale datasets and capture nonlinear relationships. This study proposes a novel deep learning-based vessel trajectory prediction framework for AIS data using Auxiliary tasks and Convolutional encoders (AIS-ACNet). The framework utilizes various features of Automatic Identification System (AIS) data, including geographical positions, and vessel dynamics such as Speed Over Ground (SOG), and Course Over Ground (COG), for trajectory prediction. The AIS-ACNet employs parallel convolutional encoder networks with feature fusion layers to control the weight of auxiliary features. The model is trained with a multi-task learning objective that includes auxiliary SOG and COG prediction tasks. This framework enhances the model's vessel trajectory prediction performance by efficiently incorporating vessel dynamics. The proposed framework is evaluated on a real-world AIS dataset retrieved from the Port of Houston, Texas, USA. The result shows that AIS-ACNet achieves a 5.31% increase in average displacement error compared to the best performing baseline model. Also, the model demonstrates ability to perform robustly on various types of trajectories.
automation & control systems,computer science, artificial intelligence,engineering, electrical & electronic, multidisciplinary