Joint-Guided Distillation Binary Neural Network via Dynamic Channel-Wise Diversity Enhancement for Object Detection

被引:4
|
作者
Xie, Yefan [1 ,2 ]
Hou, Xuan [1 ,2 ]
Guo, Yanwei [1 ,2 ]
Wang, Xiuying [3 ]
Zheng, Jiangbin [1 ,2 ]
机构
[1] Northwestern Polytech Univ, Sch Comp Sci, Natl Engn Lab Integrated Aerosp Ground Ocean Big, Xian 710129, Shaanxi, Peoples R China
[2] Northwestern Polytech Univ, Shaanxi Prov Key Lab Speech Image Informat Proc, Xian 710129, Shaanxi, Peoples R China
[3] Univ Sydney, Sch Comp Sci, Sydney, NSW 2008, Australia
基金
中国国家自然科学基金;
关键词
Binary neural network; parameter quantization; object detection; knowledge distillation;
D O I
10.1109/TCSVT.2023.3286072
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Through truncating the weights and activations of a deep neural network, conventional binary quantization imposes limitations on the representation capability of the network parameters, which hence deteriorates the detection performance of the network. In this paper, a joint-guided distillation binary neural network via dynamic channel-wise diversity enhancement for object detection (JDBNet) is proposed to mitigate the gap of quantization errors. Our JDBNet includes a dynamic channel-wise diversity scheme and real-valued joint-guided teacher assistance to enhance the representation capability of the binary neural network in the object detection tasks. In the dynamic diversity scheme, the learning channel-wise bias (LCB) layer supports adjusting the magnitude of the parameters in which the sensitivity of the model parameters to the arbitrary quantization method is reduced, thereby improving the diversity expression ability of the feature parameters. In the joint-guided strategy, the single-precision implicit knowledge from the guiding teacher in the multilevel layer is utilized to supervise and penalize the quantitative model, enhancing the fitting performance of parameters in the binary quantized model. Extensive experiments on the PASCAL VOC, MS COCO, and VisDrone-DET datasets demonstrate that our JDBNet outperforms the state-of-the-art binary object detection networks in terms of mean Average Precision.
引用
收藏
页码:448 / 460
页数:13
相关论文
共 13 条
  • [1] DYNAMIC BINARY NEURAL NETWORK BY LEARNING CHANNEL-WISE THRESHOLDS
    Zhang, Jiehua
    Su, Zhuo
    Feng, Yanghe
    Lu, Xin
    Pietikainen, Matti
    Liu, Li
    2022 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2022, : 1885 - 1889
  • [2] Underwater image enhancement via a channel-wise transmission estimation network
    Wang, Qiang
    Fu, Bo
    Fan, Huijie
    IET IMAGE PROCESSING, 2023, 17 (10) : 2958 - 2971
  • [3] Stacked U-Shape Network With Channel-Wise Attention for Salient Object Detection
    Li, Junxia
    Pan, Zefeng
    Liu, Qingshan
    Wang, Ziyang
    IEEE TRANSACTIONS ON MULTIMEDIA, 2021, 23 : 1397 - 1409
  • [4] Multi-Level Context Aggregation Network with Channel-Wise Attention for Salient Object Detection
    Jia, Zihui
    Weng, Zhenyu
    Wan, Fang
    Zhu, Yuesheng
    IEEE Access, 2020, 8 : 102303 - 102312
  • [5] Multi-Level Context Aggregation Network With Channel-Wise Attention for Salient Object Detection
    Jia, Zihui
    Weng, Zhenyu
    Wan, Fang
    Zhu, Yuesheng
    IEEE ACCESS, 2020, 8 : 102303 - 102312
  • [6] SmokeNet: Satellite Smoke Scene Detection Using Convolutional Neural Network with Spatial and Channel-Wise Attention
    Ba, Rui
    Chen, Chen
    Yuan, Jing
    Song, Weiguo
    Lo, Siuming
    REMOTE SENSING, 2019, 11 (14)
  • [7] Real-time neural speech enhancement based on temporal refinement network and channel-wise gating methods
    Lee, Jinyoung
    Kang, Hong-Goo
    DIGITAL SIGNAL PROCESSING, 2023, 133
  • [8] Video-Based Deception Detection via Capsule Network With Channel-Wise Attention and Supervised Contrastive Learning
    Gao, Shuai
    Chen, Lin
    Fang, Yuancheng
    Xiao, Shengbing
    Li, Hui
    Yang, Xuezhi
    Song, Rencheng
    IEEE OPEN JOURNAL OF THE COMPUTER SOCIETY, 2024, 5 : 660 - 670
  • [9] Shape-Guided Detection: A joint network combining object detection and underwater image enhancement together
    Yang, Chao
    Jiang, Longyu
    Li, Zhicheng
    Wu, Jie
    ROBOTICS AND AUTONOMOUS SYSTEMS, 2024, 182
  • [10] Motor Imagery Classification Using Inter-Task Transfer Learning via a Channel-Wise Variational Autoencoder-Based Convolutional Neural Network
    Lee, Do-Yeun
    Jeong, Ji-Hoon
    Lee, Byeong-Hoo
    Lee, Seong-Whan
    IEEE TRANSACTIONS ON NEURAL SYSTEMS AND REHABILITATION ENGINEERING, 2022, 30 : 226 - 237