Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Training Neural Networks with Local Error Signals

About

Supervised training of neural networks for classification is typically performed with a global loss function. The loss function provides a gradient for the output layer, and this gradient is back-propagated to hidden layers to dictate an update direction for the weights. An alternative approach is to train the network with layer-wise loss functions. In this paper we demonstrate, for the first time, that layer-wise training can approach the state-of-the-art on a variety of image datasets. We use single-layer sub-networks and two different supervised loss functions to generate local error signals for the hidden layers, and we show that the combination of these losses help with optimization in the context of local learning. Using local errors could be a step towards more biologically plausible deep learning because the global error does not have to be transported back to hidden layers. A completely backprop free variant outperforms previously reported results among methods aiming for higher biological plausibility. Code is available https://github.com/anokland/local-loss

Arild N{\o}kland, Lars Hiller Eidnes• 2019

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)--
3518
Image ClassificationMNIST (test)--
882
Image ClassificationFashion MNIST (test)--
568
Image ClassificationSTL-10 (test)
Accuracy80.8
357
Image ClassificationFashion MNIST
Accuracy95.47
225
Image ClassificationTinyImageNet (test)
Accuracy53.92
24
Image ClassificationKuzushiji-MNIST original (test)
Test Error Rate0.99
11
Image ClassificationFashion MNIST 66 (test)
Accuracy95.47
8
Image ClassificationSTL-10 standard (test)
Test Error (%)19.25
7
Image ClassificationSVHN extra train (test)--
3
Showing 10 of 10 rows

Other info

Code

Follow for update