Scalable Equilibrium Propagation via Intermediate Error Signals for Deep Convolutional CRNNs
By: Jiaqi Lin, Malyaban Bal, Abhronil Sengupta
Potential Business Impact:
Trains deep computer brains to learn much faster.
Equilibrium Propagation (EP) is a biologically inspired local learning rule first proposed for convergent recurrent neural networks (CRNNs), in which synaptic updates depend only on neuron states from two distinct phases. EP estimates gradients that closely align with those computed by Backpropagation Through Time (BPTT) while significantly reducing computational demands, positioning it as a potential candidate for on-chip training in neuromorphic architectures. However, prior studies on EP have been constrained to shallow architectures, as deeper networks suffer from the vanishing gradient problem, leading to convergence difficulties in both energy minimization and gradient computation. To address the vanishing gradient problem in deep EP networks, we propose a novel EP framework that incorporates intermediate error signals to enhance information flow and convergence of neuron dynamics. This is the first work to integrate knowledge distillation and local error signals into EP, enabling the training of significantly deeper architectures. Our proposed approach achieves state-of-the-art performance on the CIFAR-10 and CIFAR-100 datasets, showcasing its scalability on deep VGG architectures. These results represent a significant advancement in the scalability of EP, paving the way for its application in real-world systems.
Similar Papers
Equilibrium Propagation Without Limits
Machine Learning (CS)
Makes AI learn faster with bigger, bolder steps.
StochEP: Stochastic Equilibrium Propagation for Spiking Convergent Recurrent Neural Networks
Emerging Technologies
Trains brain-like computers to learn faster and better.
Toward Practical Equilibrium Propagation: Brain-inspired Recurrent Neural Network with Feedback Regulation and Residual Connections
Neural and Evolutionary Computing
Makes AI learn faster and cheaper, like a brain.