A Multi-Spatial-Scale Ocean Sound Speed Profile Prediction Model Based on a Spatio-Temporal Attention Mechanism

Abstract

Marine researchers rely heavily on ocean sound velocity, a crucial hydroacoustic environmental metric that exhibits large geographical and temporal changes. Nowadays, spatio-temporal series prediction algorithms are emerging, but their prediction accuracy requires improvement. Moreover, in terms of ocean sound speed, most of these models predict an ocean sound speed profile (SSP) at a single coordinate position, and only a few predict multi-spatial-scale SSPs. Hence, this paper proposes a new data-driven method called STA-Conv-LSTM that combines convolutional long short-term memory (Conv-LSTM) and spatio-temporal attention (STA) to predict SSPs. We used a 234-month dataset of monthly mean sound speeds in the eastern Pacific Ocean from January 2004 to June 2023 to train the prediction model. We found that using 24 months of SSPs as the inputs to predict the SSPs of the following month yielded the highest accuracy. The results demonstrate that STA-Conv-LSTM can achieve predictions with an accuracy of more than 95% for both single-point and three-dimensional scenarios. We compared it against recurrent neural network, LSTM, and Conv-LSTM models with optimal parameter settings to demonstrate the model’s superiority. With a fitting accuracy of 95.12% and the lowest root-mean-squared error of 0.8978, STA-Conv-LSTM clearly outperformed the competition with respect to prediction accuracy and stability. This model not only predicts SSPs well but also will improve the spatial and temporal forecasts of other marine environmental factors.

Keywords

• spatio-temporal attention mechanism
• deep learning
• sound velocity profile
• STA-Conv-LSTM
• multi-spatial-scale prediction