首页|基于贝叶斯优化的时间卷积网络船舶航迹预测

基于贝叶斯优化的时间卷积网络船舶航迹预测

扫码查看
[目的]为提高船舶航迹预测精度和计算效率,解决传统方法容易出现的梯度爆炸、计算时间长等问题,提出基于改进的贝叶斯优化算法与时间卷积神经网络的航迹预测模型.[方法]首先,通过引入时间模式注意力机制,提取各输入特征的权重,保证航迹历史数据的时序性,同时引入可逆残差网络,减少时间卷积神经网络模型训练过程中占用的内存;然后,再采用贝叶斯优化算法对时间卷积网络中的超参数(内核大小K、膨胀系数d)进行寻优;最后,采用五折交叉验证方法对模型进行验证,获得最优模型后进行航迹预测.[结果]采用AIS采集的航迹数据验证,结果表明,在弱耦合、中耦合和强耦合航迹预测中,均方根误差分别平均提高 5.5×10-5,3.5×10-4 和 6×10-4.[结论]所提出网络对复杂航迹具有良好的适应性,其预测精度均优于传统模型及LSTM模型,在耦合较强的航迹中仍能保持较高的预测精度.
Ship track prediction based on Bayesian optimization in temporal convolutional networks
[Objective]As the traditional ship trajectory prediction method is prone to gradient explosion and long calculation time,this paper seeks to improve its accuracy and calculation efficiency by proposing a ship trajectory prediction model based on an improved Bayesian optimization algorithm(IBOA)and temporal convolution network(TCN).[Method]A temporal pattern attention(TPA)mechanism is introduced to ex-tract the weights of each input feature and ensure the timing of the historical flight track data.At the same time,a reversible residual network(RevNet)is introduced to reduce the memory occupied by TCN model training.The IBOA is then used to find the optimality of the hyperparameters in the TCN(size of kernel K,ex-pansion coefficient d).The model is finally validated using a five-fold cross-validation method,and trajectory prediction is carried out after obtaining the optimal model.[Result]The trajectory data is collected by auto-matic identification system(AIS)and verified.The root mean square error(RMSE)is found to be increased by 5.5×10-5,3.5×10-4 and 6×10-4 in weak coupling,medium coupling and strong coupling track prediction respec-tively.[Conclusion]The proposed network has good adaptability to complex trajectories and higher accur-acy than the traditional model and long short-term memory(LSTM)model,while maintaining high prediction accuracy for trajectories with strong coupling.

navigationneural networksBayesian optimization algorithmtemporal convolutional net-workstemporal pattern attention mechanism modulereversible residual networksAIS data

李金源、朱发新、滕宪斌、毕齐林

展开 >

浙江海洋大学 船舶与海运学院,浙江 舟山 316022

广州航海学院 轮机工程学院,广东 广州 510725

导航 神经网络 贝叶斯优化算法 时间卷积网络 时间模式注意力机制模块 可逆残差网络 AIS数据

2024

中国舰船研究
中国舰船研究设计中心

中国舰船研究

CSTPCD北大核心
影响因子:0.496
ISSN:1673-3185
年,卷(期):2024.19(6)