Clipping-Based Post Training 8-Bit Quantization of Convolution Neural Networks for Object Detection

被引:2
|
作者
Chen, Leisheng [1 ]
Lou, Peihuang [1 ]
机构
[1] Nanjing Univ Aeronaut & Astronaut, Coll Mech & Elect Engn, Nanjing 210016, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2022年 / 12卷 / 23期
关键词
object detection; quantization; clipping; post-training quantization; accuracy loss;
D O I
10.3390/app122312405
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Fueled by the development of deep neural networks, breakthroughs have been achieved in plenty of computer vision problems, such as image classification, segmentation, and object detection. These models usually have handers and millions of parameters, which makes them both computational and memory expensive. Motivated by this, this paper proposes a post-training quantization method based on the clipping operation for neural network compression. By quantizing parameters of a model to 8-bit using our proposed methods, its memory consumption is reduced, its computational speed is increased, and its performance is maintained. This method exploits the clipping operation during training so that it saves a large computational cost during quantization. After training, this method quantizes the parameters to 8-bit based on the clipping value. In addition, a fully connected layer compression is conducted using singular value decomposition (SVD), and a novel loss function term is leveraged to further diminish the performance drop caused by quantization. The proposed method is validated on two widely used models, Yolo V3 and Faster R-CNN, for object detection on the PASCAL VOC, COCO, and ImageNet datasets. Performances show it effectively reduces the storage consumption at 18.84% and accelerates the model at 381%, meanwhile avoiding the performance drop (drop < 0.02% in VOC).
引用
收藏
页数:18
相关论文
共 50 条
  • [31] Effective Post-Training Quantization Of Neural Networks For Inference on Low Power Neural Accelerator
    Demidovskij, Alexander
    Smirnov, Eugene
    2020 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2020,
  • [32] Atrial Fibrillation Detection in Spectrogram Based on Convolution Neural Networks
    Guo, Jing-Ming
    Yang, Chiao-Chun
    Wang, Zong-Hui
    Hsia, Chih-Hsien
    Chang, Li-Ying
    2019 INTERNATIONAL SYMPOSIUM ON INTELLIGENT SIGNAL PROCESSING AND COMMUNICATION SYSTEMS (ISPACS), 2019,
  • [33] Optimization-Based Post-Training Quantization With Bit-Split and Stitching
    Wang, Peisong
    Chen, Weihan
    He, Xiangyu
    Chen, Qiang
    Liu, Qingshan
    Cheng, Jian
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (02) : 2119 - 2135
  • [34] Detection for Cutting Tool Wear Based on Convolution Neural Networks
    Wang, Yue
    Dai, Wei
    Xiao, Jianglin
    12TH INTERNATIONAL CONFERENCE ON RELIABILITY, MAINTAINABILITY, AND SAFETY (ICRMS 2018), 2018, : 297 - 300
  • [35] AN IMPROVED OBJECT DETECTION METHOD BASED ON DEEP CONVOLUTION NEURAL NETWORK FOR SMOKE DETECTION
    Zeng, Junying
    Lin, Zuoyong
    Qi, Chuanbo
    Zhao, Xiaoxiao
    Wang, Fan
    PROCEEDINGS OF 2018 INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND CYBERNETICS (ICMLC), VOL 1, 2018, : 184 - 189
  • [36] Speeding up inference on deep neural networks for object detection by performing partial convolution
    Wattanapong Kurdthongmee
    Journal of Real-Time Image Processing, 2020, 17 : 1487 - 1503
  • [38] A Method to Automatic Create Dataset for Training Object Detection Neural Networks
    Zhou, Shi
    Yang, Zijun
    Zhu, Miaomiao
    Li, He
    Serikawa, Seiichi
    Mizumachi, Mitsunori
    Zhang, Lifeng
    IEEE Access, 2022, 10 : 80505 - 80517
  • [39] A Method to Automatic Create Dataset for Training Object Detection Neural Networks
    Zhou, Shi
    Yang, Zijun
    Zhu, Miaomiao
    Li, He L.
    Serikawa, Seiichi
    Mizumachi, Mitsunori
    Zhang, Lifeng
    IEEE ACCESS, 2022, 10 : 80505 - 80517
  • [40] Object Detection and Tracking based on Recurrent Neural Networks
    Zhang, Yashu
    Ming, Yue
    Zhang, Runqing
    PROCEEDINGS OF 2018 14TH IEEE INTERNATIONAL CONFERENCE ON SIGNAL PROCESSING (ICSP), 2018, : 338 - 343