Optimize Deep Convolutional Neural Network with Ternarized Weights and High Accuracy
IEEE Winter Conference on Applications of Computer Vision (WACV) 2019
https://github.com/elliothe/Ternarized_Neural_Network Pytorch
本文只对权重进行三值量化,每个卷积层和全连接层可以使用多个三值网络层表示以便提升三值网络的性能
III. W EIGHT T ERNARIZATION T RAINING M ETHOD
A. Ternarization with iterative statistical scaling
权重的三值量化:

其中 scaling factor 为

量化阈值为

dot-product operations 计算如下

模型的量化是一个迭代过程:
<

该文提出了一种针对深度卷积神经网络的权重三值量化训练方法,通过迭代统计缩放优化性能。在不牺牲准确性的情况下,使用多个三值网络层来表示全精度层,特别是对于卷积层和全连接层。实验中对比了不同训练策略,如直接训练三值权重、带有迭代缩放因子的训练以及微调,证明了这些方法的有效性。
最低0.47元/天 解锁文章
998

被折叠的 条评论
为什么被折叠?



