3D Ocean Temperature Prediction Via Graph Neural Network with Optimized Attention Mechanisms

Mingyu Ou,Shijie Xu,Bin Luo,Hengan Zhou,Mingye Zhang,Pan Xu,Hongna Zhu
DOI: https://doi.org/10.1109/lgrs.2024.3398709
IF: 5.343
2024-01-01
IEEE Geoscience and Remote Sensing Letters
Abstract:Ocean temperature prediction plays crucial roles in the ocean related fields. The graph neural networks (GNNs) show advantages for modeling complex environmental issues. However, the prior methods typically focus on node features to predict ocean temperature. Note that the edge features are essential, especially for 3D ocean temperature prediction. In this paper, a graph neural network with optimized attention mechanisms (GNN-OAM) model is proposed to predict 3D ocean temperature. The GNN-OAM introduces a random forest (RF) module to capture the non-stationary temporal dependencies. Especially, the optimized attention mechanisms, via combination of multiple adjacency matrices construct edge features, are presented to capture the dynamic spatial dependencies. The prediction performance of GNN-OAM model is evaluated in the 3D ocean temperature experimental dataset, the results show that the GNN-OAM achieves ocean temperature prediction with high accuracy. And the MAEs are 0.146 and 0.26, when predicting temperatures in next one day and five days.
What problem does this paper attempt to address?