4 个月前

SVDNet 用于行人检索

SVDNet 用于行人检索

摘要

本文提出了一种用于检索问题的SVDNet,特别关注其在行人重识别(re-ID)中的应用。我们将卷积神经网络(CNN)中全连接(FC)层的每个权重向量视为一个投影基。观察发现,这些权重向量通常具有高度相关性。这一问题导致了FC描述符各元素之间的相关性,从而影响了基于欧氏距离的检索性能。为了解决这一问题,本文提出利用奇异值分解(SVD)优化深度表示学习过程。具体而言,通过约束与松弛迭代(RRI)训练方案,我们能够在CNN训练过程中逐步引入正交性约束,从而生成所谓的SVDNet。我们在Market-1501、CUHK03和Duke数据集上进行了实验,结果表明RRI有效降低了投影向量之间的相关性,生成了更具区分性的FC描述符,并显著提高了重识别精度。例如,在Market-1501数据集上,CaffeNet的首位命中率从55.3%提高到80.5%,ResNet-50的首位命中率从73.8%提高到82.3%。

基准测试

基准方法指标
person-re-identification-on-cuhk03-detectedSVDNet-ResNet50
MAP: 37.3
Rank-1: 41.5
person-re-identification-on-cuhk03-detectedSVDNet-CaffeNet
MAP: 24.9
Rank-1: 27.7
person-re-identification-on-dukemtmc-reidSVDNet
Rank-1: 76.7
mAP: 56.8
person-re-identification-on-market-1501SVDNet
Rank-1: 82.3
mAP: 62.1

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SVDNet 用于行人检索 | 论文 | HyperAI超神经