3 个月前

ICE:用于无监督行人重识别的跨实例对比编码

ICE:用于无监督行人重识别的跨实例对比编码

摘要

无监督行人重识别(ReID)旨在无需标注信息的情况下学习具有判别性的身份特征。近年来,自监督对比学习因其在无监督表示学习中的高效性而受到越来越多关注。实例对比学习的核心思想是将同一实例在不同增强视图下的表示进行匹配。然而,以往的对比学习方法对不同实例之间的关系尚未充分挖掘,尤其是在实例级别的对比损失设计方面。为解决这一问题,本文提出一种新型的跨实例对比编码方法(Inter-instance Contrastive Encoding, ICE),该方法利用实例间的成对相似度得分,以增强现有的类别级别对比ReID方法。具体而言,我们首先将成对相似度排序结果作为one-hot硬伪标签,用于强化困难实例的对比学习,从而有效降低类内差异。随后,我们采用相似度得分作为软伪标签,以增强原始视图与增强视图之间的表示一致性,使模型对数据增强扰动具有更强的鲁棒性。在多个大规模行人重识别数据集上的实验结果表明,所提出的无监督方法ICE具有显著有效性,其性能甚至可与有监督方法相媲美。相关代码已开源,地址为:https://github.com/chenhao2345/ICE。

代码仓库

chenhao2345/ICE
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
unsupervised-person-re-identification-on-4ICE
MAP: 82.3
Rank-1: 93.8
Rank-10: 98.4
Rank-5: 97.6
unsupervised-person-re-identification-on-5ICE
MAP: 69.9
Rank-1: 83.3
Rank-10: 94.1
Rank-5: 91.5

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
ICE:用于无监督行人重识别的跨实例对比编码 | 论文 | HyperAI超神经