Abstract
This paper proposes a mixed sample contrastive learning framework that constructs memory dictionary with discrete samples and reliable clusters for unsupervised vehicle re-identification. Firstly, we introduce a discrete sample separation (DSS) module including a discrete sample criterion and a discrete sample separation operation. Specifically, for a specific feature cluster, the discrete sample criterion drives the DSS to mine the discrete sample. Such that the original cluster can be separated into a more reliable cluster and discrete samples. Furthermore, a mixed sample contrastive learning (MSCL) strategy is designed to construct a mixed sample memory dictionary for training the model with more superior learning target. Moreover, a discrete sample loss (DSL) is proposed to calculate the contrastive loss of the model and dynamically update the memory dictionary during training. Extensive experiments show our method performs favorably against state of the arts. The code will be published on github after the paper is accepted.
Similar content being viewed by others
References
Zhai, X., Oliver, A., Kolesnikov, A., Beyer, L.: S4l: self-supervised semi-supervised learning. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 1476–1485 (2019)
He, K., Fan, H., Wu, Y., Xie, S., Girshick, R.: Momentum contrast for unsupervised visual representation learning. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 9729–9738 (2020)
Chen, T., Kornblith, S., Norouzi, M., Hinton, G.: A simple framework for contrastive learning of visual representations. In: International Conference on Machine Learning. PMLR, pp. 1597–1607 (2020)
Wang, Y., Zeng, D.: Deep domain adaptation on vehicle re-identification. In: 2019 IEEE Fifth International Conference on Multimedia Big Data (BigMM), pp. 416–420. IEEE (2019)
Qi, L., Wang, L., Huo, J., Zhou, L., Shi, Y., Gao, Y.: A novel unsupervised camera-aware domain adaptation framework for person re-identification. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 8080–8089 (2019)
Mekhazni, D., Bhuiyan, A., Ekladious, G., Granger, E.: Unsupervised domain adaptation in the dissimilarity space for person re-identification. In: European Conference on Computer Vision, pp. 159–174. Springer (2020)
Chen, G., Lu, Y., Lu, J., Zhou, J.: Deep credible metric learning for unsupervised domain adaptation person re-identification. In: Computer Vision–ECCV 2020: 16th European Conference, Glasgow, UK, August 23–28, 2020, Proceedings, Part VIII 16, pp. 643–659. Springer (2020)
Chen, X., Fan, H., Girshick, R., He, K.: Improved baselines with momentum contrastive learning. arXiv preprint arXiv:2003.04297 (2020)
Chen, X., He, K.: Exploring simple siamese representation learning. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 15,750–15,758 (2021)
Ester, M., Kriegel, H.-P., Sander, J., Xu X., et al.: A density-based algorithm for discovering clusters in large spatial databases with noise. In: kdd, vol. 96, no. 34, pp. 226–231 (1996)
MacQueen, J., et al.: Some methods for classification and analysis of multivariate observations. In: Proceedings of the Fifth Berkeley Symposium on Mathematical Statistics and Probability, vol. 1, no. 14, pp. 281–297. Oakland, CA, USA (1967)
Oord, A.V.D., Li, Y., Vinyals, O.: Representation Learning with Contrastive Predictive Coding. arXiv preprint arXiv:1807.03748 (2018)
Wang, D., Zhang, S.: Unsupervised person re-identification via multi-label classification. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 10,981–10,990 (2020)
Dai, Z., Wang, G., Yuan, W., Zhu, S., Tan, P.: Cluster contrast for unsupervised person re-identification. arXiv preprint arXiv:2103.11568 (2021)
Liu, X., Liu, W., Mei, T., Ma, H.: A deep learning-based approach to progressive vehicle re-identification for urban surveillance. In: European Conference on Computer Vision, pp. 869–884. Springer (2016)
Liu, H., Tian, Y., Yang, Y., Pang, L., Huang, T.: Deep relative distance learning: Tell the difference between similar vehicles. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2167–2175 (2016)
Lou, Y., Bai, Y., Liu, J., Wang, S., Duan, L.: Veri-wild: a large dataset and a new method for vehicle re-identification in the wild. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 3235–3243 (2019)
Wang, J., Zhu, X., Gong, S., Li, W.: Transferable joint attribute-identity deep learning for unsupervised person re-identification. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2275–2284 (2018)
Shan,L., Li, H., Li, C.T., Kot, A.C.: Multi-task mid-level feature alignment network for unsupervised cross-dataset person re-identification. In: BMVC 2018 (2018)
Wei, L., Zhang, S., Gao, W., Tian, Q.: Person transfer gan to bridge domain gap for person re-identification. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 79–88 (2018)
Zhong, Z., Zheng, L., Li, S., Yang, Y.: Generalizing a person retrieval model hetero-and homogeneously. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 172–188 (2018)
Song, L., Wang, C., Zhang, L., Du, B., Zhang, Q., Huang, C., Wang, X.: Unsupervised domain adaptive re-identification: theory and practice. Pattern Recogn. 102, 107173 (2020)
Yu, H.-X., Zheng, W.-S., Wu, A., Guo, X., Gong, S., Lai, J.-H.: Unsupervised person re-identification by soft multilabel learning. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 2148–2157 (2019)
Chen, G., Lu, Y., Lu, J., Zhou, J.: Deep credible metric learning for unsupervised domain adaptation person re-identification. In: Computer Vision-ECCV 2020: 16th European Conference, Glasgow, UK, August 23–28, 2020, Proceedings, Part VIII 16, pp. 643–659. Springer (2020)
Ge, Y., Chen, D., Li, H.: Mutual mean-teaching: pseudo label refinery for unsupervised domain adaptation on person re-identification. arXiv preprint arXiv:2001.01526 (2020)
Lin, Y., Dong, X., Zheng, L., Yan, Y., Yang, Y.: A bottom-up clustering approach to unsupervised person re-identification. Proc. AAAI Conf. Artif. Intell. 33(01), 8738–8745 (2019)
Wang, M., Lai, B., Huang, J., Gong, X., Hua, X.-S.: Camera-aware proxies for unsupervised person re-identification. arXiv preprint arXiv:2012.10674 (2020)
Li, J., Zhang, S.: Joint visual and temporal consistency for unsupervised domain adaptive person re-identification. In: European Conference on Computer Vision, pp. 483–499. Springer (2020)
Wang, Z., Zhang, J., Zheng, L., Liu, Y., Sun, Y., Li, Y., Wang, S.: Cycas: self-supervised cycle association for learning re-identifiable descriptions. In: Computer Vision–ECCV 2020: 16th European Conference, Glasgow, UK, August 23–28, 2020, Proceedings, Part XI 16, pp. 72–88. Springer (2020)
Ge, Y., Zhu, F., Chen, D., Zhao, R., Li, H.: Self-paced contrastive learning with hybrid memory for domain adaptive object re-id. arXiv preprint arXiv:2006.02713 (2020)
Wu, Z., Xiong, Y., Yu, S.X., Lin, D.: Unsupervised feature learning via non-parametric instance discrimination. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3733–3742 (2018)
Grill, J.-B., Strub, F., Altché, F., Tallec, C., Richemond, P.H., Buchatskaya, E., Doersch, C., Pires, B.A., Guo, Z.D., Azar M.G., et al.: Bootstrap your own latent: a new approach to self-supervised learning. arXiv preprint arXiv:2006.07733 (2020)
Bachman,P., Hjelm, R.D., Buchwalter, W.: Learning representations by maximizing mutual information across views. arXiv preprint arXiv:1906.00910 (2019)
Henaff, O.: Data-efficient image recognition with contrastive predictive coding. In: International Conference on Machine Learning. PMLR, pp. 4182–4192 (2020)
Hjelm, R.D., Fedorov, A., Lavoie-Marchildon, S., Grewal, K., Bachman, P., Trischler, A., Bengio, Y.: Learning deep representations by mutual information estimation and maximization. arXiv preprint arXiv:1808.06670 (2018)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. in: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)
Ester, M., Kriegel, H.-P., Sander, J., Xu, X.: Density-based spatial clustering of applications with noise. Int. Conf. Knowl. Discov. Data Mining 240, 6 (1996)
Zheng, A., Sun, X., Li, C., Tang, J.: Aware progressive clustering for unsupervised vehicle re-identification. IEEE Trans. Intell. Transp. Syst. (2021)
Peng, J., Wang, H., Xu, F., Fu, X.: Cross domain knowledge learning with dual-branch adversarial network for vehicle re-identification. Neurocomputing 401, 133–144 (2020)
Yi, D., Lei, Z., Liao, S., Li, S.Z.: Deep metric learning for person re-identification. In: 2014 22nd International Conference on Pattern Recognition, pp. 34–39. IEEE (2014)
Cai, J., Deng, J., Aftab, M.U., Khokhar, M.S., Kumar, R., et al.: Efficient and deep vehicle re-identification using multi-level feature extraction. Appl. Sci. 9(7), 1291 (2019)
Deng, W., Zheng, L., Ye, Q., Kang, G., Yang, Y., Jiao, J.: “Image-image domain adaptation with preserved self-similarity and domain-dissimilarity for person re-identification,” in: Proceedings of the IEEE conference on computer vision and pattern recognition, pp. 994–1003 ( 2018)
Ding, Y., Fan, H., Xu, M., Yang, Y.: Adaptive exploration for unsupervised person re-identification. ACM Trans. Multimedia Comput. Commun. Appl. (TOMM) 16(1), 1–19 (2020)
Yu, J., Oh, H.: Unsupervised vehicle re-identification via self-supervised metric learning using feature dictionary. arXiv preprint arXiv:2103.02250 (2021)
Acknowledgements
This work is supported by National Nature Science Foundation of China (Grant No. 61871106 and No. 61370152), Key R& D projects of Liaoning Province, China (Grant No. 2020JH2/10100029), and the Open Project Program Foundation of the Key Laboratory of Opto-Electronics Information Processing, Chinese Academy of Sciences (OEIP-O-202002).
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Wang, Y., Wei, Y., Ma, R. et al. Unsupervised vehicle re-identification based on mixed sample contrastive learning. SIViP 16, 2083–2091 (2022). https://doi.org/10.1007/s11760-022-02170-x
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11760-022-02170-x