我們在ACM MM 2018上的Oral論文《Attribute-Aware Attention Model for Fine-grained Representation Learning》開源啦,歡迎討論。

GitHub程式碼:

【細粒度分類、行人再識別】Attribute-Aware Attention開源啦!

在這篇論文中,我們探討了類別和屬性之間的關係和如何讓兩者互相幫助。我們的模型同時學習了類別特徵和屬性特徵。在兩種特徵的基礎上,屬性引導的注意力模組利用屬性來幫助選擇不同區域的區域性特徵,類別引導的注意力模組在類別資訊的幫助下選擇不同的屬性特徵。這是一個屬性-類別互惠的過程,最終經過注意力 機制選擇後的特徵包含更具細節和判別性的資訊,有利於類別分類以及屬性分類。

【細粒度分類、行人再識別】Attribute-Aware Attention開源啦!

Attribute-Aware Attention Model

在精細分類CUB-200-2011資料集上的實驗結果:

【細粒度分類、行人再識別】Attribute-Aware Attention開源啦!

此外,我們還在行人再識別、影象檢索等任務進行了實驗,效果也很棒。

總感覺英文論文翻譯成中文有些失真,大家感興趣直接去原文了解詳情吧:

Attribute-Aware Attention Model for Fine-grained Representation Learning

Citation

Please use the following bibtex to cite our work:

@inproceedings{han2018attribute,

title={Attribute-Aware Attention Model for Fine-grained Representation Learning},

author={Han, Kai and Guo, Jianyuan and Zhang, Chao and Zhu, Mingjian},

booktitle={Proceedings of the 26th ACM international conference on Multimedia},

pages={2040——2048},

year={2018},

organization={ACM}

}