On the Implicit Bias of Gradient Descent for Temporal Extrapolation
Not on the map. Why?
Abstract (arXiv)
When using recurrent neural networks (RNNs) it is common practice to apply trained models to sequences longer than those seen in training. This "extrapolating" usage deviates from the traditional statistical learning setup where guarantees are provided under the assumption that train and test distributions are identical. Here we set out to understand when RNNs can extrapolate, focusing on a simple case where the data generating distribution is memoryless. We first show that even with infinite training data, there exist RNN models that interpolate perfectly (i.e., they fit the training data) yet extrapolate poorly to longer sequences. We then show that if gradient descent is used for training, learning will converge to perfect extrapolation under certain assumptions on initialization. Our results complement recent studies on the implicit bias of gradient descent, showing that it plays a key role in extrapolation when learning temporal prediction models.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 14 of its 28 references that are in the map. First posted on arXiv 9 Feb 2022.
Builds on
14 of 28 references.
References · 14
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 302
- 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 259
- 2020How Neural Networks Extrapolate: From Feedforward to Graph Neural Networkscited by 218
- 2018Gradient descent aligns the layers of deep linear networkscited by 182
- 2018A Convergence Analysis of Gradient Descent for Deep Linear Neural Networkscited by 180
- 2020Neural Mechanics: Symmetry and Broken Conservation Laws in Deep Learning Dynamicscited by 93
2 more
- 2016Gradient Descent Learns Linear Dynamical Systemscited by 78
- 2021Implicit Bias of Linear RNNscited by 11
Built on by
Citing papers per year: 2022: 1; 2023: 1; 2024: 1; 2025: 3; 2026: 1.
7 papers.
- 2022Learning Low Dimensional State Spaces with Overparameterized Recurrent Neural Netscited by 7
- 2024Implicit Bias of Policy Gradient in Linear Quadratic Control: Extrapolation to Unseen Initial Statescited by 4
- 2025Regularization Implies balancedness in the deep linear networkcited by 4
- 2025On the Expressivity of Selective State-Space Layers: A Multivariate Polynomial Approachcited by 3
- 2025A Minimum Description Length Approach to Regularization in Neural Networkscited by 2
- 2023Deep Neural Networks Tend To Extrapolate Predictablycited by 1
- 2026Relative Positions Generalize, Absolute Positions Memorize: An Implicit-Bias Account of Length Generalization in Attentioncited by 0
- 2026Relative Positions Generalize, Absolute Positions Memorize: An Implicit-Bias Account of Length Generalization in Attention21 Jul 2026
- 2025Regularization Implies balancedness in the deep linear network3 Nov 2025
- 2025A Minimum Description Length Approach to Regularization in Neural Networks19 May 2025
- 2025On the Expressivity of Selective State-Space Layers: A Multivariate Polynomial Approach4 Feb 2025
- 2024Implicit Bias of Policy Gradient in Linear Quadratic Control: Extrapolation to Unseen Initial States12 Feb 2024
- 2023Deep Neural Networks Tend To Extrapolate Predictably2 Oct 2023
- 2022Learning Low Dimensional State Spaces with Overparameterized Recurrent Neural Nets25 Oct 2022