4 个月前

使用局部误差信号训练神经网络

使用局部误差信号训练神经网络

摘要

神经网络分类的监督训练通常使用全局损失函数进行。损失函数为输出层提供梯度,该梯度通过反向传播到隐藏层来指导权重的更新方向。另一种方法是使用逐层损失函数来训练网络。在本文中,我们首次证明了逐层训练可以在多种图像数据集上接近最先进的水平。我们使用单层子网络和两种不同的监督损失函数来生成隐藏层的局部误差信号,并展示了这些损失函数的组合有助于在局部学习背景下进行优化。利用局部误差可能是实现更符合生物学原理的深度学习的一个步骤,因为全局误差不需要再传输回隐藏层。一种完全不依赖反向传播的方法在追求更高生物学合理性的方法中表现优于之前报道的结果。代码可从以下地址获取:https://github.com/anokland/local-loss

代码仓库

anokland/local-loss
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-classification-on-cifar-10VGG11B(2x) + LocalLearning + CO
Percentage correct: 96.4
image-classification-on-cifar-100VGG11B(3x) + LocalLearning
Percentage correct: 79.9
image-classification-on-fashion-mnistVGG8B(2x) + LocalLearning + CO
Percentage error: 4.14
image-classification-on-kuzushiji-mnistVGG8B(2x) + LocalLearning + CO
Accuracy: 99.01
Error: 0.99
image-classification-on-mnistVGG8B + LocalLearning + CO
Percentage error: 0.26
image-classification-on-stl-10VGG8B + LocalLearning + CO
Percentage correct: 80.75
image-classification-on-svhnVGG8B + LocalLearning + CO
Percentage error: 1.65

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用局部误差信号训练神经网络 | 论文 | HyperAI超神经