arrow
Return

Multi-Granularity Part Sampling Attention for Fine-Grained Visual Classification

delete2024-01-01
delete0
PRE
AI
J
Jiahui Wang
Q
Qin Xu *
B
Bo Jiang *
B
Bin Luo
唐金辉 cover
唐金辉 (Jinhui Tang)
DOI:10.1109/TIP.2024.3441813delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Fine-grained visual classification aims to classify similar sub-categories with the challenges of large variations within the same sub-category and high visual similarities between different sub-categories. Recently, methods that extract semantic parts of the discriminative regions have attracted increasing attention. However, most existing methods extract the part features via rectangular bounding boxes by object detection module or attention mechanism, which makes it difficult to capture the rich shape information of objects. In this paper, we propose a novel Multi-Granularity Part Sampling Attention (MPSA) network for fine-grained visual classification. First, a novel multi-granularity part retrospect block is designed to extract the part information of different scales and enhance the high-level feature representation with discriminative part features of different granularities. Then, to extract part features of various shapes at each granularity, we propose part sampling attention, which can sample the implicit semantic parts on the feature maps comprehensively. The proposed part sampling attention not only considers the importance of sampled parts but also adopts the part dropout to reduce the overfitting issue. In addition, we propose a novel multi-granularity fusion method to highlight the foreground features and suppress the background noises with the assistance of the gradient class activation map. Experimental results demonstrate that the proposed MPSA achieves state-of-the-art performance on four commonly used fine-grained visual classification benchmarks. The source code is publicly available at https://github.com/mobulan/MPSA.
Keywords:
Feature extraction
Semantics
Visualization
Shape
Location awareness
Attention mechanisms
Transformers
Fine-grained visual classification
multi-granularity
part sampling
attention mechanism

Journal

IEEE Transactions on Image Processing cover
IEEE Transactions on Image Processing
IF:
13.7
Papers:
1.0W
Citations:
8.4W

Organization

A
anhui university
Scholars:
1.9W
Papers: 1.2W
Citations: 24