ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
712
Citations
88
Influential Citations
Frontiers in Computational Neuroscience
Venue
2017
Year
We introduce Equilibrium Propagation, a learning framework for energy-based models. It involves only one kind of neural computation, performed in both the first phase (when the prediction is made) and the second phase of training (after the target or prediction error is revealed). Although this algorithm computes the gradient of an objective function just like Backpropagation, it does not need a special computation or circuit for the second phase, where errors are implicitly propagated. Equilibrium Propagation shares similarities with Contrastive Hebbian Learning and Contrastive Divergence while solving the theoretical issues of both algorithms: our algorithm computes the gradient of a well-defined objective function. Because the objective function is defined in terms of local perturbations, the second phase of Equilibrium Propagation corresponds to only nudging the prediction (fixed point or stationary distribution) toward a configuration that reduces prediction error. In the case of a recurrent multi-layer supervised network, the output units are slightly nudged toward their target in the second phase, and the perturbation introduced at the output layer propagates backward in the hidden layers. We show that the signal "back-propagated" during this second phase corresponds to the propagation of error derivatives and encodes the gradient of the objective function, when the synaptic update corresponds to a standard form of spike-timing dependent plasticity. This work makes it more plausible that a mechanism similar to Backpropagation could be implemented by brains, since leaky integrator neural computation performs both inference and error back-propagation in our model. The only local difference between the two phases is whether synaptic changes are allowed or not. We also show experimentally that multi-layer recurrently connected networks with 1, 2, and 3 hidden layers can be trained by Equilibrium Propagation on the permutation-invariant MNIST task.
Equilibrium Propagation (EqProp) addresses a fundamental question in both machine learning and neuroscience: how can the brain implement a learning algorithm as effective as backpropagation without a dedicated error-propagation circuit? Backpropagation, the workhorse of modern deep learning, requires a separate backward pass that computes error derivatives using the transpose of forward weights—a mechanism widely considered biologically implausible. EqProp offers a compelling alternative by showing that a single type of neural computation, performed in two phases, can compute the gradient of a well-defined objective function.
The paper bridges the gap between energy-based models and backpropagation, two frameworks that have historically been treated separately. Energy-based models provide a principled probabilistic framework but often rely on contrastive divergence, which lacks a well-defined objective. EqProp solves this by defining an objective in terms of local perturbations, ensuring that the learning rule performs gradient descent on a valid cost function. This theoretical clarity is a significant advance over contrastive Hebbian learning and contrastive divergence.
Moreover, EqProp makes backpropagation more plausible as a brain mechanism. The only difference between the inference and training phases is whether synaptic changes are allowed, and the algorithm can be implemented with leaky integrator neurons and spike-timing dependent plasticity. This has profound implications for our understanding of credit assignment in biological neural networks.
The paper demonstrates that multi-layer recurrently connected networks with 1, 2, and 3 hidden layers can be trained using EqProp on the permutation-invariant MNIST task. While the abstract does not provide specific accuracy numbers, the successful training of such networks validates the algorithm's practical viability. The authors show that the signal back-propagated during the second phase corresponds to the propagation of error derivatives, confirming the theoretical analysis.
EqProp represents a significant step toward unifying energy-based models with backpropagation and offers a biologically plausible mechanism for credit assignment. By showing that a single neural computation can perform both inference and error back-propagation, it challenges the notion that backpropagation requires a specialized circuit. This work has inspired subsequent research in biologically plausible deep learning and neuromorphic computing, and it provides a foundation for developing learning algorithms that could be implemented in energy-efficient hardware or biological systems. Its impact extends to both theoretical neuroscience and practical machine learning, where energy-based models continue to be relevant.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba