跳到主要导航 跳到搜索 跳到主要内容

BiRe-ID: Binary Neural Network for Efficient Person Re-ID

  • Sheng Xu
  • , Chang Liu
  • , Baochang Zhang*
  • , Jinhu Lü
  • , Guodong Guo
  • , David Doermann
  • *此作品的通讯作者
  • Beihang University
  • Shenzhen Academy of Aerospace Technology
  • Baidu Inc
  • National Engineering Laboratory for Deep Learning Technology and Applications
  • SUNY Buffalo

科研成果: 期刊稿件文章同行评审

摘要

Person re-identification (Re-ID) has been promoted by the significant success of convolutional neural networks (CNNs). However, the application of such CNN-based Re-ID methods depends on the tremendous consumption of computation and memory resources, which affects its development on resource-limited devices such as next generation AI chips. As a result, CNN binarization has attracted increasing attention, which leads to binary neural networks (BNNs). In this article, we propose a new BNN-based framework for efficient person Re-ID (BiRe-ID). In this work, we discover that the significant performance drop of binarized models for Re-ID task is caused by the degraded representation capacity of kernels and features. To address the issues, we propose the kernel and feature refinement based on generative adversarial learning (KR-GAL and FR-GAL) to enhance the representation capacity of BNNs. We first introduce an adversarial attention mechanism to refine the binarized kernels based on their real-valued counterparts. Specifically, we introduce a scale factor to restore the scale of 1-bit convolution. And we employ an effective generative adversarial learning method to train the attention-aware scale factor. Furthermore, we introduce a self-supervised generative adversarial network to refine the low-level features using the corresponding high-level semantic information. Extensive experiments demonstrate that our BiRe-ID can be effectively implemented on various mainstream backbones for the Re-ID task. In terms of the performance, our BiRe-ID surpasses existing binarization methods by significant margins, at the level even comparable with the real-valued counterparts. For example, on Market-1501, BiRe-ID achieves 64.0% mAP on ResNet-18 backbone, with an impressive 12.51× speedup in theory and 11.75× storage saving. In particular, the KR-GAL and FR-GAL methods show strong generalization on multiple tasks such as Re-ID, image classification, object detection, and 3D point cloud processing.

源语言英语
期刊论文编号26
期刊ACM Transactions on Multimedia Computing, Communications and Applications
18
1s
DOI
出版状态已出版 - 2月 2022

学术指纹

探究 'BiRe-ID: Binary Neural Network for Efficient Person Re-ID' 的科研主题。它们共同构成独一无二的学术指纹。

引用此