OceanVP: A HYCOM based benchmark dataset and a relational spatiotemporal predictive network for oceanic variable prediction

Zhensheng Shi,Haiyong Zheng,Junyu Dong
DOI: https://doi.org/10.1016/j.oceaneng.2024.117748
IF: 5
2024-04-17
Ocean Engineering
Abstract:Oceanic variable prediction (OVP) is significant in marine climate, environment, resources, and engineering. Traditional OVP approaches are mainly based on numerical ocean model (NOM) methods. Recently, deep learning (DL) has provided new methods and techniques for OVP, and diverse data and model studies have emerged for specific OVP tasks involving ocean temperature and current prediction, etc. Although these efforts have made remarkable progress, there is still no common benchmark for DL-based OVP data and method research. To address this problem, we construct a benchmark dataset, dubbed OceanVP, which is based on the HY brid C oordinate O cean M odel (HYCOM) reanalysis data. In addition, we propose a REL ational S patio T emporal P redictive N etwork (REL-STPN) as the benchmark method to learn the spatiotemporal relations from the data of oceanic variables, and adopt three main variables (water temperature, velocity, and salinity) from HYCOM to conduct the experiments. The experiments on the OceanVP dataset indicate that our REL-STPN method achieves state-of-the-art (SOTA) performance and fast model iterations on a single GPU with a training duration of less than 4 h and an inference speed of over 1000 FPS. Furthermore, we improve a feature visualization method to visually analyze and explain the changing process of the learned predictive features.
engineering, civil, ocean, marine,oceanography
What problem does this paper attempt to address?