SR-GNN: Spatial Relation-aware Graph Neural Network for Fine-Grained Image Categorization

Abstract

Over the past few years, a significant progress has been made in deep convolutional neural networks (CNNs)-based image recognition. This is mainly due to the strong ability of such networks in mining discriminative object pose and parts information from texture and shape. This is often inappropriate for fine-grained visual classification (FGVC) since it exhibits high intra-class and low inter-class variances due to occlusions, deformation, illuminations, etc. Thus, an expressive feature representation describing global structural information is a key to characterize an object/ scene. To this end, we propose a method that effectively captures subtle changes by aggregating context-aware features from most relevant image-regions and their importance in discriminating fine-grained categories avoiding the bounding-box and/or distinguishable part annotations. Our approach is inspired by the recent advancement in self-attention and graph neural networks (GNNs) approaches to include a simple yet effective relation-aware feature transformation and its refinement using a context-aware attention mechanism to boost the discriminability of the transformed feature in an end-to-end learning process. Our model is evaluated on eight benchmark datasets consisting of fine-grained objects and human-object interactions. It outperforms the state-of-the-art approaches by a significant margin in recognition accuracy.

arXiv commentAccepted manuscript - IEEE Transaction on Image Processing

References in corpus (10)

• A Comprehensive Survey on Graph Neural Networks
• The Devil is in the Channels: Mutual-Channel Loss for Fine-Grained Image Classification
• Object-Part Attention Model for Fine-grained Image Classification
• Weakly Supervised Attention Pyramid Convolutional Neural Network for Fine-Grained Visual Classification
• Weakly Supervised Fine-Grained Image Categorization
• Fine-graind Image Classification via Combining Vision and Language
• Pay attention to the activations: a modular attention mechanism for fine-grained image recognition
• Attend and Guide (AG-Net): A Keypoints-driven Attention-based Deep Network for Image Recognition
• Classification-Specific Parts for Improving Fine-Grained Visual Categorization
• Regional Attention Network (RAN) for Head Pose and Fine-grained Gesture Recognition

Cited by in corpus (3)

• Fine-Grained Sports, Yoga, and Dance Postures Recognition: A Benchmark Analysis
• Deep Neural Networks Fused with Textures for Image Classification
• D3-GNN: Dynamic Distributed Dataflow for Streaming Graph Neural Networks