Uncertainty-Aware Multi-Shot Knowledge Distillation for Image-Based Object Re-Identification
About
Object re-identification (re-id) aims to identify a specific object across times or camera views, with the person re-id and vehicle re-id as the most widely studied applications. Re-id is challenging because of the variations in viewpoints, (human) poses, and occlusions. Multi-shots of the same object can cover diverse viewpoints/poses and thus provide more comprehensive information. In this paper, we propose exploiting the multi-shots of the same identity to guide the feature learning of each individual image. Specifically, we design an Uncertainty-aware Multi-shot Teacher-Student (UMTS) Network. It consists of a teacher network (T-net) that learns the comprehensive features from multiple images of the same object, and a student network (S-net) that takes a single image as input. In particular, we take into account the data dependent heteroscedastic uncertainty for effectively transferring the knowledge from the T-net to S-net. To the best of our knowledge, we are the first to make use of multi-shots of an object in a teacher-student learning manner for effectively boosting the single image based re-id. We validate the effectiveness of our approach on the popular vehicle re-id and person re-id datasets. In inference, the S-net alone significantly outperforms the baselines and achieves the state-of-the-art performance.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Vehicle Re-identification | VeRi-776 (test) | Rank-195.8 | 232 | |
| Vehicle Re-identification | VehicleID (Small) | R-180.9 | 61 | |
| Vehicle Re-identification | VehicleID (Large) | R-176.1 | 39 | |
| Vehicle Re-identification | VeRi-Wild (test 3000) | R1 Accuracy84.5 | 25 | |
| Vehicle Re-identification | VeRi-Wild (test5000) | Rank-1 Accuracy79.3 | 24 | |
| Vehicle Re-identification | VeRi-Wild (Test10000) | R1 Accuracy72.8 | 24 | |
| Vehicle Re-identification | VehicleID | Rank-1 Accuracy80.9 | 23 | |
| Vehicle Re-identification | VehicleID small (test) | Rank-1 Accuracy80.9 | 17 | |
| Vehicle Re-identification | VehicleID (Medium) | Rank-178.8 | 9 | |
| Vehicle Re-identification | VeRi-776 | mAP75.9 | 9 |