3 个月前

基于在线关系正则化的结构化领域自适应方法用于无监督行人重识别

基于在线关系正则化的结构化领域自适应方法用于无监督行人重识别

摘要

无监督域自适应(Unsupervised Domain Adaptation, UDA)旨在将基于带标签源域数据集训练的模型迁移到未标注的目标域数据集上。在开放集行人重识别(re-ID)任务中,UDA面临更大的挑战,因为源域与目标域之间的身份类别(类)完全不重叠。以往的主要研究方向之一是基于域迁移的方法,然而近年来该方法因性能逊于基于伪标签的方法而逐渐失去关注。我们认为,域迁移在挖掘源域有价值数据方面具有巨大潜力,但现有方法在迁移过程中缺乏有效的正则化机制。具体而言,以往方法仅关注迁移图像的身份一致性,而忽略了迁移过程中样本间的相互关系。为应对上述挑战,本文提出一种端到端的结构化域自适应框架,并引入在线关系一致性正则化项。在训练过程中,行人特征编码器被优化以实时建模样本间的相互关系,从而监督关系一致性的域迁移过程;而经过有效迁移生成的高质量图像又能反向提升编码器的性能。在此基础上,编码器还可通过伪标签进一步优化:利用具有真实身份标签的源域到目标域迁移图像,以及具有伪身份标签的目标域图像,进行联合训练。实验结果表明,所提出的框架在多个行人重识别无监督域自适应任务中均达到了当前最优性能。此外,借助本框架生成的“合成到真实”域迁移图像,我们在2020年视觉域自适应挑战赛(Visual Domain Adaptation Challenge, VisDA)中取得了第二名的优异成绩。

代码仓库

GJTNB/reading-memo
pytorch
GitHub 中提及
yxgeee/VisDA-ECCV20
官方
pytorch
yxgeee/SDA
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
unsupervised-domain-adaptation-on-duke-toSDA
mAP: 70.0
rank-1: 86.9
rank-10: 96.3
rank-5: 94.4
unsupervised-domain-adaptation-on-duke-to-1SDA
mAP: 25.6
rank-1: 54.4
rank-10: 71.3
rank-5: 66.4
unsupervised-domain-adaptation-on-market-toSDA
mAP: 61.4
rank-1: 76.5
rank-10: 89.7
rank-5: 86.6
unsupervised-domain-adaptation-on-market-to-1SDA
mAP: 23.2
rank-1: 49.5
rank-10: 67.7
rank-5: 62.2

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于在线关系正则化的结构化领域自适应方法用于无监督行人重识别 | 论文 | HyperAI超神经