Skip to main content
Log in

Hierarchical pseudo-label learning for one-shot person re-identification

  • Published:
Applied Intelligence Aims and scope Submit manuscript

Abstract

The aim of one-shot person re-identification (person re-ID) is to recognize the target person from abundant pedestrians with only one labeled image of each person in training samples. The major problems faced by one-shot person re-ID algorithms are information loss and label noise. In this work, we propose a novel approach targeting one-shot person re-identification, planning to solve both the problems. It overcomes the information loss problem by introducing a hierarchical pseudo-label assignment strategy to fully exploit the unlabeled data information. Further, we also enhance the independence of two network branches by multiplying them by different weights for co-training in our asymmetric mutual teaching framework, which can generate soft labels to reduce hard pseudo-label noise. Extensive experiments on three benchmark datasets suggest that our proposed method significantly outperforms the existing state-of-the-art algorithm of mAP, respectively on Market-1501 and on DukeMTMC-reID, by 31.5 and 18.1. Our method can also be applied to the few-shots setting and achieve respectable performance.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3

Similar content being viewed by others

References

  1. Gheissari N, Sebastian T B, Hartley R (2006) Person reidentification using spatiotemporal appearance. In: IEEE Computer Society Conference on Computer Vision & Pattern Recognition(CVPR)

  2. Su C, Li J, Zhang S, Xing J, Gao W, Tian Q (2017) Pose-driven deep convolutional model for person re-identification. In: 2017 IEEE International Conference on Computer Vision (ICCV), pp 3980–3989

  3. Lin Y, Zheng L, Zheng Z, Wu Y, Yang Y (2019) Improving person re-identification by attribute and identity learning. arXiv:1703.07220

  4. Li W, Zhao R, Xiao T, Wang X (2014) Deepreid: Deep filter pairing neural network for person re-identification. In: 2014 IEEE Conference on Computer Vision and Pattern Recognition(CVPR), pp 152–159

  5. Liu X, Song M, Tao D, Zhou X, Chen C, Bu J (2014) Semi-supervised coupled dictionary learning for person re-identification. In: Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition(CVPR), pp 3550–3557

  6. Liu X, Song M, Tao D, Zhou X, Chen C, Bu J (2014) Semi-supervised coupled dictionary learning for person re-identification. In: Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition(CVPR), pp 3550–3557

  7. Qi L, Wang L, Huo J, Shi Y, Gao Y (2020) Progressive cross-camera soft-label learning for semi-supervised person re-identification. IEEE Trans Circ Syst Video Technol 30(9):2815–2829

    Article  Google Scholar 

  8. Bak S, Carr P (2017) One-shot metric learning for person re-identification. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). IEEE Computer Society, pp 1571–1580

  9. Wu Y, Lin Y, Dong X, Yan Y, Bian W, Yang Y (2019) Progressive learning for person re-identification with one example. IEEE Trans Image Process PP(6):1–1

    MathSciNet  MATH  Google Scholar 

  10. Xu T, Li J, Wu H, Yang H, Chen Y (2019) Feature space regularization for person re-identification with one sample. In: 2019 IEEE 31st International Conference on Tools with Artificial Intelligence (ICTAI)

  11. Li H, Xiao J, Sun M, Lim E G, Zhao Y (2021) Progressive sample mining and representation learning for one-shot person re-identification. Pattern Recogn 110:107614

    Article  Google Scholar 

  12. Ge Y, Chen D, Li H (2020) Mutual mean-teaching: Pseudo label refinery for unsupervised domain adaptation on person re-identification. In: International Conference on Learning Representations(ICLR)

  13. Santoro A, Bartunov S, Botvinick M, Wierstra D, Lillicrap T (2016) One-shot Learning with Memory-Augmented Neural Networks. arXiv:1605.06065

  14. Vinyals O, Blundell C, Lillicrap T, Wierstra D et al (2016) Matching networks for one shot learning. Adv Neural Inf Process Syst 29:3630–3638

    Google Scholar 

  15. Ravi S, Larochelle H (2017) Optimization as a model for few-shot learning. International Conference on Learning Representations. IEEE, Toulon

  16. Sun Q, Li X, Li L, Liu X, Liu F, Jiao L (2019) Semi-supervised complex-valued gan for polarimetric sar image classification. In: IGARSS 2019 - 2019 IEEE International Geoscience and Remote Sensing Symposium, pp 3245–3248

  17. Vapnik V, Sterin A (1977) On structural risk minimization or overall risk in a problem of pattern recognition. Autom Remote Control 10(3):1495–1503

    MATH  Google Scholar 

  18. Iscen A, Tolias G, Avrithis Y, Chum O (2020) Label propagation for deep semi-supervised learning. In: 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)

  19. Blum A, Mitchell T (1998) Combining labeled and unlabeled data with co-training. In: Proceedings of the 11th Annual Conference on Computational Learning Theory

  20. Xian Y, Hu H (2018) Enhanced multi-dataset transfer learning method for unsupervised person re-identification using co-training strategy. IET Comput Vis 12(8):1219–1227

    Article  Google Scholar 

  21. Lin Y, Dong X, Zheng L, Yan Y, Yang Y (2019) A bottom-up clustering approach to unsupervised person re-identification. Proc AAAI Conf Artif Intell 33:8738–8745

    Google Scholar 

  22. Yu H X, Zheng W S, Wu A, Guo X, Lai J H (2019) Unsupervised person re-identification by soft multilabel learning. In: 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)

  23. Wang J, Zhu X, Gong S, Li W (2018) Transferable joint attribute-identity deep learning for unsupervised person re-identification. In: 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR), pp 2275–2284

  24. Wu Y, Lin Y, Dong X, Yan Y, Yang Y (2018) Exploit the unknown gradually: One-shot video-based person re-identification by stepwise learning. In: 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)

  25. Li Z, Ko B, Choi H (2018) Pseudo-labeling using gaussian process for semi-supervised deep learning. In: 2018 IEEE International Conference on Big Data and Smart Computing (BigComp). IEEE Computer Society, pp 263–269

  26. Hu Q, Wu J, Cheng J, Wu L, Lu H (2017) Pseudo label based unsupervised deep discriminative hashing for image retrieval. In: Proceedings of the 25th ACM International Conference on Multimedia, pp 1584–1590

  27. Fan H, Zheng L, Yan C, Yang Y (2019) Unsupervised person re-identification: Clustering and fine-tuning. ACM Trans Multimed Comput Commun Appl 14

  28. Zhong Z, Zheng L, Luo Z, Li S, Yang Y (2020) Invariance matters: Exemplar memory for domain adaptive person re-identification. In: 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)

  29. Zheng L, Shen L, Tian L, Wang S, Tian Q (2015) Scalable person re-identification: A benchmark. In: 2015 IEEE International Conference on Computer Vision (ICCV)

  30. Ristani E, Solera F, Zou R, Cucchiara R, Tomasi C (2016) Performance measures and a data set for multi-target, multi-camera tracking. In: European Computer on Computer Vision Workshops(ECCV)

  31. Zhong Z, Zheng L, Cao D, Li S (2017) Re-ranking person re-identification with k-reciprocal encoding. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp 3652–3661

  32. He K, Zhang X, Ren S, Sun J (2016) Deep residual learning for image recognition. In: 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). IEEE Computer Society, pp 770–778

  33. Chang X, Ma Z, Wei X, Hong X, Gong Y (2020) Transductive semi-supervised metric learning for person re-identification. Pattern Recogn:107569

  34. Xin X, Wang J, Xie R, Zhou S, Huang W, Zheng N (2019) Semi-supervised person re-identification using multi-view clustering. Pattern Recogn 88:285–297

    Article  Google Scholar 

Download references

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Xiaoyu Ma.

Additional information

Publisher’s note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Shao, J., Ma, X. Hierarchical pseudo-label learning for one-shot person re-identification. Appl Intell 52, 9225–9238 (2022). https://doi.org/10.1007/s10489-021-02959-2

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10489-021-02959-2

Keywords

Navigation