Biologically plausible heavy-tailed connectivity enhances generalizations on cognitive tasks in recurrent neural networks
Zhe Jiao, Xiaodong He, Shanglin Zhou
Abstract
While heavy-tailed synaptic weight distributions are pervasive in biological neural networks, their computational role---particularly in relation to generalization---remains poorly understood. To address this, we develop a novel optimal-transport-based optimization algorithm that incorporates key biological constraints, including Dale’s principle and heavy-tailed synaptic statistics, to train recurrent neural networks (RNNs) on a wide range of cognitive tasks. We show that these biologically constrained, heavy-tailed RNNs exhibit substantially improved generalization, which we further characterize within a PAC-Bayes framework. Our theoretical analysis and numerical experiments reveal two complementary mechanisms underlying this generalization enhancement. Topologically, heavy-tailed connectivity induces an effectively low-rank structure, which in turn yields low-dimensional neural dynamics. Geometrically, heavy-tailed connectivity intrinsically shapes task variable representations to lie near a linear manifold, thereby improving generalization for a linear readout strategy. Together, these results identify heavy-tailed connectivity as a biologically grounded intrinsic mechanism that promotes low-rank structure and favorable representational geometry, leading to improved generalization in flexible cognitive tasks.
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 368c96a9-c263-4750-a4db-deecffb420cbBuilds on4
- The Heavy-Tail Phenomenon in SGDMert Gürbüzbalaban, Umut Simsekli, Lingjiong ZhuICML 2021 · 165 citations
- Hausdorff Dimension, Heavy Tails, and Generalization in Neural NetworksUmut Simsekli, Ozan Sener, George Deligiannidis, Murat A. ErdogduNeurIPS 2020 · 79 citations
- Slow Transition to Low-Dimensional Chaos in Heavy-Tailed Recurrent Neural NetworksEva Xie, Stefan Mihalas, Lukasz KusmierzNeurIPS 2025 · 7 citations
- Discovering alternative solutions beyond the simplicity bias in recurrent neural networksWilliam Qian, Cengiz PehlevanICLR 2026 · 5 citations
Related papers
- The computational and learning benefits of Daleian neural networksAdam Haber, Elad SchneidmanNeurIPS 2022 · 10 citations
- A theory of weight distribution-constrained learningWeishun Zhong, Ben Sorscher, Daniel Lee, Haim SompolinskyNeurIPS 2022 · 5 citations
- Structured flexibility in recurrent neural networks via neuromodulationJulia Costacurta, Shaunak Bhandarkar, David M. Zoltowski, Scott W. LindermanNeurIPS 2024 · 21 citations
- Identifying Connectivity Distributions from Neural Dynamics Using FlowsTimothy Kim, Ulises Obilinovic, Yiliu Wang, Eric SheaBrown et al.ICML 2026 · 1 citation
- Beyond accuracy: generalization properties of bio-plausible temporal credit assignment rulesYuhan Helena Liu, Arna Ghosh, Blake A. Richards, Eric Shea-Brown et al.NeurIPS 2022 · 10 citations
