Global Credit Assignment via Dynamical Criticality
Wentao Wang, Keren Gao, Guozhang Chen
Abstract
Efficiently training recurrent neural networks on long sequences remains an open challenge. The standard global paradigm, backpropagation through time (BPTT), suffers from vanishing and exploding gradients and memory costs that scale linearly with sequence length. Conversely, biologically inspired local learning rules are memory-efficient but typically introduce severe bias. To bridge this gap, we introduce Criticality-driven Online Local Alignment (COLA). By leveraging the long-range spatiotemporal correlations inherent to the critical regime, COLA enables a strictly local learning rule to approximate global error propagation, thereby combining online efficiency with gradient descent precision. Theoretically, for a recurrent neural network with hidden units, COLA requires only an auxiliary state and constant activation memory, completely independent of sequence length. Empirically, COLA is competitive with BPTT on standard benchmarks and demonstrates superior robustness on stability-sensitive tasks. Finally, we conduct a rigorous analysis of the approximation error to provide a theoretical foundation for reliable online learning. Code is available at https://github.com/Criticality-Cognitive-Computation-Lab/COLA
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 65cc928f-8af4-4399-8d8b-eb1a01834e08Builds on3
- Training Recurrent Neural Networks via Forward Propagation Through TimeAnil Kag, Venkatesh SaligramaICML 2021 · 48 citations
- Practical Real Time Recurrent Learning with a Sparse ApproximationJacob Menick, Erich Elsen, Utku Evci, Simon Osindero et al.ICLR 2021 · 18 citations
- Feedback control guides credit assignment in recurrent neural networksKlara Kaleb, Barbara Feulner, Juan Gallego, Claudia ClopathNeurIPS 2024 · 5 citations
Related papers
- Dynamics and Representation Structure of Local Approximations to Gradient-Based Learning in Linear Recurrent Neural NetworksEzekiel Williams, Alexandre Payeur, Guillaume LajoieICML 2026
- RNNs Incrementally Evolving on an Equilibrium Manifold: A Panacea for Vanishing and Exploding Gradients?Anil Kag, Ziming Zhang, Venkatesh SaligramaICLR 2020 · 51 citations
- Training Recurrent Neural Networks Online by Learning Explicit State VariablesSomjit Nath, Vincent Liu, Alan Chan, Xin Li et al.ICLR 2020 · 9 citations
- Untangling tradeoffs between recurrence and self-attention in artificial neural networksGiancarlo Kerg, Bhargav Kanuparthi, Anirudh Goyal, Kyle Goyette et al.NeurIPS 2020 · 16 citations
- UnICORNN: A recurrent model for learning very long time dependenciesT. Konstantin Rusch, Siddhartha MishraICML 2021 · 76 citations
