4 个月前

ABD-Net:注意力驱动且多样化的行人重识别

ABD-Net:注意力驱动且多样化的行人重识别

摘要

注意力机制已被证明在行人重识别(Re-ID)中具有有效性。然而,所学习到的注意力特征嵌入通常既不具备自然多样性,也不具备非相关性,这会降低基于欧氏距离的检索性能。我们认为,强制实现多样性可以极大地补充注意力机制的优势。为此,我们提出了一种注意且多样化的网络(ABD-Net),该网络在整个网络中无缝集成了注意力模块和多样性正则化,以学习更具代表性、鲁棒性和区分性的特征。具体而言,我们引入了一对互补的注意力模块,分别关注通道聚合和位置感知。此外,还推导出一种新的高效正交约束形式,用于强制隐藏激活和权重的正交性。通过细致的消融研究,我们验证了所提出的注意力和多样性项各自对ABD-Net性能提升的贡献。在三个流行的基准数据集上,ABD-Net的表现始终优于现有的最先进方法。

代码仓库

TAMU-VITA/ABD-Net
官方
pytorch
GitHub 中提及
hsfzxjy/dprn
pytorch
GitHub 中提及
vita-group/abd-net
pytorch
GitHub 中提及

基准测试

基准方法指标
person-re-identification-on-dukemtmc-reidABD-Net (ResNet-50)
Rank-1: 89.0
mAP: 78.59
person-re-identification-on-market-1501ABD-Net (ResNet-50)
Rank-1: 95.6
mAP: 88.28
person-re-identification-on-market-1501-cABD-Net
Rank-1: 29.65
mAP: 9.81
mINP: 0.26
person-re-identification-on-msmt17ABD-Net (ResNet-50)
Rank-1: 82.3
mAP: 60.8

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
ABD-Net:注意力驱动且多样化的行人重识别 | 论文 | HyperAI超神经