二值网络:训练权重和激活值限制在+1或-1的神经网络

该论文提出了一种训练二值神经网络(BNN)的方法,其中权重和激活值限于+1或-1。在MNIST,CIFAR-10和SVHN数据集上的实验表明,BNN在减少内存消耗和提高能效的同时,保持了高精度。对比确定性和随机二值化,研究主要采用确定性二值化,但训练过程中的激活可能会使用随机二值化。BNN的梯度计算通过二值权重和激活进行,并使用实数权重的SGD进行更新。尽管sign函数的导数问题,但依然能够通过特定方法传播梯度。
摘要由CSDN通过智能技术生成

Binarized Neural Networks: Training Neural Networks with Weights and Activations Constrained to +1 or -1

介绍

这篇论文主要作了以下改进:
1.引入一种训练二值神经网络的方法。在运行时,以及在训练的过程中计算参数梯度时,神经网络具有二值的权重和激活值。
2.做了两套实验,每一套在不同的框架下完成,分别是Torch7以及Theano,他们在基于MNIST ,CIFAR-10,SVHN等数据集上训练BNN时都几乎实现了最佳的结果。
3.在前传的时候(不管是在运行还是在推断的时候),BNN极大地减少了内存的消耗(大小以及访存数),把大多数的算术操作以bit位运算代替,这极大地增加了能效。精心设计的硬件能够减少60%的时间复杂度
4.MNIST BNN的速度是未经优化的GPU核的7倍,而分类精确度却不会损失。

1.1 Deterministic vs Stochastic Binarization 确定二值与随机二值

Deterministic Binarization
在这里插入图片描述
Stochastic Binarization
在这里插入图片描述
随机二值化看起来好&#

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值