On the Stability and Scalability of Node Perturbation LearningDownload PDF

Published: 31 Oct 2022, 18:00, Last Modified: 12 Oct 2022, 17:53NeurIPS 2022 AcceptReaders: Everyone
Keywords: Biologically Plausible Deep Networks, Credit Assignment, Neuroscience, Dynamical Systems
TL;DR: Node perturbation is scalable against overparameterization but unstable in the presence of model mismatch.
Abstract: To survive, animals must adapt synaptic weights based on external stimuli and rewards. And they must do so using local, biologically plausible, learning rules -- a highly nontrivial constraint. One possible approach is to perturb neural activity (or use intrinsic, ongoing noise to perturb it), determine whether performance increases or decreases, and use that information to adjust the weights. This algorithm -- known as node perturbation -- has been shown to work on simple problems, but little is known about either its stability or its scalability with respect to network size. We investigate these issues both analytically, in deep linear networks, and numerically, in deep nonlinear ones. We show analytically that in deep linear networks with one hidden layer, both learning time and performance depend very weakly on hidden layer size. However, unlike stochastic gradient descent, when there is model mismatch between the student and teacher networks, node perturbation is always unstable. The instability is triggered by weight diffusion, which eventually leads to very large weights. This instability can be suppressed by weight normalization, at the cost of bias in the learning rule. We confirm numerically that a similar instability, and to a lesser extent scalability, exist in deep nonlinear networks trained on both a motor control task and image classification tasks. Our study highlights the limitations and potential of node perturbation as a biologically plausible learning rule in the brain.
Supplementary Material: pdf
25 Replies

Loading