FET-FGVC: Feature-enhanced transformer for fine-grained visual classification

被引:20
作者
Chen, Huazhen [1 ]
Zhang, Haimiao [2 ]
Liu, Chang [2 ]
An, Jianpeng [1 ]
Gao, Zhongke [1 ]
Qiu, Jun [2 ]
机构
[1] Tianjin Univ, Sch Elect & Informat Engn, Tianjin 300072, Peoples R China
[2] Beijing Informat Sci & Technol Univ, Inst Appl Math, Beijing 100192, Peoples R China
关键词
Fine-grained visual classification (FGVC); Transformer; Graph convolutional network (GCN); Feature enhancement; NETWORK;
D O I
10.1016/j.patcog.2024.110265
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The challenge of Fine-grained visual classification (FGVC) comes from the small variations between classes and the large variations within classes. Inspired by the fact that identifying bird species focuses not only on the global features of the subject area but also on the subtle details of the local area, we propose a featureenhanced Transformer to improve the performance of FGVC. Our proposed method consists of a Dynamic Swin Transformer backbone for extracting comprehensive global image features through continuous attention aggregation, a GCN-based local branch for separating and enhancing local features in different regions, and a pairwise feature interaction (PFI) module for enhancing global features through interactions between image pairs. We conducted extensive experiments on five FGVC datasets to demonstrate the superiority of our method. By fusing the enhanced global and local features, our method achieves the best accuracy compared to existing methods. Our method has an advantage in terms of computational efficiency.
引用
收藏
页数:13
相关论文
共 61 条
[31]   Complemental Attention Multi-Feature Fusion Network for Fine-Grained Classification [J].
Miao, Zhuang ;
Zhao, Xun ;
Wang, Jiabao ;
Li, Yang ;
Li, Hang .
IEEE SIGNAL PROCESSING LETTERS, 2021, 28 :1983-1987
[32]  
Raghu M, 2021, ADV NEUR IN, V34
[33]   Dynamic Spatial Sparsification for Efficient Vision Transformers and Convolutional Neural Networks [J].
Rao, Yongming ;
Liu, Zuyan ;
Zhao, Wenliang ;
Zhou, Jie ;
Lu, Jiwen .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (09) :10883-10897
[34]   Learning Affinity from Attention: End-to-End Weakly-Supervised Semantic Segmentation with Transformers [J].
Ru, Lixiang ;
Zhan, Yibing ;
Yu, Baosheng ;
Du, Bo .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :16825-16834
[35]  
Sun GL, 2020, AAAI CONF ARTIF INTE, V34, P12047
[36]   SIM-Trans: Structure Information Modeling Transformer for Fine-grained Visual Categorization [J].
Sun, Hongbo ;
He, Xiangteng ;
Peng, Yuxin .
PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2022, 2022, :5853-5861
[37]   The iNaturalist Species Classification and Detection Dataset [J].
Van Horn, Grant ;
Mac Aodha, Oisin ;
Song, Yang ;
Cui, Yin ;
Sun, Chen ;
Shepard, Alex ;
Adam, Hartwig ;
Perona, Pietro ;
Belongie, Serge .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :8769-8778
[38]  
Van Horn G, 2015, PROC CVPR IEEE, P595, DOI 10.1109/CVPR.2015.7298658
[39]  
Wah Catherine, 2011, Technical report, V2, P6
[40]   Multi-Path Deep CNNs for Fine-Grained Car Recognition [J].
Wang, Huibing ;
Peng, Jinjia ;
Zhao, Yanzhu ;
Fu, Xianping .
IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY, 2020, 69 (10) :10484-10493