Share your thoughts, 1 month free Claude Pro on usSee more
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)--
894
Image ClassificationFashion MNIST (test)--
592
Image ClassificationSTL-10 (test)
Accuracy80.8
357
Image ClassificationFashion MNIST
Accuracy95.47
240
Image ClassificationCIFAR-100 standard (test)
Top-1 Accuracy55
141
Image ClassificationCIFAR-10 standard (test)
Accuracy81
81
Image ClassificationTinyImageNet (test)
Accuracy53.92
24
Image ClassificationKuzushiji-MNIST original (test)
Test Error Rate0.99
11
Image ClassificationFashion MNIST 66 (test)
Accuracy95.47
8
Showing 10 of 12 rows

Other info

Code

Follow for update