中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Margin-aware binarized weight networks for image classification

文献类型:会议论文

作者Xu, Ting-Bing1,2; Yang, Peipei1,2; Zhang, Xu-Yao1,2; Liu, Cheng-Lin1,2
出版日期2017
会议日期13-15 Sep.
会议地点Shanghai, China
关键词Deep Network Compression Binarized Weight Networks Binary-l2 Regularization
英文摘要

Deep neural networks (DNNs) have achieved remarkable successes in many vision tasks. However, due to the dependence on large memory and high-performance GPUs, it is extremely hard to deploy DNNs on low-power devices. For compressing and accelerating deep neural networks, many techniques have been proposed recently. Particularly, binarized weight networks, which store one weight using only one bit and replace complex floating operations with simple calculations, are attractive from the perspective of hardware implementation. In this paper, we propose a simple strategy to learn better binarized weight networks. Motivated by the phenomenon that the stochastic binarization approach usually converges with real-valued weights close to two boundaries {−1, +1} and gives better performance than deterministic binarization, we construct a margin-aware binarization strategy by adding a weight constraint into the objective function of deterministic scheme to minimize the margins between real-valued weights and boundaries. This constraint can be easily realized by a Binary-L2 regularization without suffering from the complex random number generation. Experimental results on MNIST and CIFAR-10 datasets show that the proposed method yields better performance than recent network binarization schemes and the full precision network counterpart. 

会议录Proc. International Conference on Image and Graphics (ICIG 2017)
源URL[http://ir.ia.ac.cn/handle/173211/19984]  
专题自动化研究所_模式识别国家重点实验室_模式分析与学习团队
通讯作者Xu, Ting-Bing
作者单位1.National Laboratory of Pattern Recognition, Institute of Automation of Chinese Academy of Sciences, Beijing, China
2.University of Chinese Academy of Sciences (UCAS), Beijing, China
推荐引用方式
GB/T 7714
Xu, Ting-Bing,Yang, Peipei,Zhang, Xu-Yao,et al. Margin-aware binarized weight networks for image classification[C]. 见:. Shanghai, China. 13-15 Sep..

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。