Brain-like learning with exponentiated gradients
Jonathan H. CornfordiD, Roman PogodiniD, Arna GhoshiD, Kaiwen ShengiD, Brendan A. BicknelliD, Olivier CodoliD, Beverley A. ClarkiD, Guillaume LajoieiD, Blake A. RichardsiD
Abstract
Computational neuroscience relies on gradient descent (GD) for training artificial neural network (ANN) models of the brain. The advantage of GD is that it is effective at learning difficult tasks. However, it produces ANNs that are a poor phenomenological fit to biology, making them less relevant as models of the brain. Specifically, it violates Dales law, by allowing synapses to change from excitatory to inhibitory, and leads to synaptic weights that are not log-normally distributed, contradicting experimental data. Here, starting from first principles of optimisation theory, we present an alternative learning algorithm, exponentiated gradient (EG), that respects Dales Law and produces log-normal weights, without losing the power of learning with gradients. We also show that in biologically relevant settings EG outperforms GD, including learning from sparsely relevant signals and dealing with synaptic pruning. Altogether, our results show that EG is a superior learning algorithm for modelling the brain with ANNs.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 7
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Learning to live with Dale's principle: ANNs with separate excitatory and inhibitory units
Jonathan H. Cornford, Damjan Kalajdzievski, … Blake A. Richards·bioRxiv·Nov 2, 2020
- A rapid and efficient learning rule for biological neural circuits
Eren Sezener, Agnieszka Grabska‐Barwińska, … Peter E. Latham·bioRxiv·Mar 10, 2021
- Evolutionary learning in the brain by heterosynaptic plasticity
Zedong Bi, Guozhang Chen, … Liang Tian·bioRxiv·Dec 14, 2021
- Learning better with Dale's Law: A Spectral Perspective
Pingsheng Li, Jonathan H. Cornford, … Blake Richards·neural information processing systemsPeer-reviewed·Jan 1, 2023
- Learning to learn online with neuromodulated synaptic plasticity in spiking neural networks
Samuel Schmidgall, Joe Hays·bioRxiv·Jun 24, 2022
- A solution to the learning dilemma for recurrent networks of spiking neurons
Guillaume Bellec, Franz Scherr, … Wolfgang Maass·Nature CommunicationsPeer-reviewed·Jul 17, 2020