2020/06/06 by Axel Laborieux, Laborieux, Axel, Maxence Ernoult +9 · 16 citations
Computer Science · Engineering · Mathematics · Neuroscience · #Advanced Memory and Neural Computing #Algorithm #Artificial intelligence #Artificial neural network #Backpropagation #Computer science #Deep learning #Domain Adaptation and Few-Shot Learning #Estimator #Locality #MNIST database #Mathematics #Neural dynamics and brain function #Physics #Scalability #Scale (ratio) #Scaling #cs.NE
paper · pdf · doi:10.48550/arxiv.2006.03824
published in arXiv (Cornell University) (Cornell University)
arxiv created 2020/06/06 · openalex publication_date 2020/06/06 · arxiv updated 2020/06/09 · openalex created_date 2022/07/25 · openalex updated_date 2026/07/28
Equilibrium Propagation (EP) is a biologically-inspired algorithm for\nconvergent RNNs with a local learning rule that comes with strong theoretical\nguarantees. The parameter updates of the neural network during the credit\nassignment phase have been shown mathematically to approach the gradients\nprovided by Backpropagation Through Time (BPTT) when the network is\ninfinitesimally nudged toward its target. In practice, however, training a\nnetwork with the gradient estimates provided by EP does not scale to visual\ntasks harder than MNIST. In this work, we show that a bias in the gradient\nestimate of EP, inherent in the use of finite nudging, is responsible for this\nphenomenon and that cancelling it allows training deep ConvNets by EP. We show\nthat this bias can be greatly reduced by using symmetric nudging (a positive\nnudging and a negative one). We also generalize previous EP equations to the\ncase of cross-entropy loss (by opposition to squared error). As a result of\nthese advances, we are able to achieve a test error of 11.7% on CIFAR-10 by EP,\nwhich approaches the one achieved by BPTT and provides a major improvement with\nrespect to the standard EP approach with same-sign nudging that gives 86% test\nerror. We also apply these techniques to train an architecture with asymmetric\nforward and backward connections, yielding a 13.2% test error. These results\nhighlight EP as a compelling biologically-plausible approach to compute error\ngradients in deep neural networks.\n