3 个月前

基于耦合优化的领域自适应行人重识别

基于耦合优化的领域自适应行人重识别

摘要

领域自适应行人重识别(Domain Adaptive Person Re-Identification, ReID)由于目标场景中存在领域差异以及标注数据稀缺,面临严峻挑战。为应对上述两大难题,本文提出一种耦合优化方法,包含领域不变映射(Domain-Invariant Mapping, DIM)与全局-局部距离优化(Global-Local Distance Optimization, GLO)两个核心模块。与以往分两阶段进行知识迁移的方法不同,DIM通过将标注数据集与无标注数据集中的图像映射至共享特征空间,实现了更高效的单阶段知识迁移。GLO专为在目标域上采用无监督设置训练ReID模型而设计,不依赖于传统面向监督学习的优化策略,而是引入更多样本参与距离优化过程,显著提升了对噪声标签预测的鲁棒性。同时,GLO融合了全局数据集与局部训练批次中的距离优化机制,从而展现出更优的训练效率。在三个大规模数据集(Market-1501、DukeMTMC-reID 和 MSMT17)上的大量实验表明,所提出的耦合优化方法在性能上大幅超越现有最先进方法。此外,该方法在无监督训练场景下亦表现优异,甚至超越若干近期提出的领域自适应方法。

代码仓库

liu-xb/DIM_GLO
官方
pytorch

基准测试

基准方法指标
unsupervised-person-re-identification-onDIM+GLO
Rank-1: 76.2
mAP: 58.3
unsupervised-person-re-identification-on-1DIM+GLO
Rank-1: 88.3
mAP: 65.1
unsupervised-person-re-identification-on-2DIM+GLO
Rank-1: 49.7
mAP: 20.7
unsupervised-person-re-identification-on-3DIM+GLO
Rank-1: 56.5
mAP: 24.4

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于耦合优化的领域自适应行人重识别 | 论文 | HyperAI超神经