HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Learning Class Unique Features in Fine-Grained Visual Classification

Runkai Zheng Zhijia Yu Yinqi Zhang Chris Ding Hei Victor Cheng Li Liu

Learning Class Unique Features in Fine-Grained Visual Classification

Abstract

A major challenge in Fine-Grained Visual Classification (FGVC) is distinguishing various categories with high inter-class similarity by learning the feature that differentiate the details. Conventional cross entropy trained Convolutional Neural Network (CNN) fails this challenge as it may suffer from producing inter-class invariant features in FGVC. In this work, we innovatively propose to regularize the training of CNN by enforcing the uniqueness of the features to each category from an information theoretic perspective. To achieve this goal, we formulate a minimax loss based on a game theoretic framework, where a Nash equilibria is proved to be consistent with this regularization objective. Besides, to prevent from a feasible solution of minimax loss that may produce redundant features, we present a Feature Redundancy Loss (FRL) based on normalized inner product between each selected feature map pair to complement the proposed minimax loss. Superior experimental results on several influential benchmarks along with visualization show that our method gives full play to the performance of the baseline model without additional computation and achieves comparable results with state-of-the-art models.

Benchmarks

BenchmarkMethodologyMetrics
fine-grained-image-classification-on-cub-200-1DenseNet161+MM+FRL
Accuracy: 88.5
fine-grained-image-classification-on-fgvcDenseNet161+MM+FRL
Accuracy: 94.0 %
fine-grained-image-classification-on-stanfordDenseNet161+MM+FRL
Accuracy: 95.2%
image-classification-on-cifar-10ResNet-18+MM+FRL
Percentage correct: 95.33
image-classification-on-cifar-100ResNet-18+MM+FRL
Percentage correct: 76.64
image-classification-on-stl-10ResNet-18+MM+FRL
Percentage correct: 85.42

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
Learning Class Unique Features in Fine-Grained Visual Classification | Papers | HyperAI