arXiv reaDer
SwinLSTM:Swin Transformer と LSTM を使用した時空間予測精度の向上
SwinLSTM:Improving Spatiotemporal Prediction Accuracy using Swin Transformer and LSTM
CNN と RNN を統合して時空間依存関係をキャプチャすることは、時空間予測タスクの一般的な戦略です。ただし、局所的な空間情報を学習する CNN の特性により、時空間依存関係を捕捉する効率が低下し、その結果、予測精度が制限されます。この論文では、Swin Transformer ブロックと、ConvLSTM の畳み込み構造をセルフ アテンション メカニズムに置き換える拡張機能である簡略化された LSTM を統合する、新しいリカレント セル SwinLSTM を提案します。さらに、SwinLSTMセルを核とした時空間予測ネットワークを構築します。 SwinLSTM は、独自のトリックを使用せずに、Moving MNIST、Human3.6m、TaxiBJ、および KTH データセットに対して最先端の手法を上回るパフォーマンスを発揮します。特に、ConvLSTM と比較して予測精度が大幅に向上しています。私たちの競合的な実験結果は、モデルが時空間依存関係を捉えるには、大域的な空間依存関係を学習する方が有利であることを示しています。私たちは、SwinLSTM が時空間予測精度の向上を促進するための強固なベースラインとして機能することを期待しています。コードは https://github.com/SongTang-x/SwinLSTM で公開されています。
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 https://github.com/SongTang-x/SwinLSTM.
updated: Sat Aug 19 2023 03:08:28 GMT+0000 (UTC)
published: Sat Aug 19 2023 03:08:28 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト