EAAI Journal 2025 Journal Article
A three-dimensional dynamic spatial-temporal graph neural network for ocean temperature field prediction
- Shuai Zhang
- ZhuoLin Li
- Xiaoyu He
- Jie Yu
- LingYu Xu
Accurate prediction of the ocean temperature field is vital for the protection of marine ecosystems under climate change. However, most existing methods only consider temporal changes, ignoring the rich three-dimensional (3D) dynamic spatial characteristics of the ocean temperature field. To improve prediction accuracy, we propose a fine-grained modeling method that uses the spatial correlations in the ocean temperature field, called the 3D Dynamic Spatial-Temporal Graph Neural Network (3D-DSTGN). Specifically, according to the 3D spatial features and dynamic spatial dependencies of the ocean temperature field, we first decompose the spatial correlation of the ocean temperature field into long-term static and short-term dynamic parts through statistical analysis of real ocean temperature datasets. We then build a 3D dynamic graph structure learning module to create static and dynamic graph structures with 3D spatial features to model and capture the corresponding spatial correlations. Next, based on the two graph structures, we apply a dual-mode graph convolution block to fully capture the dynamic spatial dependencies of the ocean temperature field. Furthermore, we use a multi-scale temporal convolution block to capture complex temporal dependencies from historical ocean temperature data. Finally, the dual-mode graph convolution block and the multi-scale temporal convolution block construct the spatio-temporal recurrent module, which extracts complex dynamic spatio-temporal contextual dependencies. On a large-scale real ocean temperature dataset from the sea surface to a subsurface depth of approximately 2, 000 m, 3D-DSTGN outperforms other baselines in experiments across different temporal and spatial scales, effectively modeling the dynamic three-dimensional spatial relationships of the ocean temperature field.