Bidirectional Attention-Recognition Model for Fine-Grained Object Classification

被引:39
作者
Liu, Chuanbin [1 ]
Xie, Hongtao [1 ]
Zha, Zhengjun [1 ]
Yu, Lingyun [1 ]
Chen, Zhineng [2 ]
Zhang, Yongdong [1 ]
机构
[1] Univ Sci & Technol China, Sch Informat Sci & Technol, Hefei 230026, Peoples R China
[2] Chinese Acad Sci, Inst Automat, Beijing 100190, Peoples R China
关键词
Fine-grained object classification; interpretable machine learning; visual attention; pattern recognition; data augmentation;
D O I
10.1109/TMM.2019.2954747
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Fine-grained object classification (FGOC) is a challenging research topic in multimedia computing with machine learning, which faces two pivotal conundrums: focusing attention on the discriminate part regions, and then processing recognition with the part-based features. Existing approaches generally adopt a unidirectional two-step structure, that first locate the discriminate parts and then recognize the part-based features. However, they neglect the truth that part localization and feature recognition can be reinforced in a bidirectional process. In this paper, we propose a novel bidirectional attention-recognition model (BARM) to actualize the bidirectional reinforcement for FGOC. The proposed BARM consists of one attention agent for discriminate part regions proposing and one recognition agent for feature extraction and recognition. Meanwhile, a feedback flow is creatively established to optimize the attention agent directly by recognition agent. Therefore, in BARM the attention agent and the recognition agent can reinforce each other in a bidirectional way and the overall framework can be trained end-to-end without neither object nor parts annotations. Moreover, a novel Multiple Random Erasing data augmentation is proposed, and it exhibits impressive pertinency and superiority for FGOC. Conducted on several extensive FGOC benchmarks, BARM outperforms the present state-of-the-art methods in classification accuracy. Furthermore, BARM exhibits a clear interpretability and keeps consistent with the human perception in visualization experiments.
引用
收藏
页码:1785 / 1795
页数:11
相关论文
共 67 条
[41]   Hide-and-Seek: Forcing a Network to be Meticulous for Weakly-supervised Object and Action Localization [J].
Singh, Krishna Kumar ;
Lee, Yong Jae .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :3544-3553
[42]   Multi-Attention Multi-Class Constraint for Fine-grained Image Recognition [J].
Sun, Ming ;
Yuan, Yuchen ;
Zhou, Feng ;
Ding, Errui .
COMPUTER VISION - ECCV 2018, PT XVI, 2018, 11220 :834-850
[43]   Selective Search for Object Recognition [J].
Uijlings, J. R. R. ;
van de Sande, K. E. A. ;
Gevers, T. ;
Smeulders, A. W. M. .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2013, 104 (02) :154-171
[44]  
VASWANI A, 2017, ADV NEURAL INFORM PR, P5998, DOI DOI 10.48550/ARXIV.1706.03762
[45]  
Wah C, 2011, CALTECH UCSD BIRDS 2
[46]   Learning a Discriminative Filter Bank within a CNN for Fine-grained Recognition [J].
Wang, Yaming ;
Morariu, Vlad I. ;
Davis, Larry S. .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :4148-4157
[47]   Mining Discriminative Triplets of Patches for Fine-Grained Classification [J].
Wang, Yaming ;
Choi, Jonghyun ;
Morariu, Vlad I. ;
Davis, Larry S. .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :1163-1172
[48]   Mask-CNN: Localizing parts and selecting descriptors for fine-grained bird species categorization [J].
Wei, Xiu-Shen ;
Xie, Chen-Wei ;
Wu, Jianxin ;
Shen, Chunhua .
PATTERN RECOGNITION, 2018, 76 :704-714
[49]   Low-Complexity Tree Search-based Detection Algorithms for Generalized Spatial Modulation Aided Single Carrier Systems [J].
Xiao, Lixia ;
Yang, Ping ;
Zhao, Yan ;
Xiao, Yue ;
Liu, Jiang ;
Li, Shaoqian .
2016 IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS (ICC), 2016,
[50]  
Xiao TJ, 2015, PROC CVPR IEEE, P842, DOI 10.1109/CVPR.2015.7298685