HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

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

Asish Bera Zachary Wharton Yonghuai Liu Nik Bessis Ardhendu Behera

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.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
fine-grained-image-classification-on-cub-200SR-GNN
Accuracy: 91.9%
fine-grained-image-classification-on-fgvcSR-GNN
Accuracy: 95.4
FLOPS: 9.8
PARAMS: 30.9
fine-grained-image-classification-on-nabirdsSR-GNN
Accuracy: 91.2%
fine-grained-image-classification-on-oxfordSR-GNN
Accuracy: 97.9%
FLOPS: 9.8
PARAMS: 30.9
fine-grained-image-classification-on-stanfordSR-GNN
Accuracy: 96.1
FLOPS: 9.8
PARAMS: 30.9
fine-grained-image-classification-on-stanford-1MP
Accuracy: 97.3%

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
SR-GNN: Spatial Relation-aware Graph Neural Network for Fine-Grained Image Categorization | Papers | HyperAI