SwinLSTM:Improving Spatiotemporal Prediction Accuracy using Swin Transformer and LSTM

Song Tang,Chuang Li,Pu Zhang,RongNian Tang

2023-12-23

Abstract:Integrating CNNs and RNNs to capture spatiotemporal dependencies is a prevalent strategy for spatiotemporal prediction tasks. However, the property of CNNs to learn local spatial information decreases their efficiency in capturing spatiotemporal dependencies, thereby limiting their prediction accuracy. In this paper, we propose a new recurrent cell, SwinLSTM, which integrates Swin Transformer blocks and the simplified LSTM, an extension that replaces the convolutional structure in ConvLSTM with the self-attention mechanism. Furthermore, we construct a network with SwinLSTM cell as the core for spatiotemporal prediction. Without using unique tricks, SwinLSTM outperforms state-of-the-art methods on Moving MNIST, Human3.6m, TaxiBJ, and KTH datasets. In particular, it exhibits a significant improvement in prediction accuracy compared to ConvLSTM. Our competitive experimental results demonstrate that learning global spatial dependencies is more advantageous for models to capture spatiotemporal dependencies. We hope that SwinLSTM can serve as a solid baseline to promote the advancement of spatiotemporal prediction accuracy. The codes are publicly available at <a class="link-external link-https" href="https://github.com/SongTang-x/SwinLSTM" rel="external noopener nofollow">this https URL</a>.

Computer Vision and Pattern Recognition,Artificial Intelligence

What problem does this paper attempt to address?

The paper aims to address the accuracy issue in spatiotemporal prediction tasks. Specifically, it proposes a new recurrent unit—SwinLSTM, which combines the Swin Transformer module and a simplified version of LSTM. Compared to traditional methods that integrate Convolutional Neural Networks (CNN) and Recurrent Neural Networks (RNN), SwinLSTM leverages the self-attention mechanism to capture global spatial dependencies, thereby improving the accuracy of spatiotemporal predictions.

The authors conducted experimental validation on multiple datasets, including Moving MNIST, TaxiBJ, Human3.6m, and KTH. The experimental results show that SwinLSTM outperforms existing state-of-the-art methods on these datasets. In particular, SwinLSTM significantly improves prediction accuracy compared to ConvLSTM. Additionally, the paper analyzes the impact of different components on model performance through ablation studies and further explains the working mechanism of the model through feature map visualization. Overall, SwinLSTM demonstrates effectiveness and generalization capability in spatiotemporal prediction tasks.