Revisiting Multi-Granularity Representation via Group Contrastive Learning for Unsupervised Vehicle Re-identification

Zhigang Chang,Shibao Zheng
2024-10-29
Abstract:Vehicle re-identification (Vehicle ReID) aims at retrieving vehicle images across disjoint surveillance camera views. The majority of vehicle ReID research is heavily reliant upon supervisory labels from specific human-collected datasets for training. When applied to the large-scale real-world scenario, these models will experience dreadful performance declines due to the notable domain discrepancy between the source dataset and the target. To address this challenge, in this paper, we propose an unsupervised vehicle ReID framework (MGR-GCL). It integrates a multi-granularity CNN representation for learning discriminative transferable features and a contrastive learning module responsible for efficient domain adaptation in the unlabeled target domain. Specifically, after training the proposed Multi-Granularity Representation (MGR) on the labeled source dataset, we propose a group contrastive learning module (GCL) to generate pseudo labels for the target dataset, facilitating the domain adaptation process. We conducted extensive experiments and the results demonstrated our superiority against existing state-of-the-art methods.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is **the domain adaptation problem in unsupervised vehicle re - identification (Vehicle Re - ID)**. Specifically, when existing vehicle re - identification models are applied in large - scale real - world scenarios, due to the significant domain discrepancy between the source domain and the target domain, their performance will drop substantially. To address this challenge, the authors propose a new unsupervised vehicle re - identification framework, named **MGR - GCL** (Multi - Granularity Representation via Group Contrastive Learning). ### Specific manifestations of the problem 1. **Domain discrepancy**: Existing methods rely on specific manually - annotated datasets for training. When these models are applied to large - scale real - world scenarios, due to the domain discrepancy between the source domain and the target domain, their performance will drop significantly. 2. **Under - utilization of target - domain data**: Although traditional unsupervised domain adaptation methods can alleviate the domain discrepancy, they fail to fully utilize the unlabeled data in the target domain, resulting in limited performance improvement. 3. **Lack of fine - grained features**: Existing methods usually use the overall convolutional neural network to extract global features, ignoring more fine - grained transferable features, thus affecting the effect of domain adaptation. ### Solutions To solve the above problems, the paper proposes the following innovations: 1. **Multi - Granularity Representation Learning (MGR)**: A two - directional fine - grained partial feature learning network is designed to extract fine - grained features in both horizontal and vertical directions, in order to learn more discriminative and transferable features. 2. **Group Contrastive Learning Module (GCL)**: Pseudo - labels are generated through clustering, and group contrastive learning is used for efficient learning in the target domain, further improving the domain adaptation performance. 3. **Iterative Adaptation Learning Scheme**: Combining MGR and GCL, the model is continuously optimized in an iterative manner, gradually improving the performance on the target domain. ### Experimental verification The paper has carried out extensive experiments on three typical vehicle re - identification datasets (VeRi - 776, VehicleID, and VehicleX). The results show that the proposed MGR - GCL framework significantly outperforms the existing state - of - the - art methods in all settings. ### Summary By introducing the multi - granularity representation learning and group contrastive learning modules, this paper effectively solves the domain adaptation problem in unsupervised vehicle re - identification and significantly improves the performance of the model on the target domain.