Style Normalization and Restitution for Generalizable Person Re-identification
About
Existing fully-supervised person re-identification (ReID) methods usually suffer from poor generalization capability caused by domain gaps. The key to solving this problem lies in filtering out identity-irrelevant interference and learning domain-invariant person representations. In this paper, we aim to design a generalizable person ReID framework which trains a model on source domains yet is able to generalize/perform well on target domains. To achieve this goal, we propose a simple yet effective Style Normalization and Restitution (SNR) module. Specifically, we filter out style variations (e.g., illumination, color contrast) by Instance Normalization (IN). However, such a process inevitably removes discriminative information. We propose to distill identity-relevant feature from the removed information and restitute it to the network to ensure high discrimination. For better disentanglement, we enforce a dual causal loss constraint in SNR to encourage the separation of identity-relevant features and identity-irrelevant features. Extensive experiments demonstrate the strong generalization capability of our framework. Our models empowered by the SNR modules significantly outperform the state-of-the-art domain generalization approaches on multiple widely-used person ReID benchmarks, and also show superiority on unsupervised domain adaptation.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Person Re-Identification | Market1501 (test) | Rank-1 Accuracy94.4 | 1264 | |
| Person Re-Identification | Duke MTMC-reID (test) | Rank-185.5 | 1018 | |
| Person Re-Identification | Market 1501 | mAP84.7 | 999 | |
| Person Re-Identification | DukeMTMC-reID | Rank-1 Acc85.5 | 648 | |
| Person Re-Identification | MSMT17 (test) | Rank-1 Acc22 | 499 | |
| Person Re-Identification | Market-1501 (test) | Rank-194.4 | 384 | |
| Person Re-Identification | VIPeR | Rank-155.1 | 182 | |
| Person Re-Identification | Market-1501 to DukeMTMC-reID (test) | Rank-176.3 | 172 | |
| Person Re-Identification | VIPeR (test) | Top-1 Accuracy55.1 | 113 | |
| Person Re-Identification | MSMT17 source: DukeMTMC-reID (test) | Rank-1 Acc69.2 | 83 |