Semi-supervised pedestrian re-identification via a teacher-student model with similarity-preserving generative adversarial networks

被引:3
作者
Zhao, Botong [1 ,2 ]
Wang, Yanjie [1 ,2 ]
Su, Keke [1 ,2 ]
Ren, Hong [1 ]
Han, Xiyu [1 ]
机构
[1] Chinese Acad Sci, Changchun Inst Opt Fine Mech & Phys, Changchun 130033, Peoples R China
[2] Univ Chinese Acad Sci, Beijing 100049, Peoples R China
关键词
Deep learning; Pedestrian re-identification; Semi-supervised learning;
D O I
10.1007/s10489-022-03218-8
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper describes a pedestrian re-identification algorithm, which was developed by integrating semi-supervised learning and similarity-preserving generative adversarial networks (SPGAN). The pedestrian re-identification task aimed to rapidly capture the same target using different cameras. Importantly, this process can be applied in the field of security. Because real-life environments are complex, the number of detected identities is uncertain, and the cost of manual labeling is high; therefore, it is difficult to apply the re-identification model based on supervised learning in real-life scenarios. To use the existing labeled dataset and a large amount of unlabeled data in the application environment, this report proposes a semi-supervised pedestrian re-identification model, which combines a teacher-student model with SPGAN. SPGAN was used to reduce the difference between the target domain and the source domain by transferring the style of the labeled dataset from the source domain. Additionally, the dataset from the source domain was used after the style transfer to pre-train the model; this enabled the model to adapt more rapidly to the target domain. The teacher-student model and the transformer model were then employed to generate soft pseudo-labels and hard pseudo-labels (via iterative training) and to update the parameters through distillation learning. Thus, it retained the learned features while adapting to the target domain. Experimental results indicated that the maps of the applied method on the Market-to-Duke, Duke-to-Market, Market-to-MSMT, and Duke-to-MSMT domains were 70.2, 79.3, 30.2, and 33.4, respectively.
引用
收藏
页码:1605 / 1618
页数:14
相关论文
共 33 条
[1]  
Chang XB, 2019, AAAI CONF ARTIF INTE, P3288
[2]   StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Translation [J].
Choi, Yunjey ;
Choi, Minje ;
Kim, Munyoung ;
Ha, Jung-Woo ;
Kim, Sunghun ;
Choo, Jaegul .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :8789-8797
[3]   Dual-Refinement: Joint Label and Feature Refinement for Unsupervised Domain Adaptive Person Re-Identification [J].
Dai, Yongxing ;
Liu, Jun ;
Bai, Yan ;
Tong, Zekun ;
Duan, Ling-Yu .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :7815-7829
[4]  
Delorme G, 2019, ARXIV 190401308
[5]   Image-Image Domain Adaptation with Preserved Self-Similarity and Domain-Dissimilarity for Person Re-identification [J].
Deng, Weijian ;
Zheng, Liang ;
Ye, Qixiang ;
Kang, Guoliang ;
Yang, Yi ;
Jiao, Jianbin .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :994-1003
[6]  
Ding GD, 2019, ARXIV 190601308
[7]  
Ding YH, 2019, ARXIV 190704194
[8]   Unsupervised Person Re-identification: Clustering and Fine-tuning [J].
Fan, Hehe ;
Zheng, Liang ;
Yan, Chenggang ;
Yang, Yi .
ACM TRANSACTIONS ON MULTIMEDIA COMPUTING COMMUNICATIONS AND APPLICATIONS, 2018, 14 (04)
[9]   Self-similarity Grouping: A Simple Unsupervised Cross Domain Adaptation Approach for Person Re-identification [J].
Fu, Yang ;
Wei, Yunchao ;
Wang, Guanshuo ;
Zhou, Yuqian ;
Shi, Honghui ;
Huang, Thomas S. .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :6111-6120
[10]  
Ge Y, 2020, ARXIV 200101526