BDNN: Binary convolution neural networks for fast object detection

被引:20
作者
Peng, Hanyu [1 ,2 ]
Chen, Shifeng [1 ]
机构
[1] Chinese Acad Sci, Shenzhen Inst Adv Technol, Multimedia Lab, 1068 Xueyuan Ave, Shenzhen, Shenzhen Univer, Peoples R China
[2] Univ Chinese Acad Sci, Beijing, Peoples R China
基金
中国国家自然科学基金;
关键词
Deep learning; Object detection; Network compression; DESCRIPTOR;
D O I
10.1016/j.patrec.2019.03.026
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Object detection methods based on neural networks have made considerable progress. However, methods like Faster RCNN and SSD that adopt large neural networks as the base models. It's still a challenge to deploy such large detection networks in mobile or embedded devices. In this paper, we propose a low bit-width weight optimization approach to train Binary Neural Network for object detection using binary weights in training and testing. We introduce a greedy layer-wise method to train the detection network. This method boosts the performance instead of training the entire network at the same time. Our binary detection neural network (BDNN) can reduce the computational requirements and storage with competitive performance. For example, the binary network based on Faster RCNN with VGG16 can save 95% compression. In our experiments, BDNN achieves comparable performance with mAP 63.3% and outperforms SPPNet by 4.4% on PASCAL VOC 2007. (C) 2019 Elsevier B.V. All rights reserved.
引用
收藏
页码:91 / 97
页数:7
相关论文
共 33 条
[1]  
[Anonymous], P 3 INT C LEARNING R
[2]  
[Anonymous], 2017, ARXIV170401344
[3]  
[Anonymous], 2016, ARXIV161200212
[4]  
[Anonymous], 2015, CORR
[5]  
[Anonymous], 2013, CoRR abs/1308.3432
[6]  
[Anonymous], 2016, ABS160606160 ARXIV
[7]  
[Anonymous], 2015, ARXIV150404788
[8]  
[Anonymous], 2015, PROC CVPR IEEE
[9]  
[Anonymous], ADV NEURAL INFORM PR, DOI DOI 10.1109/TPAMI.2016.2577031
[10]  
[Anonymous], 2016, ARXIV160305279