摘要

行人重识别（Person Re-Identification, Re-ID）旨在跨不同摄像头识别同一行人的图像。然而，不同数据集之间的域差异给将基于一个数据集训练的Re-ID模型迁移到另一个数据集带来了显著挑战。当前最先进的无监督域自适应方法通过在目标域上使用聚类算法生成伪标签，并基于这些伪标签优化模型，从而实现从源域到目标域的知识迁移。尽管这些方法取得了当前最优的性能，但聚类过程不可避免地引入了标签噪声，而这一问题在现有方法中被忽视。这些带有噪声的伪标签严重制约了模型在目标域上进一步优化特征表示的能力。为缓解噪声伪标签的影响，本文提出一种新颖的无监督框架——互惠均值教学（Mutual Mean-Teaching, MMT），通过离线精炼的硬伪标签与在线精炼的软伪标签，以交替训练的方式，实现对目标域特征的更优学习。此外，主流的Re-ID模型通常同时采用分类损失（classification loss）与三元组损失（triplet loss）以达到最佳性能。然而，传统三元组损失无法直接与软化后的伪标签协同工作。为此，本文提出一种新型的软化Softmax三元组损失（soft softmax-triplet loss），支持基于软化伪三元组标签的联合学习，从而实现更优的域自适应性能。所提出的MMT框架在四个无监督域自适应任务上均取得了显著提升：在Market-to-Duke、Duke-to-Market、Market-to-MSMT和Duke-to-MSMT任务中，mAP分别提升了14.4%、18.2%、13.1%和16.4%。代码已开源，地址为：https://github.com/yxgeee/MMT。