基于Terrain-CGAN-Att-RF的水下滑翔机航点预测

Outlet Location Prediction of Underwater Gliders Based on Terrain-CGAN-Att-RF

  • 摘要: 水下滑翔机在复杂海洋环境中执行任务时,航点(出水点)空间位置的不确定性显著影响任务规划的有效性和准确性。为提升航点预测精度,针对现有机器学习方法在复杂地形和小样本等场景下预测性能受限的问题,本文首先构建一种融合地形与海流数据的水下滑翔机航点预测模型,即地形-条件生成对抗网络-注意力加权-随机森林模型(Terrain-CGAN-Att-RF)。该模型通过引入地形信息作为环境数据,增强模型对不同地形条件下水下滑翔机运动状态的分析能力;其次,有针对性地利用条件生成对抗网络(Conditional Generative Adversarial Network, CGAN),并将DBSCAN聚类(Density-Based Spatial Clustering of Applications with Noise)标签作为条件输入进行样本增强,缓解样本数量稀缺问题,提升生成样本的多样性与合理性,并设计注意力加权机制的邻域增强模块用以计算目标样本与邻域样本间的空间与环境相似性,提取更具相关性的输入特征。最终,本文以增强后的特征为输入构建了基于随机森林的预测模型。基于南海海试数据的验证结果表明,本文提出Terrain-CGAN-Att-RF模型在小样本条件下仍表现出优异性能,航程预测的平均绝对误差(MAE)、均方根误差(RMSE)和决定系数(R2)分别为291.34 m、401.88 m和0.9356,航向预测的MAE、RMSE和R2分别为24.28°、33.55°和0.8504

     

    Abstract: When underwater gliders perform missions in complex oceanic environments, the spatial uncertainty of waypoints (surfacing points) significantly affects the effectiveness and accuracy of mission planning. To improve waypoint prediction accuracy and address the limited performance of existing machine learning methods in scenarios involving complex terrain and small sample sizes, this paper proposes an underwater glider waypoint prediction model that integrates terrain and ocean current data, namely the Terrain-Conditional Generative Adversarial Network-Attention-weighted-Random Forest (Terrain-CGAN-Att-RF) model. First, the model introduces topographic information as environmental data to enhance its capability to analyze the motion states of underwater gliders under diverse terrain conditions. Second, a Conditional Generative Adversarial Network (CGAN) is purposefully employed, utilizing DBSCAN (Density-Based Spatial Clustering of Applications with Noise) clustering labels as conditional inputs for sample augmentation. This approach alleviates data scarcity while improving the diversity and rationality of the generated samples. Furthermore, a neighborhood enhancement module with an attention-weighted mechanism is designed to calculate the spatial and environmental similarity between target and neighborhood samples, thereby extracting more relevant input features. Finally, a Random Forest-based prediction model is constructed using these enhanced features as inputs. Validation results based on sea trial data from the South China Sea demonstrate that the proposed Terrain-CGAN-Att-RF model maintains superior performance even under small-sample conditions. For dead reckoning distance prediction, the Mean Absolute Error (MAE), Root Mean Square Error (RMSE), and Coefficient of Determination (R2) are 291.34 m, 401.88 m, and 0.9356, respectively. For heading prediction, the MAE, RMSE, and R2 are 24.28°, 33.55°, and 0.8504, respectively.

     

/

返回文章
返回