3 个月前

自 paced 对比学习与混合记忆机制在域适应目标重识别中的应用

自 paced 对比学习与混合记忆机制在域适应目标重识别中的应用

摘要

领域自适应目标重识别(Domain Adaptive Object Re-ID)旨在将标注源域中学习到的知识迁移至未标注的目标域,以应对开放类重识别问题。尽管当前基于伪标签的先进方法已取得显著进展,但由于域间差异以及聚类性能不佳,这些方法未能充分挖掘所有有价值的信息。为解决上述问题,本文提出一种新颖的自步对比学习框架,该框架结合了混合记忆机制。该混合记忆能够动态生成三类监督信号:源域的类别级信号、目标域的聚类级信号以及未聚类实例级信号,用于指导特征表示的学习。与传统的对比学习策略不同,所提出的框架能够联合区分源域类别、目标域聚类以及未聚类实例。尤为重要的是,所提出的自步学习机制可逐步构建更可靠的聚类结构,从而持续优化混合记忆与学习目标,这一机制被证实是实现卓越性能的关键所在。实验结果表明,本方法在多个目标重识别领域自适应任务中均优于现有最先进方法,甚至在无需额外标注的情况下,还能提升源域上的性能。此外,该方法在无监督目标重识别任务中的泛化版本,在Market-1501和MSMT17基准测试上分别超越现有最先进算法16.7%和7.9%。

代码仓库

GJTNB/reading-memo
pytorch
GitHub 中提及
open-mmlab/OpenUnReID
pytorch
GitHub 中提及
yxgeee/SpCL
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
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

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
自 paced 对比学习与混合记忆机制在域适应目标重识别中的应用 | 论文 | HyperAI超神经