Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Hierarchical Mask-Enhanced Dual Reconstruction Network for Few-Shot Fine-Grained Image Classification

About

Few-shot fine-grained image classification (FS-FGIC) is challenging as it requires distinguishing visually similar subclasses with extremely limited labeled examples. Existing methods suffer from critical limitations: metric-based methods lose spatial information and misalign local features, while reconstruction-based methods underuse hierarchical feature information and lack selective focus on discriminative key regions. We propose the Hierarchical Mask-enhanced Dual Reconstruction Network (HMDRN), integrating dual-layer feature reconstruction with mask-enhanced feature processing. HMDRN leverages complementary visual information from different network hierarchies via learnable weights, balancing high-level semantic representations with mid-level structural details. It incorporates a spatial binary mask-enhanced transformer module that selectively enhances discriminative regions while filtering background noise. On three fine-grained datasets, HMDRN consistently outperforms state-of-the-art methods with both Conv-4 and ResNet-12 backbones. Ablation studies validate each component's effectiveness, showing dual-layer reconstruction enhances inter-class discrimination while mask-enhanced transformation reduces intra-class variations.

Ning Luo, Meiyin Hu, Huan Wan, Yanyan Yang, Zhuohang Jiang, Xin Wei• 2025

Related benchmarks

TaskDatasetResultRank
Few-shot classificationCUB--
104
5-way Few-shot ClassificationCUB
Accuracy93.5
99
5-way Few-shot ClassificationCars
Accuracy92.8
98
5-way Few-shot ClassificationDogs
Accuracy82.06
42
Few-shot Image ClassificationStanfordCars--
33
Few-shot classificationStanford Dogs
Accuracy (1-shot)78.88
27
Showing 6 of 6 rows

Other info

Follow for update