Minimal Convolutional RNNs Accelerate Spatiotemporal Learning
Not on the map. Why?
Abstract (arXiv)
We introduce MinConvLSTM and MinConvGRU, two novel spatiotemporal models that combine the spatial inductive biases of convolutional recurrent networks with the training efficiency of minimal, parallelizable RNNs. Our approach extends the log-domain prefix-sum formulation of MinLSTM and MinGRU to convolutional architectures, enabling fully parallel training while retaining localized spatial modeling. This eliminates the need for sequential hidden state updates during teacher forcing - a major bottleneck in conventional ConvRNN models. In addition, we incorporate an exponential gating mechanism inspired by the xLSTM architecture into the MinConvLSTM, which further simplifies the log-domain computation. Our models are structurally minimal and computationally efficient, with reduced parameter count and improved scalability. We evaluate our models on two spatiotemporal forecasting tasks: Navier-Stokes dynamics and real-world geopotential data. In terms of training speed, our architectures significantly outperform standard ConvLSTMs and ConvGRUs. Moreover, our models also achieve lower prediction errors in both domains, even in closed-loop autoregressive mode. These findings demonstrate that minimal recurrent structures, when combined with convolutional input aggregation, offer a compelling and efficient alternative for spatiotemporal sequence modeling, bridging the gap between recurrent simplicity and spatial complexity.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 28 references that are in the map. First posted on arXiv 5 Aug 2025.
Builds on
25 of 28 references.
References · 25
- 2017Attention Is All You Needcited by 60,361
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,057
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2014Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modelingcited by 2,367
- 2015Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcastingcited by 1,436
- 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
- 2022Simplified State Space Layers for Sequence Modelingcited by 788
- 2021Combining Recurrent, Convolutional, and Continuous-time Models with Linear State-Space Layerscited by 608
- 2020HiPPO: Recurrent Memory with Optimal Polynomial Projectionscited by 605
- 2023Resurrecting Recurrent Neural Networks for Long Sequencescited by 408
- 2024xLSTM: Extended Long Short-Term Memorycited by 327
13 more
- 2015LSTM: A Search Space Odysseycited by 316
- 2021Skillful Precipitation Nowcasting using Deep Generative Models of Radarcited by 311
- 2016Unsupervised Learning for Physical Interaction through Video Predictioncited by 260
- 2023WeatherBench 2: A benchmark for the next generation of data-driven global weather modelscited by 236
- 2024The Illusion of State in State-Space Modelscited by 187
- 2017Deep Learning for Precipitation Nowcasting: A Benchmark and A New Modelcited by 182
- 2015Delving Deeper into Convolutional Networks for Learning Video Representationscited by 148
- 2017Parallelizing Linear Recurrent Neural Nets Over Sequence Lengthcited by 127
- 2023Hierarchically Gated Recurrent Neural Network for Sequence Modelingcited by 102
- 2024Were RNNs All We Needed?cited by 38
- 2019A Distributed Neural Network Architecture for Robust Non-Linear Spatio-Temporal Predictioncited by 7
- 2023Efficient Parallelization of a Ubiquitous Sequential Computationcited by 5
- 2024Comparing and Contrasting DLWP Backbones on Navier-Stokes and Atmospheric Dynamicscited by 3
Built on by
Not cited yet.