HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Self-paced Contrastive Learning with Hybrid Memory for Domain Adaptive Object Re-ID

Yixiao Ge Feng Zhu Dapeng Chen Rui Zhao Hongsheng Li

Self-paced Contrastive Learning with Hybrid Memory for Domain Adaptive Object Re-ID

Abstract

Domain adaptive object re-ID aims to transfer the learned knowledge from the labeled source domain to the unlabeled target domain to tackle the open-class re-identification problems. Although state-of-the-art pseudo-label-based methods have achieved great success, they did not make full use of all valuable information because of the domain gap and unsatisfying clustering performance. To solve these problems, we propose a novel self-paced contrastive learning framework with hybrid memory. The hybrid memory dynamically generates source-domain class-level, target-domain cluster-level and un-clustered instance-level supervisory signals for learning feature representations. Different from the conventional contrastive learning strategy, the proposed framework jointly distinguishes source-domain classes, and target-domain clusters and un-clustered instances. Most importantly, the proposed self-paced method gradually creates more reliable clusters to refine the hybrid memory and learning targets, and is shown to be the key to our outstanding performance. Our method outperforms state-of-the-arts on multiple domain adaptation tasks of object re-ID and even boosts the performance on the source domain without any extra annotations. Our generalized version on unsupervised object re-ID surpasses state-of-the-art algorithms by considerable 16.7% and 7.9% on Market-1501 and MSMT17 benchmarks.

Code Repositories

GJTNB/reading-memo
pytorch
Mentioned in GitHub
open-mmlab/OpenUnReID
pytorch
Mentioned in GitHub
yxgeee/SpCL
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
unsupervised-domain-adaptation-on-duke-toSpCL
mAP: 76.7
rank-1: 90.3
rank-10: 97.7
rank-5: 96.2
unsupervised-domain-adaptation-on-duke-to-1SpCL
mAP: 26.5
rank-1: 53.1
rank-10: 70.5
rank-5: 65.8
unsupervised-domain-adaptation-on-market-toSpCL
mAP: 68.8
rank-1: 82.9
rank-10: 92.5
rank-5: 90.1
unsupervised-domain-adaptation-on-market-to-1SpCl
mAP: 25.4
rank-1: 51.6
rank-10: 69.7
rank-5: 64.3
unsupervised-person-re-identification-on-4SpCL
MAP: 72.6
Rank-1: 87.7
Rank-10: 96.9
Rank-5: 95.2
unsupervised-person-re-identification-on-5SpCL
MAP: 65.3
Rank-1: 81.2
Rank-10: 92.2
Rank-5: 90.3

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Self-paced Contrastive Learning with Hybrid Memory for Domain Adaptive Object Re-ID | Papers | HyperAI