Abstract:Sea surface temperature (SST) is a key parameter in ocean hydrology. Currently, existing SST prediction methods fail to fully utilize the potential spatial correlation between variables. To address this challenge, we propose a spatiotenporal UNet (ST-UNet) model based on the UNet model. In particular, in the encoding phase of ST-UNet, we use parallel convolution with different kernel sizes to efficiently extract spatial features, and use ConvLSTM to capture temporal features based on the utilization of spatial features. Atrous Spatial Pyramid Pooling (ASPP) module is placed at the bottleneck of the network to further incorporate the multi-scale features, allowing the spatial features to be fully utilized. The final prediction is then generated in the decoding stage using parallel convolution with different kernel sizes similar to the encoding stage. We conducted a series of experiments on the Bohai Sea and Yellow Sea SST data set, as well as the South China Sea SST data set, using SST data from the past 35 days to predict SST data for 1, 3, and 7 days in the future. The model was trained using data spanning from 2010 to 2021, with data from 2022 being utilized to assess the model's predictive performance. The experimental results show that the model proposed in this research paper achieves excellent results at different prediction scales in both sea areas, and the model consistently outperforms other methods. Specifically, in the Bohai Sea and Yellow Sea sea areas, when the prediction scales are 1, 3, and 7 days, the MAE of ST-UNet outperforms the best results of the other three compared models by 17%, 12%, and 2%, and the MSE by 16%, 18%, and 9%, respectively. In the South China Sea, when the prediction ranges are 1, 3, and 7 days, the MAE of ST-UNet is 27%, 18%, and 3% higher than the best of the other three compared models, and the MSE is 46%, 39%, and 16% higher, respectively. Our results highlight the effectiveness of the ST-UNet model in capturing spatial correlations and accurately predicting SST. The proposed model is expected to improve marine hydrographic studies.

STCANet: Spatiotemporal Coupled Attention Network for Ocean Surface Current Prediction

Ca-STANet: Spatiotemporal Attention Network for Chlorophyll-a Prediction with Gap-Filled Remote Sensing Data.

A Physics-Guided Attention-Based Neural Network for Sea Surface Temperature Prediction

Arctic Sea Ice Concentration Prediction Using Spatial Attention Deep Learning

Spatial-temporal Offshore Current Field Forecasting Using Residual-learning Based Purely CNN Methodology with Attention Mechanism

Spatial-Temporal Self-Attention Network for Flow Prediction

MSSTN: a multi-scale spatio-temporal network for traffic flow prediction

HiCAN: Hierarchical Convolutional Attention Network for Sequence Modeling.

Spatio-Temporal Traffic Flow Prediction Based on Coordinated Attention

StHCFormer: A Multivariate Ocean Weather Predicting Method Based on Spatiotemporal Hybrid Convolutional Attention Networks

Prediction of Sea Surface Temperature Using U-Net Based Model

Spatial–temporal combination and multi-head flow-attention network for traffic flow prediction

Spatio-Temporal Self-Attention Network for Video Saliency Prediction

SDPNet: A Novel DeepLearning Method for Ocean Surface Current Prediction

Vessel trajectory prediction based on spatio-temporal graph convolutional network for complex and crowded sea areas

SFANet: Spatial-Frequency Attention Network for Weather Forecasting

A deep learning model with spatio-temporal graph convolutional networks for river water quality prediction

Hybrid Attention Spatial-Temporal Network for Video Saliency Prediction

ST-DCN: A Spatial-Temporal Densely Connected Networks for Crowd Flow Prediction.

Sea Surface Temperature Prediction Using ConvLSTM-Based Model with Deformable Attention

Interpretable Crowd Flow Prediction with Spatial-Temporal Self-Attention