Welcome to Journal of Graphics share: 

Journal of Graphics ›› 2022, Vol. 43 ›› Issue (4): 677-684.DOI: 10.11996/JG.j.2095-302X.2022040677

• Computer Graphics and Virtual Reality • Previous Articles     Next Articles

Sea surface temperature prediction algorithm combined with residual spatial-temporal attention mechanism

  

  1. 1. Department of Information Technology, Shanghai Ocean University, Shanghai 201306, China;
    2. East China Sea Standard Metrology Center, State Oceanic Administration, Shanghai 201306, China
  • Online:2022-08-31 Published:2022-08-15
  • Contact: HE Qi (1979), associate professor, Ph.D. Her main research interests cover ocean big data analysis, big data storage, workflow and business process management, and service computing, etc
  • Supported by:
    National Natural Science Foundation of China (61972240); Youth Project of National Natural Science Foundation of China (41906179);
    Capacity Building Project of Some Local Universities of Shanghai Science and Technology Commission (20050501900)

Abstract:

Sea surface temperature (SST) is closely related to global climate change, ocean disasters, and ocean ecosystems, so the accurate prediction of SST is an important topic. The existing regional SST prediction methods treat the time series of SST data as a series of matrixes, each corresponding to the regional SST at a particular time. The spatial and temporal features are extracted from the matrix series for later SST prediction. However, the existing SST prediction methods fail to fully consider the imbalanced influence of temporal and spatial features on the SST, leading to the neglection of some key information and limiting the improvement of prediction accuracy. To address this problem, we proposed a regional SST prediction method (CRA-ConvLSTM) combining temporal attention mechanism and spatial attention mechanism. This enabled the model to dynamically assign different influence weights to the temporal features at different times and spatial features at different locations, thereby improving the accuracy of SST prediction. Specifically, the input regional SST time series was first encoded into multi-layer feature vectors by a convolutional neural network (CNN), and local features were extracted. Then, the residual time attention module was constructed to learn the attention weight at different moments adaptively, and the key features of the time dimension were extracted. The residual spatial attention module was designed to extract the key features of different points in the region in terms of the spatial dimension. In addition, the attention mechanism combined with the residual structure can avoid performance degradation caused by information reduction in the network. Experimental results show that the proposed model could achieve 0.19 and 99.43% respectively in terms of the root mean square error (RMSE) and prediction accuracy (PACC), which is superior to other methods and effectively improves the prediction accuracy of SST.


Key words:

CLC Number: