中文

用于细粒度表示学习的属性感知注意力模型

计算机视觉与模式识别 2019-12-23 v2

摘要

如何学习具有判别性的细粒度表示是许多计算机视觉应用的关键,如行人重识别、细粒度分类、细粒度图像检索等。以往的大多数方法专注于学习度量或集成以推导更好的全局表示,通常缺乏局部信息。基于上述考虑,我们提出了一种新颖的属性感知注意力模型(A3MA^3M),该模型能够以端到端的方式同时学习局部属性表示和全局类别表示。所提出的模型包含两个注意力模型:属性引导注意力模块利用属性信息帮助选择不同区域的类别特征,同时类别引导注意力模块借助类别线索选择不同属性的局部特征。通过这种属性与类别互惠的过程,局部和全局特征相互增益。最终,所得特征包含了更多用于图像识别的内在信息,而非噪声和无关特征。在 Market-1501、CompCars、CUB-200-2011 和 CARS196 上进行的大量实验证明了 A3MA^3M 的有效性。代码可在 https://github.com/iamhankai/attribute-aware-attention 获取。

关键词

引用

@article{arxiv.1901.00392,
  title  = {Attribute-Aware Attention Model for Fine-grained Representation Learning},
  author = {Kai Han and Jianyuan Guo and Chao Zhang and Mingjian Zhu},
  journal= {arXiv preprint arXiv:1901.00392},
  year   = {2019}
}

备注

Accepted by ACM Multimedia 2018 (Oral). Code is available at https://github.com/iamhankai/attribute-aware-attention